82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
import socket
|
|
import random
|
|
import struct
|
|
from abc import ABCMeta, abstractmethod
|
|
from model.host import VictimHost
|
|
|
|
__author__ = 'itamar'
|
|
|
|
|
|
class NetworkRange(object):
|
|
__metaclass__ = ABCMeta
|
|
|
|
def __init__(self, base_address, shuffle=True):
|
|
self._base_address = base_address
|
|
self._shuffle = shuffle
|
|
self._config = __import__('config').WormConfiguration
|
|
|
|
@abstractmethod
|
|
def _get_range(self):
|
|
raise NotImplementedError()
|
|
|
|
def __iter__(self):
|
|
base_range = self._get_range()
|
|
if self._shuffle:
|
|
random.shuffle(base_range)
|
|
|
|
for x in base_range:
|
|
yield VictimHost(socket.inet_ntoa(struct.pack(">L", self._base_address + x)))
|
|
|
|
|
|
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>" % (socket.inet_ntoa(struct.pack(">L", self._base_address + 1)),
|
|
socket.inet_ntoa(struct.pack(">L", 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>" % (socket.inet_ntoa(struct.pack(">L", self._base_address - self._size)),
|
|
socket.inet_ntoa(struct.pack(">L", 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 FixedRange(NetworkRange):
|
|
def __init__(self, fixed_addresses=None, shuffle=True):
|
|
base_address = 0
|
|
super(FixedRange, self).__init__(base_address, shuffle=shuffle)
|
|
if not fixed_addresses:
|
|
self._fixed_addresses = self._config.range_fixed
|
|
else:
|
|
if type(fixed_addresses) is str:
|
|
self._fixed_addresses = [fixed_addresses]
|
|
else:
|
|
self._fixed_addresses = list(fixed_addresses)
|
|
|
|
def __repr__(self):
|
|
return "<FixedRange %s>" % (",".join(self._fixed_addresses))
|
|
|
|
def _get_range(self):
|
|
address_range = []
|
|
for address in self._fixed_addresses:
|
|
if not address: # Empty string
|
|
continue
|
|
address_range.append(struct.unpack(">L", socket.inet_aton(address))[0])
|
|
return address_range
|