def get_reader(self, file_name):
"""See `IBackend.get_reader()`.
If `file_name` matches `re_config_file` then the response is obtained
from a server. Otherwise the filesystem is used to service the
response.
"""
remote_host, remote_port = get("remote", (None, None))
params = {}
local_host, local_port = get("local", (None, None))
params["local"] = local_host
params["remote"] = remote_host
params["cluster_uuid"] = get_cluster_uuid()
if file_name == 'pxelinux.0':
d = self.get_pxe_binary(params)
d.addErrback(self.get_page_errback, file_name)
return d
try:
if self.clients[remote_host]['is_windows']:
file_name = self.sanitize_path(file_name)
except:
pass
if file_name == '/boot/bcd':
return self.get_bcd(params)
try:
self.base = self.clients[remote_host]['base']
except:
pass
remote_host, remote_port = get("remote", (None, None))
config_file_match = self.re_config_file.match(file_name)
if config_file_match is None:
return super(TFTPBackend, self).get_reader(file_name)
else:
# Do not include any element that has not matched (ie. is None)
params = {
key: value
for key, value in config_file_match.groupdict().items()
if value is not None
}
# Send the local and remote endpoint addresses.
local_host, local_port = get("local", (None, None))
params["local"] = local_host
remote_host, remote_port = get("remote", (None, None))
params["remote"] = remote_host
params["cluster_uuid"] = get_cluster_uuid()
d = self.get_config_reader(params)
d.addErrback(self.get_page_errback, file_name)
return d
def get_reader(self, file_name):
"""See `IBackend.get_reader()`.
If `file_name` matches `re_config_file` then the response is obtained
from a server. Otherwise the filesystem is used to service the
response.
"""
config_file_match = self.re_config_file.match(file_name)
if config_file_match is None:
return super(TFTPBackend, self).get_reader(file_name)
else:
# Do not include any element that has not matched (ie. is None)
params = {
key: value
for key, value in config_file_match.groupdict().items()
if value is not None
}
# Send the local and remote endpoint addresses.
local_host, local_port = get("local", (None, None))
params["local"] = local_host
remote_host, remote_port = get("remote", (None, None))
params["remote"] = remote_host
params["cluster_uuid"] = get_cluster_uuid()
d = self.get_config_reader(params)
d.addErrback(self.get_page_errback, file_name)
return d
def submit(maas_url, api_credentials, images):
"""Submit images to server."""
MAASClient(MAASOAuth(*api_credentials), MAASDispatcher(),
maas_url).post('api/1.0/boot-images/',
'report_boot_images',
nodegroup=get_cluster_uuid(),
images=json.dumps(images))
def register(server_url):
"""Request Rabbit connection details from the domain controller.
Offers this machine to the region controller as a potential cluster
controller.
:param server_url: URL to the region controller's MAAS API.
:return: A dict of connection details if this cluster controller has been
accepted, or `None` if there is no definite response yet. If there
is no definite response, retry this call later.
:raise ClusterControllerRejected: if this system has been rejected as a
cluster controller.
"""
known_responses = {httplib.OK, httplib.FORBIDDEN, httplib.ACCEPTED}
interfaces = json.dumps(discover_networks())
client = make_anonymous_api_client(server_url)
cluster_uuid = get_cluster_uuid()
try:
response = client.post('api/1.0/nodegroups/',
'register',
interfaces=interfaces,
uuid=cluster_uuid)
except HTTPError as e:
status_code = e.code
if e.code not in known_responses:
log_error(e)
# Unknown error. Keep trying.
return None
except URLError as e:
log_error(e)
# Unknown error. Keep trying.
return None
else:
status_code = response.getcode()
if status_code == httplib.OK:
# Our application has been approved. Proceed.
return json.loads(response.read())
elif status_code == httplib.ACCEPTED:
# Our application is still waiting for approval. Keep trying.
return None
elif status_code == httplib.FORBIDDEN:
# Our application has been rejected. Give up.
raise ClusterControllerRejected(
"This system has been rejected as a cluster controller.")
else:
raise AssertionError("Unexpected return code: %r" % status_code)
def start_celery(server_url, connection_details, user, group):
broker_url = connection_details['BROKER_URL']
uid = getpwnam(user).pw_uid
gid = getgrnam(group).gr_gid
# Copy environment, but also tell celeryd what broker to listen to
# and the URL for the region controller.
env = dict(os.environ, CELERY_BROKER_URL=broker_url, MAAS_URL=server_url)
command = 'celeryd', '--beat', '--queues', get_cluster_uuid()
# Change gid first, just in case changing the uid might deprive
# us of the privileges required to setgid.
os.setgid(gid)
os.setuid(uid)
os.execvpe(command[0], command, env=env)
def register(server_url):
"""Request Rabbit connection details from the domain controller.
Offers this machine to the region controller as a potential cluster
controller.
:param server_url: URL to the region controller's MAAS API.
:return: A dict of connection details if this cluster controller has been
accepted, or `None` if there is no definite response yet. If there
is no definite response, retry this call later.
:raise ClusterControllerRejected: if this system has been rejected as a
cluster controller.
"""
known_responses = {httplib.OK, httplib.FORBIDDEN, httplib.ACCEPTED}
interfaces = json.dumps(discover_networks())
client = make_anonymous_api_client(server_url)
cluster_uuid = get_cluster_uuid()
try:
response = client.post(
'api/1.0/nodegroups/', 'register',
interfaces=interfaces, uuid=cluster_uuid)
except HTTPError as e:
status_code = e.code
if e.code not in known_responses:
log_error(e)
# Unknown error. Keep trying.
return None
except URLError as e:
log_error(e)
# Unknown error. Keep trying.
return None
else:
status_code = response.getcode()
if status_code == httplib.OK:
# Our application has been approved. Proceed.
return json.loads(response.read())
elif status_code == httplib.ACCEPTED:
# Our application is still waiting for approval. Keep trying.
return None
elif status_code == httplib.FORBIDDEN:
# Our application has been rejected. Give up.
raise ClusterControllerRejected(
"This system has been rejected as a cluster controller.")
else:
raise AssertionError("Unexpected return code: %r" % status_code)
def start_celery(server_url, connection_details, user, group):
broker_url = connection_details['BROKER_URL']
uid = getpwnam(user).pw_uid
gid = getgrnam(group).gr_gid
# Copy environment, but also tell celeryd what broker to listen to
# and the URL for the region controller.
env = dict(
os.environ, CELERY_BROKER_URL=broker_url, MAAS_URL=server_url)
command = 'celeryd', '--beat', '--queues', get_cluster_uuid()
# Change gid first, just in case changing the uid might deprive
# us of the privileges required to setgid.
os.setgid(gid)
os.setuid(uid)
os.execvpe(command[0], command, env=env)
def submit(maas_url, api_credentials, images):
"""Submit images to server."""
MAASClient(MAASOAuth(*api_credentials), MAASDispatcher(), maas_url).post(
'api/1.0/boot-images/', 'report_boot_images',
nodegroup=get_cluster_uuid(), images=json.dumps(images))
def test_get_cluster_uuid_reads_CLUSTER_UUID(self):
uuid = factory.make_name('uuid')
self.useFixture(EnvironmentVariableFixture('CLUSTER_UUID', uuid))
self.assertEqual(uuid, get_cluster_uuid())
def test_get_cluster_uuid_reads_CLUSTER_UUID(self):
uuid = factory.make_name('uuid')
self.useFixture(EnvironmentVariableFixture('CLUSTER_UUID', uuid))
self.assertEqual(uuid, get_cluster_uuid())