Bugfix. Make both OSes use MT and not MP to avoid pickling issues.

This commit is contained in:
Daniel Goldberg 2019-10-03 15:06:18 +03:00
parent c767250760
commit 4cba1e9327
1 changed files with 1 additions and 6 deletions
monkey/infection_monkey/network

View File

@ -1,17 +1,12 @@
import time import time
import logging import logging
from multiprocessing.dummy import Pool
from common.network.network_range import NetworkRange from common.network.network_range import NetworkRange
from infection_monkey.config import WormConfiguration from infection_monkey.config import WormConfiguration
from infection_monkey.model.victim_host_generator import VictimHostGenerator from infection_monkey.model.victim_host_generator import VictimHostGenerator
from infection_monkey.network.info import local_ips, get_interfaces_ranges from infection_monkey.network.info import local_ips, get_interfaces_ranges
from infection_monkey.network import TcpScanner, PingScanner from infection_monkey.network import TcpScanner, PingScanner
from infection_monkey.utils.environment import is_windows_os
if is_windows_os():
from multiprocessing.dummy import Pool
else:
from multiprocessing import Pool
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)