利用多线程实现ssh并发访问

需求说明


1.在文件中取出所有远程主机IP地址

2. 在shell命令行中接受远程服务器IP地址文件、远程服务器密码以及在远程主机上执行的命令 

3.通过多线程实现在所有的远程服务器上并发执行命令


安装paramiko


paramiko 遵循SSH2协议,支持以加密和认证的方式,进行远程服务器的连接,可以实现远程文件的上传,下载或通过ssh远程执行命令。

pip3 install paramiko


测试是否安装成功


import paramiko


源代码


#导入模块

import paramiko

import sys

import getpass

import threading

import os

#创建函数实现远程连接主机,服务器密码以及远程在主机上执行的命令的功能

def remote_comm(host,pwd,command):

#创建用于连接ssh服务器的实例

    ssh=paramiko.SSHClient()

#设置自动添加主机秘钥

    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

#连接ssh服务器,添加连接的主机,用户名,密码填好

    ssh.connect(hostname=host,username='root',password=pwd)

#在ssh服务器上执行指定命令,返回三项类对象,分别是:输入/输出/错误

    stdin,stdout,stderr=ssh.exec_command(command)

#读取输出

    out =stdout.read()

#读取错误

    error=stderr.read()

#如果有输出

    if out:

#打印主机输出内容

        print('[%s]OUT:\n%s'%(host,out.decode('utf8')))

#如果有错误

    if error:

#打印主机错误信息

        print('[%s]ERROR:\n%s'%(host,out.decode('utf8')))

#程序结束

    ssh.close()

if __name__ == '__main__':

    #设定sys.argv长度,确保remote_comm函数中;参数数量

    if len(sys.argv)!=3:

        print('Usage:%s ipaddr_file "command"'%sys.argv[0])

        exit(1)

    #判断命令行上输入如果不是文件,确保输入的是文件

    if not  os.path.isfile(sys.argv[1]):

        print('NO such file:',sys.argv[1])

        exit(2)

    #fname为存储远程主机ip的文件,用sys.argv位置参数的方法,可以在执行脚本时再输入文件名,更为灵活

    fname=sys.argv[1]

    #command为在远程主机上执行的命令,用sys.argv方法,可以在执行脚本时再输入响应的命令,command为remote_comm函数的第三个参数

    command=sys.argv[2]

    #通过getpass输入远程服务器密码,pwd为remote_comm函数第二个参数

    pwd=getpass.getpass()

    #打开存有远程主机ip的文件

    with open(fname)as fobj:

    #将遍历文件将ip以列表形式存入ips,line.strip()可以去掉每行后的换行符\n

        ips=[line.strip() for line in fobj]

    #循环遍历文件,获取ip地址,ip为remote_comm函数第一个参数

    for ip in ips:

    #将读取到的IP地址作为remote_comm函数实际参数传递给函数,ips中有几个ip地址循环几次

    #创建多线程

        t=threading.Thread(target=remote_comm,args=(ip,pwd,command))

    #启用多线程

    t.start()


代码运行结果


先创建虚拟机,并将虚拟机的ip地址写入文件

这里以一台为例子,可以写多台

vim server_addr.txt

192.168.1.10


#查询虚拟机上root用户的信息

python3 sshcommit.py server_addr.txt "id root"

#输入链接虚拟机的root用户密码

Password:

[192.168.1.10]OUT:

#可以看到执行到的命令结果

uid=0(root) gid=0(root) groups=0(root)


#为虚拟机添加用户zjh

python3 sshcommit.py server_addr.txt "useradd zjh"

#输入虚拟机密码

Password:


#更改刚才创建的zjh用户的密码

python3 sshcommit.py server_addr.txt "echo zjh |passwd --stdin zjh"

Password:

[192.168.1.10]OUT:

#显示zjh用户的密码修改成功

Changing password for user zjh.

passwd: all authentication tokens updated successfully.


在虚拟机上检测是否有zjh用户


[root@zabbix ~]# id zjh

uid=1000(zjh) gid=1000(zjh) 组=1000(zjh)

#可以看到是存在的,证明脚本执行成功


错误检测


参数给少了效果如下:

python3 sshcommit.py server_addr.txt

Usage:sshcommit.py ipaddr_file "command"

可以显示出我们自定义的报错内容信息


参数给多了效果如下:

python3 sshcommit.py server_addr.txt id zjh

Usage:sshcommit.py ipaddr_file "command"

可以显示出我们自定义的报错内容信息


©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容