def insert(self, item):
if item.randomized and len(self) == 0:
self.isRandomized = True
if not self.isAny:
BaseMap.insert(self, item)
if self.checkForPortscans and not item.portNumber == -1:
self.elements.insert(item)
class PortsMap(BaseMap):
"""A map of ports."""
def __init__(self, npa, checkForPortscans=False):
BaseMap.__init__(self)
self.classinfo = PortsMap
self.NUM_PORTS_ANY = npa
self.separator = ", "
self.isRange = False
self.isRandomized = False
self.checkForPortscans = checkForPortscans
if self.checkForPortscans:
self.elements = BaseMap()
self.elements.separator = ", "
def insert(self, item):
if item.randomized and len(self) == 0:
self.isRandomized = True
if not self.isAny:
BaseMap.insert(self, item)
if self.checkForPortscans and not item.portNumber == -1:
self.elements.insert(item)
def extend(self, other):
if (len(self) + len(other)) < self.NUM_PORTS_ANY and not self.isAny and not other.isAny:
BaseMap.extend(self, other)
else:
self.clear()
self.isAny = True
self.insert(Port(-1, False))
if self.checkForPortscans:
self.elements.extend(other.elements.values())
def doChecks(self, numChecks, distRange, numAny):
l = len(self)
if l >= numChecks:
if l > numAny:
self.clear()
self.insert(Port(-1))
self.isAny = True
if not self.isAny and not self.isRange:
self.checkForRange(distRange)
def checkForRange(self, border):
portMap = self.values()
portMap.sort()
if portMap[len(portMap) - 1] - portMap[0] <= border:
self.isRange = True
self.clear()
self.insert(portMap[0])
self.insert(portMap[len(portMap) - 1])
self.separator = ":"
def __str__(self):
if self.isAny:
return ""
out = BaseMap.__str__(self)
if len(self) > 1 and not self.isRange:
out = "{ " + out + " }"
return out
def extend(self, other):
if (len(self) + len(other)) < self.NUM_PORTS_ANY and not self.isAny and not other.isAny:
BaseMap.extend(self, other)
else:
self.clear()
self.isAny = True
self.insert(Port(-1, False))
if self.checkForPortscans:
self.elements.extend(other.elements.values())
def extend(self, other):
if not other.isAny:
BaseMap.extend(self, other)
else:
self.clear()
self.insert(IPv4Address("0.0.0.0"))
self.isAny = True
if self.checkForPortscans:
self.elements.extend(other.elements.values())
def __init__(self, npa, checkForPortscans=False):
BaseMap.__init__(self)
self.classinfo = PortsMap
self.NUM_PORTS_ANY = npa
self.separator = ", "
self.isRange = False
self.isRandomized = False
self.checkForPortscans = checkForPortscans
if self.checkForPortscans:
self.elements = BaseMap()
self.elements.separator = ", "
def __init__(self, checkForPortscans=False):
BaseMap.__init__(self)
self.classinfo = IPsMap
self.separator = ", "
self.isNetworks = False
self.checkForPortscans = checkForPortscans
if self.checkForPortscans:
self.elements = BaseMap()
self.elements.separator = ", "
# List for problems, which need a sorted list
# must be filled with current values() before used
self.ipList = []
def __str__(self):
if self.isAny:
return ""
out = BaseMap.__str__(self)
if len(self) > 1 and not self.isRange:
out = "{ " + out + " }"
return out
def __str__(self):
self.ipList = self.values()
for i in range(len(self.ipList)):
if not isinstance(self.ipList[i], IPv4Network):
self.ipList[i] = IPv4Network(self.ipList[i], 32)
self._mapToDict()
out = "any" if self.isAny else BaseMap.__str__(self)
out = out.replace("/32", "")
if len(self) > 1 and not self.isAny:
out = "{ " + out + " }"
self.ipList = self.values()
for i in range(len(self.ipList)):
if self.ipList[i].prefixlen == 32:
self.ipList[i] = IPv4Address(self.ipList[i].ip)
self._mapToDict()
return out
class IPsMap(BaseMap):
"""A map of IPs."""
def __init__(self, checkForPortscans=False):
BaseMap.__init__(self)
self.classinfo = IPsMap
self.separator = ", "
self.isNetworks = False
self.checkForPortscans = checkForPortscans
if self.checkForPortscans:
self.elements = BaseMap()
self.elements.separator = ", "
# List for problems, which need a sorted list
# must be filled with current values() before used
self.ipList = []
def insert(self, item):
if not self.isAny:
BaseMap.insert(self, item)
if self.checkForPortscans and isinstance(item, IPv4Address):
self.elements.insert(item)
def extend(self, other):
if not other.isAny:
BaseMap.extend(self, other)
else:
self.clear()
self.insert(IPv4Address("0.0.0.0"))
self.isAny = True
if self.checkForPortscans:
self.elements.extend(other.elements.values())
def ipInBoth(self, otherMap):
for value1 in self:
if isinstance(value1, IPv4Address):
if value1 in otherMap:
return True
return False
def doChecks(self, numChecks, numExpand, numAny, slashSize, minIpsForSlash):
self.ipList = self.values()
self._sort_ipList()
self.makeSlashNetwork(slashSize, minIpsForSlash)
if len(self) >= numChecks:
if len(self.ipList) > numAny:
self.ipList = [IPv4Address("0.0.0.0")]
self.isAny = True
if self.isNetworks or self.isAny:
self._mapToDict()
def checkForJoinedNetwork(self, otherIPs, slashSize):
"""Checks if the ips from otherIPs would be
in size slashSize networks made from the ips in self.
If the ips from otherIPs fit into these networks,
the ips in self are made into corresponding
slashSize networks."""
self.ipList = self.values()
length = len(self.ipList)
if length > len(otherIPs):
length = len(otherIPs)
self.ipList.extend(otherIPs.values())
self.ipList = list(set(self.ipList))
self._sort_ipList()
self.makeSlashNetwork(slashSize, 2, length, True)
if self.isNetworks:
self._mapToDict()
if self.checkForPortscans:
self.elements.extend(otherIPs.elements.values())
return self.isNetworks
def checkForNetworks(self, border, numExpand):
"""Checks if by summarizing the address range of ips
the number of networks is <= number of ips.
If that is the case, ips are replaced with the networks
from the summarized address range."""
if len(self.ipList) >= border:
network = summarize_address_range(self.ipList[0], self.ipList[len(self.ipList) - 1])
network = collapse_address_list(network)
if len(network) <= (len(self) / 2):
self.isNetworks = True
self.ipList = network
def makeSlashNetwork(self, slashSize, minIPsPerNet, maxFoundNetw=1000, joinedNetworksCheck=False):
"""If minIPsPerNet can be found for a slashSize networks,
instead of the ips, the corresponding network is saved.
MaxFoundNetworks and onlyOne are only relevant for check from
checkForJoinedNetwork. If there are more networks, than
maxFoundNetwork, the ips which are checked are not in a
joined, but in separate networks."""
foundNetworks = []
ips = self.ipList
while len(ips) > 0:
if joinedNetworksCheck and len(foundNetworks) > maxFoundNetw:
self.isNetworks = False
return
startIp = ips[0]
if not isinstance(startIp, IPv4Network):
network = IPv4Network("%s/%d" % (startIp, slashSize))
network = network.masked()
elif startIp.prefixlen == 32:
network = IPv4Network("%s/%d" % (startIp.ip, slashSize))
network = network.masked()
else:
network = startIp
# If it is a network already, minIPsPerNet is not relevant
minIPsPerNet = 2
i = 0
matchingIPsToCurrentNetwork = True
while matchingIPsToCurrentNetwork:
if not isinstance(ips[i], IPv4Network):
tmpNetwork = IPv4Network("%s/%d" % (ips[i], slashSize))
elif ips[i].prefixlen == 32:
tmpNetwork = IPv4Network("%s/%d" % (ips[i].ip, slashSize))
tmpNetwork = network.masked()
else:
tmpNetwork = ips[i]
if len(ips) == 1:
i = i + 1
if tmpNetwork == network and len(ips) > 1:
i = i + 1
if i >= len(ips):
self._writeIPs(ips, foundNetworks, network, i, minIPsPerNet)
matchingIPsToCurrentNetwork = False
else:
self._writeIPs(ips, foundNetworks, network, i, minIPsPerNet)
matchingIPsToCurrentNetwork = False
if joinedNetworksCheck and len(foundNetworks) > maxFoundNetw:
self.isNetworks = False
else:
self.ipList = foundNetworks
self.isNetworks = True
def _sort_ipList(self):
for i in range(len(self.ipList)):
if not isinstance(self.ipList[i], IPv4Network):
self.ipList[i] = IPv4Network(self.ipList[i], 32)
self.ipList.sort()
for i in range(len(self.ipList)):
if self.ipList[i].prefixlen == 32:
self.ipList[i] = IPv4Address(self.ipList[i].ip)
def _expandedNetworkSearch(self, border):
"""This method tries to find a network by
expanding the range
WARNING: This really takes some time,
be careful what you choose for
border!"""
startIp = self.ipList[0]
endIp = self.ipList[len(self.ipList) - 1] + 1
for i in range(border):
for j in range(border):
network = summarize_address_range(startIp, endIp)
if len(network) == 1:
self.ipList = network
self.isNetworks = True
endIp = endIp + 1
endIp = self.ipList[len(self.ipList) - 1]
startIp = startIp - 1
def _writeIPs(self, ips, foundNetworks, network, i, minIPsPerNet):
if i >= minIPsPerNet:
foundNetworks.append(network)
else:
for j in range(i):
foundNetworks.append(ips[j])
for j in range(i):
ips.remove(ips[0])
def _mapToDict(self):
isAny = self.isAny
self.clear()
for item in self.ipList:
self.insert(item)
del self.ipList
self.isAny = isAny
def __str__(self):
self.ipList = self.values()
for i in range(len(self.ipList)):
if not isinstance(self.ipList[i], IPv4Network):
self.ipList[i] = IPv4Network(self.ipList[i], 32)
self._mapToDict()
out = "any" if self.isAny else BaseMap.__str__(self)
out = out.replace("/32", "")
if len(self) > 1 and not self.isAny:
out = "{ " + out + " }"
self.ipList = self.values()
for i in range(len(self.ipList)):
if self.ipList[i].prefixlen == 32:
self.ipList[i] = IPv4Address(self.ipList[i].ip)
self._mapToDict()
return out
def insert(self, item):
if not self.isAny:
BaseMap.insert(self, item)
if self.checkForPortscans and isinstance(item, IPv4Address):
self.elements.insert(item)
def __init__(self):
BaseMap.__init__(self)
self.classinfo = RuleSet
