From 019f2c1403239292f7b9153681a866a8f7539809 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 31 Jan 2022 19:54:40 -0500 Subject: [PATCH] Agent: Implement ping scanning in Puppet Fixes #1602 PR #1691 --- monkey/infection_monkey/puppet/puppet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/puppet/puppet.py b/monkey/infection_monkey/puppet/puppet.py index 41ed99250..af550e4cb 100644 --- a/monkey/infection_monkey/puppet/puppet.py +++ b/monkey/infection_monkey/puppet/puppet.py @@ -2,6 +2,7 @@ import logging import threading from typing import Dict +from infection_monkey import network from infection_monkey.i_puppet import ( ExploiterResultData, FingerprintData, @@ -33,7 +34,7 @@ class Puppet(IPuppet): return self._mock_puppet.run_pba(name, options) def ping(self, host: str, timeout: float = 1) -> PingScanData: - return self._mock_puppet.ping(host, timeout) + return network.ping(host, timeout) def scan_tcp_port(self, host: str, port: int, timeout: float = 3) -> PortScanData: return self._mock_puppet.scan_tcp_port(host, port, timeout)