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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
# Copyright 2012 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Model definition for NodeGroup which models a collection of Nodes."""
from __future__ import (
absolute_import,
print_function,
unicode_literals,
)
__metaclass__ = type
__all__ = [
'NodeGroup',
]
from django.db.models import (
CharField,
ForeignKey,
IPAddressField,
Manager,
)
from maasserver import DefaultMeta
from maasserver.dhcp import is_dhcp_management_enabled
from maasserver.models.timestampedmodel import TimestampedModel
from maasserver.refresh_worker import refresh_worker
from piston.models import (
KEY_SIZE,
Token,
)
from provisioningserver.omshell import generate_omapi_key
from provisioningserver.tasks import (
add_new_dhcp_host_map,
write_dhcp_config,
)
class NodeGroupManager(Manager):
"""Manager for the NodeGroup class.
Don't import or instantiate this directly; access as `<Class>.objects` on
the model class it manages.
"""
def new(self, name, worker_ip, subnet_mask=None, broadcast_ip=None,
router_ip=None, ip_range_low=None, ip_range_high=None,
dhcp_key='', dhcp_interface=''):
"""Create a :class:`NodeGroup` with the given parameters.
This method will generate API credentials for the nodegroup's
worker to use.
"""
# Avoid circular imports.
from maasserver.models.user import create_auth_token
from maasserver.worker_user import get_worker_user
dhcp_values = [
dhcp_interface,
subnet_mask,
broadcast_ip,
router_ip,
ip_range_low,
ip_range_high,
]
assert all(dhcp_values) or not any(dhcp_values), (
"Provide all DHCP settings, or none at all.")
api_token = create_auth_token(get_worker_user())
nodegroup = NodeGroup(
name=name, worker_ip=worker_ip, subnet_mask=subnet_mask,
broadcast_ip=broadcast_ip, router_ip=router_ip,
ip_range_low=ip_range_low, ip_range_high=ip_range_high,
api_token=api_token, api_key=api_token.key, dhcp_key=dhcp_key,
dhcp_interface=dhcp_interface)
nodegroup.save()
return nodegroup
def ensure_master(self):
"""Obtain the master node group, creating it first if needed."""
# Avoid circular imports.
from maasserver.models import Node
try:
master = self.get(name='master')
except NodeGroup.DoesNotExist:
# The master did not exist yet; create it on demand.
master = self.new(
'master', '127.0.0.1', dhcp_key=generate_omapi_key())
# If any legacy nodes were still not associated with a node
# group, enroll them in the master node group.
Node.objects.filter(nodegroup=None).update(nodegroup=master)
return master
def get_by_natural_key(self, name):
"""For Django, a node group's name is a natural key."""
return self.get(name=name)
def refresh_workers(self):
"""Send refresh tasks to all node-group workers."""
for nodegroup in self.all():
refresh_worker(nodegroup)
class NodeGroup(TimestampedModel):
class Meta(DefaultMeta):
"""Needed for South to recognize this model."""
objects = NodeGroupManager()
# A node group's name is also used for the group's DNS zone.
name = CharField(
max_length=80, unique=True, editable=True, blank=False, null=False)
# Credentials for the worker to access the API with.
api_token = ForeignKey(Token, null=False, editable=False, unique=True)
api_key = CharField(
max_length=KEY_SIZE, null=False, blank=False, editable=False,
unique=True)
# Address of the worker.
worker_ip = IPAddressField(null=False, editable=True, unique=True)
# DHCP server settings.
dhcp_key = CharField(
blank=True, editable=False, max_length=255, default='')
# Network interface to serve DHCP on.
dhcp_interface = CharField(
blank=True, editable=False, max_length=255, default='')
subnet_mask = IPAddressField(
editable=True, unique=False, blank=True, null=True, default='')
broadcast_ip = IPAddressField(
editable=True, unique=False, blank=True, null=True, default='')
router_ip = IPAddressField(
editable=True, unique=False, blank=True, null=True, default='')
ip_range_low = IPAddressField(
editable=True, unique=True, blank=True, null=True, default='')
ip_range_high = IPAddressField(
editable=True, unique=True, blank=True, null=True, default='')
def __repr__(self):
return "<NodeGroup %r>" % self.name
def set_up_dhcp(self):
"""Write the DHCP configuration file and restart the DHCP server."""
# Circular imports.
from maasserver.dns import get_dns_server_address
write_dhcp_config.delay(
subnet=self.ip_range_low,
next_server=self.worker_ip,
omapi_key=self.dhcp_key, subnet_mask=self.subnet_mask,
broadcast_ip=self.broadcast_ip, router_ip=self.router_ip,
dns_servers=get_dns_server_address(),
ip_range_low=self.ip_range_low, ip_range_high=self.ip_range_high)
def is_dhcp_enabled(self):
"""Is the DHCP for this nodegroup enabled?"""
return is_dhcp_management_enabled() and all([
self.subnet_mask,
self.broadcast_ip,
self.ip_range_low,
self.ip_range_high
])
def add_dhcp_host_maps(self, new_leases):
if self.is_dhcp_enabled() and len(new_leases) > 0:
# XXX JeroenVermeulen 2012-08-21, bug=1039362: the DHCP
# server is currently always local to the worker system, so
# use 127.0.0.1 as the DHCP server address.
add_new_dhcp_host_map.delay(
new_leases, '127.0.0.1', self.dhcp_key)
|