在同一個程序中運行2個阻塞循環

Question

我想在同一個程序中運行2個阻塞循環。 在我的程序中，我使用nfqueue來攔截數據包。當隊列被創建時，它開始等待數據包並阻止程序。當數據包到達時，它會調用cb（）函數，然後再次開始監聽新數據包。

這裏是我的程序：

import nfqueue, socket 
from scapy.all import * 
import os 

os.system('iptables -t mangle -A PREROUTING -j NFQUEUE --queue-num 1') 
os.system('iptables -t mangle -A POSTROUTING -j NFQUEUE --queue-num 2') 


count = 0 

def cb(payload): 
    global count 
    count +=1 
    data = payload.get_data() 
    p = IP(data) 

    print str(count) + ": TOS  = " + str(p.tos) 
    payload.set_verdict(nfqueue.NF_ACCEPT) 


def run_queue(queue_num): 
    print "Preparing the queue" 
    q = nfqueue.queue() 
    q.open() 
    q.unbind(socket.AF_INET) 
    q.bind(socket.AF_INET) 
    q.set_callback(cb) 
    q.create_queue(queue_num) 

    try: 
     print "Running the queue" 
     q.try_run() 

    except KeyboardInterrupt, e: 
     print "interruption" 
     q.unbind(socket.AF_INET) 
     q.close() 


run_queue(1) 
run_queue(2) 

我怎麼能在同一程序運行2個以上這些封環？

任何幫助將不勝感激。 謝謝！

Answer 1

您可以在自己的線程運行的每個循環中，像這樣：

import nfqueue, socket 
from threading import Thread 
from scapy.all import * 
import os 

os.system('iptables -t mangle -A PREROUTING -j NFQUEUE --queue-num 1') 
os.system('iptables -t mangle -A POSTROUTING -j NFQUEUE --queue-num 2') 


count = 0 

def cb(payload): 
    global count 
    count +=1 
    data = payload.get_data() 
    p = IP(data) 

    print str(count) + ": TOS  = " + str(p.tos) 
    payload.set_verdict(nfqueue.NF_ACCEPT) 


def run_queue(queue_num): 
    print "Preparing the queue" 
    q = nfqueue.queue() 
    q.open() 
    q.unbind(socket.AF_INET) 
    q.bind(socket.AF_INET) 
    q.set_callback(cb) 
    q.create_queue(queue_num) 

    try: 
     print "Running the queue" 
     q.try_run() 

    except KeyboardInterrupt, e: 
     print "interruption" 
     q.unbind(socket.AF_INET) 
     q.close() 

if __name__ == "__main__": 
    t1 = Thread(target=run_queue, args=(1,)) 
    t1.start() 
    t2 = Thread(target=run_queue, args=(2,)) 
    t2.start() 
    t1.join() 
    t2.join()