对python多线程SSH登录并发脚本详解

时间:2021-05-22

测试系统中有一项记录ssh登录日志,需要对此进行并发压力测试。

于是用多线程进行python并发记录

因为需要安装的一些依赖和模块比较麻烦,脚本完成后再用pyinstaller打成exe包分发给其他测试人员一起使用。

1.脚本编写

# -*- coding: utf-8 -*-import paramikoimport threadingimport timelt = []def ssh(a,xh,sp): count = 0 for i in range(0,xh): try: ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect('ip地址',22,'用户名', '密码') ssh.close() print u"线程[%s]第[%s]次登录"%(a,i) if sp != 0: time.sleep(sp) count += 1 except: print u"线程异常,已处理" lt.append(count) if __name__ == "__main__": figlet = ''' _____ _____ _ | ___| | _ \ | | | |__ | |_| | | | | __| | _ { | | | | | |_| | | | |_| |_____/ |_| Code by FBI. ''' print figlet print u"认证攻击次数=线程数*每个线程认证攻击次数" print u"请输入线程数:" xc = raw_input() print u"请输入每个线程攻击次数:" xh = raw_input() print u"请输入每个线程延迟时间(秒),0为不休眠:" sp = raw_input() try: print u"预计总共发送认证攻击%s次"%(int(xc)*int(xh)) threads = [] for j in range(int(xc)): threads.append(threading.Thread(target=ssh,args=(j,int(xh),int(sp),))) for t in threads: t.start() print t.name t.join() print lt count = 0 for count in lt: count += count print u"程序执行完毕总共发送认证攻击【%s】次" % count except ValueError,e: print u"因为输入不规范导致程序出现错误,请输入数字"

2.pyinstaller制作exe程序

下载pyinstaller后

在根目录中cmd中执行python setup.py install安装pyinstaller

安装完成后执行命令打成exe文件

python pyinstaller.py -F 文件路径

3.执行效果

如图:

以上这篇对python多线程SSH登录并发脚本详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章