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

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

2021-05-28 00:12黑面狐 Python

今天小编就为大家分享一篇对python多线程SSH登录并发脚本详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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

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

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

1.脚本编写

				?

									# -*- coding: utf-8 -*-

									import paramiko

									import threading

									import time

									lt = []

									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"因为输入不规范导致程序出现错误，请输入数字"