forked from p15670423/monkey
parent
9f2fe5e513
commit
019f2c1403
|
@ -2,6 +2,7 @@ import logging
|
||||||
import threading
|
import threading
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
|
from infection_monkey import network
|
||||||
from infection_monkey.i_puppet import (
|
from infection_monkey.i_puppet import (
|
||||||
ExploiterResultData,
|
ExploiterResultData,
|
||||||
FingerprintData,
|
FingerprintData,
|
||||||
|
@ -33,7 +34,7 @@ class Puppet(IPuppet):
|
||||||
return self._mock_puppet.run_pba(name, options)
|
return self._mock_puppet.run_pba(name, options)
|
||||||
|
|
||||||
def ping(self, host: str, timeout: float = 1) -> PingScanData:
|
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:
|
def scan_tcp_port(self, host: str, port: int, timeout: float = 3) -> PortScanData:
|
||||||
return self._mock_puppet.scan_tcp_port(host, port, timeout)
|
return self._mock_puppet.scan_tcp_port(host, port, timeout)
|
||||||
|
|
Loading…
Reference in New Issue