27 lines
863 B
Python
27 lines
863 B
Python
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import subprocess
|
||
|
|
from network import HostScanner
|
||
|
|
from model.host import VictimHost
|
||
|
|
|
||
|
|
__author__ = 'itamar'
|
||
|
|
|
||
|
|
PING_COUNT_FLAG = "-n" if "win32" == sys.platform else "-c"
|
||
|
|
PING_TIMEOUT_FLAG = "-w" if "win32" == sys.platform else "-W"
|
||
|
|
|
||
|
|
class PingScanner(HostScanner):
|
||
|
|
def __init__(self):
|
||
|
|
self._config = __import__('config').WormConfiguration
|
||
|
|
self._devnull = open(os.devnull, "w")
|
||
|
|
|
||
|
|
def is_host_alive(self, host):
|
||
|
|
assert isinstance(host, VictimHost)
|
||
|
|
|
||
|
|
return 0 == subprocess.call(["ping",
|
||
|
|
PING_COUNT_FLAG, "1",
|
||
|
|
PING_TIMEOUT_FLAG, str(self._config.ping_scan_timeout),
|
||
|
|
host.ip_addr],
|
||
|
|
stdout=self._devnull,
|
||
|
|
stderr=self._devnull)
|