1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
# $Id$
"""
Copyright (C) 2010, 2011 SPARTA, Inc. dba Cobham Analytic Solutions
Permission to use, copy, modify, and distribute this software for any
purpose with or without fee is hereby granted, provided that the above
copyright notice and this permission notice appear in all copies.
THE SOFTWARE IS PROVIDED "AS IS" AND SPARTA DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL SPARTA BE LIABLE FOR ANY SPECIAL, DIRECT,
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
"""
from rpki.gui.app import misc, models
from rpki import resource_set
class AllocationTree(object):
'''Virtual class representing a tree of unallocated resource ranges.
Keeps track of which subsets of a resource range have been
allocated.'''
def __init__(self, resource):
self.resource = resource
self.range = resource.as_resource_range()
self.need_calc = True
def calculate(self):
if self.need_calc:
self.children = []
self.alloc = self.__class__.set_type()
self.unalloc = self.__class__.set_type()
if self.is_allocated():
self.alloc.append(self.range)
else:
for child in self.resource.children.all():
c = self.__class__(child)
if c.unallocated():
self.children.append(c)
self.alloc = self.alloc.union(c.alloc)
total = self.__class__.set_type()
total.append(self.range)
self.unalloc = total.difference(self.alloc)
self.need_calc=False
def unallocated(self):
self.calculate()
return self.unalloc
def as_ul(self):
'''Returns a string of the tree as an unordered HTML list.'''
s = []
s.append('<a href="%s">%s</a>' % (self.resource.get_absolute_url(), self.resource))
# when the unallocated range is a subset of the current range,
# display the missing ranges
u = self.unallocated()
if len(u) != 1 or self.range != u[0]:
s.append(' (missing: ')
s.append(', '.join(str(x) for x in u))
s.append(')')
# quick access links
if self.resource.parent:
s.append(' | <a href="%s/delete">delete</a>' % (self.resource.get_absolute_url(),))
s.append(' | <a href="%s/allocate">give</a>' % (self.resource.get_absolute_url(),))
if self.range.min != self.range.max:
s.append(' | <a href="%s/split">split</a>' % (self.resource.get_absolute_url(),))
# add type-specific actions
a = self.supported_actions()
if a:
s.extend(a)
if self.children:
s.append('\n<ul>\n')
for c in self.children:
s.append('<li>' + c.as_ul())
s.append('\n</ul>')
return ''.join(s)
def supported_actions(self):
'''Virtual method allowing subclasses to add actions to the HTML list.'''
return None
@classmethod
def from_resource_range(cls, resource):
if isinstance(resource, resource_set.resource_range_as):
return AllocationTreeAS(resource)
if isinstance(resource, resoute_set.resource_range_ip):
return AllocationTreeIP(resource)
raise ValueError, 'Unsupported resource range type'
class AllocationTreeAS(AllocationTree):
set_type = resource_set.resource_set_as
def __init__(self, *args, **kwargs):
AllocationTree.__init__(self, *args, **kwargs)
self.conf = misc.top_parent(self.resource).from_cert.all()[0].parent.conf
def is_allocated(self):
'''Returns true if this AS has been allocated to a child or
used in a ROA request.'''
# FIXME: detect use in ROA requests
if self.resource.allocated:
return True
# for individual ASNs
if self.range.min == self.range.max:
# is this ASN used in any roa?
if self.conf.roas.filter(asn=self.range.min):
return True
return False
class AllocationTreeIP(AllocationTree):
'''virtual class representing a tree of IP address ranges.'''
@classmethod
def from_prefix(cls, prefix):
r = prefix.as_resource_range()
if isinstance(r, resource_set.resource_range_ipv4):
return AllocationTreeIPv4(prefix)
elif isinstance(r, resource_set.resource_range_ipv6):
return AllocationTreeIPv6(prefix)
raise ValueError, 'Unsupported IP range type'
def supported_actions(self):
'''add a link to issue a ROA for this IP range'''
if self.resource.is_prefix():
return [' | <a href="%s/roa">roa</a>' % self.resource.get_absolute_url()]
else:
return []
def is_allocated(self):
'''Return True if this IP range is allocated to a child or used
in a ROA request.'''
return self.resource.allocated or self.resource.roa_requests.count()
class AllocationTreeIPv4(AllocationTreeIP):
set_type = resource_set.resource_set_ipv4
class AllocationTreeIPv6(AllocationTreeIP):
set_type = resource_set.resource_set_ipv6
# vim:sw=4 ts=8 expandtab
|
