python如何判断ssh端口

Python可以通过使用socket库连接指定的主机和端口、使用paramiko库进行SSH连接、使用nmap库进行端口扫描等方式来判断SSH端口是否开放。在这几种方法中，使用socket库是一种简单且直接的方式。

下面我们将详细介绍如何使用这些方法来判断SSH端口的状态。

一、使用socket库进行连接测试

socket库是Python内置的标准库，提供了对网络接口的底层访问。我们可以通过尝试连接目标主机的SSH端口（默认是22）来判断端口是否开放。

导入socket库并设置超时时间

首先，我们需要导入socket库，并为socket连接设置一个超时时间，以避免因连接尝试失败而长时间等待。

import socket
def check_ssh_port(host, port=22, timeout=5):
    # 创建一个socket对象
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 设置超时时间
    sock.settimeout(timeout)
    try:
        # 尝试连接目标主机的指定端口
        sock.connect((host, port))
        # 如果连接成功，返回True
        return True
    except (socket.timeout, socket.error):
        # 如果连接失败，返回False
        return False
    finally:
        # 关闭socket连接
        sock.close()

使用函数检查SSH端口

使用上述函数，我们可以轻松地检查目标主机的SSH端口是否开放。

host = '192.168.1.1'
if check_ssh_port(host):
    print(f'SSH port on {host} is open.')
else:
    print(f'SSH port on {host} is closed.')

二、使用paramiko库进行SSH连接

paramiko是一个用Python实现的SSHv2协议库，提供了建立SSH连接的高层接口。我们可以通过尝试建立SSH连接来判断端口是否开放。

安装paramiko库

要使用paramiko库，我们需要先安装它。可以通过pip命令进行安装：

pip install paramiko

使用paramiko尝试建立SSH连接

import paramiko
def check_ssh_with_paramiko(host, port=22, username='your_username', password='your_password'):
    client = paramiko.SSHClient()
    # 自动添加主机到known_hosts
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        # 尝试连接目标主机的SSH服务
        client.connect(host, port=port, username=username, password=password, timeout=5)
        # 如果连接成功，返回True
        return True
    except (paramiko.ssh_exception.SSHException, socket.error):
        # 如果连接失败，返回False
        return False
    finally:
        # 关闭SSH连接
        client.close()
host = '192.168.1.1'
if check_ssh_with_paramiko(host):
    print(f'SSH connection to {host} is successful.')
else:
    print(f'SSH connection to {host} failed.')

三、使用nmap库进行端口扫描

nmap是一款开源的网络探测和安全审计工具。我们可以使用Python的nmap库来扫描目标主机的SSH端口。

安装nmap库

可以通过pip命令安装nmap库：

pip install python-nmap

使用nmap扫描端口

import nmap
def scan_ssh_port(host, port=22):
    nm = nmap.PortScanner()
    try:
        # 扫描目标主机的指定端口
        nm.scan(host, str(port))
        # 获取端口扫描结果
        port_state = nm[host]['tcp'][port]['state']
        return port_state == 'open'
    except Exception as e:
        print(f"Error scanning {host}: {e}")
        return False
host = '192.168.1.1'
if scan_ssh_port(host):
    print(f'SSH port on {host} is open.')
else:
    print(f'SSH port on {host} is closed.')