Python Scapy использует несколько процессоров

Question

Я борюсь с тем, что мои функции в python, для создания пакетов для Scapy, используют только 1 процессор за раз. Я искал множественную обработку, но я действительно не знаю, как я мог ее реализовать. Я также пытался использовать ThreadPool, но результат тот же.

Вот отрезок моего кода.

def code(a,b,c) 

    while a < b : 
     ## build pkt 100000 times 

    send(pkt) 

def main(): 
    ##defining some intregers a b c 
    code(a,b,c) 

if __name__ == "__main__": 

При этом функция while вычисляется только одним процессором. Это занимает слишком много времени, чтобы генерировать тысячи пакетов.

Есть ли простой способ сделать это?

Answer 1

Вы должны разбить список пакетов в кусках и позволить процессам выполнять эту работу за вас.

EDIT: требуется Python3 или параллельный модуль. Предыдущие примеры просто дали идею. Это рабочее решение. Не стесняйтесь изменять постоянные значения.

from concurrent.futures import ProcessPoolExecutor 


START = 0 
END = 1000000 
CHUNK_SIZE = 1000 


chunks = range(START, END, CHUNK_SIZE) 


def code(offset): 
    packets = [] 
    chunk = offset + CHUNK_SIZE 

    for index in range(offset, chunk): 
     packet = build_packet(index) 
     packets.append(packet) 

    return packets 


with ProcessPoolExecutor() as executor: 
    packets = [] 

    for chunk in executor.map(code, chunks): 
     packets += chunk 

send(packets)