p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'feature/support-subnet-in-config' into feature/detect-cross-segment-traffic

This commit is contained in:
Itay Mizeretz 2018-02-26 17:35:48 +02:00
parent daef61e492 816be5191b
commit c369b8b19b
1 changed files with 4 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
infection_monkey/network/range.py
View File

@ -29,6 +29,10 @@ class NetworkRange(object):
for x in base_range: for x in base_range:
yield VictimHost(self._number_to_ip(self._base_address + x)) yield VictimHost(self._number_to_ip(self._base_address + x))
@abstractmethod
def is_in_range(self, ip_address):
raise NotImplementedError()
@abstractmethod @abstractmethod
def _get_range(self): def _get_range(self):
raise NotImplementedError() raise NotImplementedError()
@ -53,35 +57,6 @@ class NetworkRange(object):
return socket.inet_ntoa(struct.pack(">L", num)) return socket.inet_ntoa(struct.pack(">L", num))
class ClassCRange(NetworkRange):
def __init__(self, base_address, shuffle=True):
base_address = struct.unpack(">L", socket.inet_aton(base_address))[0] & 0xFFFFFF00
super(ClassCRange, self).__init__(base_address, shuffle=shuffle)
def __repr__(self):
return "<ClassCRange %s-%s>" % (self._number_to_ip(self._base_address + 1),
self._number_to_ip(self._base_address + 254))
def _get_range(self):
return range(1, 254)
class RelativeRange(NetworkRange):
def __init__(self, base_address, shuffle=True):
base_address = struct.unpack(">L", socket.inet_aton(base_address))[0]
super(RelativeRange, self).__init__(base_address, shuffle=shuffle)
self._size = 1
def __repr__(self):
return "<RelativeRange %s-%s>" % (self._number_to_ip(self._base_address - self._size),
self._number_to_ip(self._base_address + self._size))
def _get_range(self):
lower_end = -(self._size / 2)
higher_end = lower_end + self._size
return range(lower_end, higher_end + 1)
class CidrRange(NetworkRange): class CidrRange(NetworkRange):
def __init__(self, cidr_range, shuffle=True): def __init__(self, cidr_range, shuffle=True):
base_address = 0 base_address = 0