Agent: Fix should_propagate import in automated_master.py

This commit is contained in:
vakarisz 2022-06-22 14:04:15 +03:00
parent 734fc80c92
commit 5a48bfcfcc
2 changed files with 2 additions and 3 deletions

View File

@ -3,8 +3,6 @@ import threading
import time
from typing import Any, Callable, Dict, Iterable, List, Mapping, Optional, Tuple
from utils.propagation import should_propagate
from common.utils import Timer
from infection_monkey.credential_store import ICredentialsStore
from infection_monkey.i_control_channel import IControlChannel, IslandCommunicationError
@ -15,6 +13,7 @@ from infection_monkey.network import NetworkInterface
from infection_monkey.telemetry.credentials_telem import CredentialsTelem
from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
from infection_monkey.telemetry.post_breach_telem import PostBreachTelem
from infection_monkey.utils.propagation import should_propagate
from infection_monkey.utils.threading import create_daemon_thread, interruptible_iter
from . import Exploiter, IPScanner, Propagator

View File

@ -78,8 +78,8 @@ from infection_monkey.utils.monkey_dir import (
remove_monkey_dir,
)
from infection_monkey.utils.monkey_log_path import get_agent_log_path
from infection_monkey.utils.propagation import should_propagate
from infection_monkey.utils.signal_handler import register_signal_handlers, reset_signal_handlers
from utils.propagation import should_propagate
logger = logging.getLogger(__name__)
logging.getLogger("urllib3").setLevel(logging.INFO)