143 lines
4.5 KiB
Python
143 lines
4.5 KiB
Python
# Copyright 2019 Red Hat, Inc.
|
|
#
|
|
# This work is licensed under the GNU GPLv2 or later.
|
|
# See the COPYING file in the top-level directory.
|
|
|
|
import os
|
|
import random
|
|
import re
|
|
import string
|
|
import tempfile
|
|
|
|
from ..logger import log
|
|
|
|
|
|
class CloudInitData():
|
|
disable = None
|
|
root_password_generate = None
|
|
root_password_file = None
|
|
generated_root_password = None
|
|
root_ssh_key = None
|
|
clouduser_ssh_key = None
|
|
user_data = None
|
|
meta_data = None
|
|
network_config = None
|
|
|
|
def _generate_password(self):
|
|
if not self.generated_root_password:
|
|
self.generated_root_password = ""
|
|
for dummy in range(16):
|
|
self.generated_root_password += random.choice(
|
|
string.ascii_letters + string.digits)
|
|
return self.generated_root_password
|
|
|
|
def _get_password(self, pwdfile):
|
|
with open(pwdfile, "r") as fobj:
|
|
return fobj.readline().rstrip("\n\r")
|
|
|
|
def get_password_if_generated(self):
|
|
if self.root_password_generate:
|
|
return self._generate_password()
|
|
|
|
def get_root_password(self):
|
|
if self.root_password_file:
|
|
return self._get_password(self.root_password_file)
|
|
return self.get_password_if_generated()
|
|
|
|
def get_root_ssh_key(self):
|
|
if self.root_ssh_key:
|
|
return self._get_password(self.root_ssh_key)
|
|
|
|
def get_clouduser_ssh_key(self):
|
|
if self.clouduser_ssh_key:
|
|
return self._get_password(self.clouduser_ssh_key)
|
|
|
|
|
|
def _create_metadata_content(cloudinit_data):
|
|
content = ""
|
|
if cloudinit_data.meta_data:
|
|
log.debug("Using meta-data content from path=%s",
|
|
cloudinit_data.meta_data)
|
|
content = open(cloudinit_data.meta_data).read()
|
|
return content
|
|
|
|
|
|
def _create_userdata_content(cloudinit_data):
|
|
if cloudinit_data.user_data:
|
|
log.debug("Using user-data content from path=%s",
|
|
cloudinit_data.user_data)
|
|
return open(cloudinit_data.user_data).read()
|
|
|
|
content = "#cloud-config\n"
|
|
|
|
if cloudinit_data.root_password_generate or cloudinit_data.root_password_file:
|
|
rootpass = cloudinit_data.get_root_password()
|
|
content += "chpasswd:\n"
|
|
content += " list: |\n"
|
|
content += " root:%s\n" % rootpass
|
|
|
|
if cloudinit_data.root_password_generate:
|
|
content += " expire: True\n"
|
|
elif cloudinit_data.root_password_file:
|
|
content += " expire: False\n"
|
|
|
|
if cloudinit_data.root_ssh_key or cloudinit_data.clouduser_ssh_key:
|
|
content += "users:\n"
|
|
rootkey = cloudinit_data.get_root_ssh_key()
|
|
userkey = cloudinit_data.get_clouduser_ssh_key()
|
|
|
|
if rootkey:
|
|
content += " - name: root\n"
|
|
content += " ssh-authorized-keys:\n"
|
|
content += " - %s\n" % rootkey
|
|
if userkey:
|
|
content += " - ssh-authorized-keys: %s\n" % userkey
|
|
|
|
if cloudinit_data.disable:
|
|
content += "runcmd:\n"
|
|
content += ('- echo "Disabled by virt-install" > '
|
|
"/etc/cloud/cloud-init.disabled\n")
|
|
|
|
clean_content = re.sub(r"root:(.*)", 'root:[SCRUBBLED]', content)
|
|
if "VIRTINST_TEST_SUITE_PRINT_CLOUDINIT" in os.environ:
|
|
print(clean_content)
|
|
|
|
log.debug("Generated cloud-init userdata: \n%s", clean_content)
|
|
return content
|
|
|
|
|
|
def _create_network_config_content(cloudinit_data):
|
|
content = ""
|
|
if cloudinit_data.network_config:
|
|
log.debug("Using network-config content from path=%s",
|
|
cloudinit_data.network_config)
|
|
content = open(cloudinit_data.network_config).read()
|
|
return content
|
|
|
|
|
|
def create_files(scratchdir, cloudinit_data):
|
|
metadata = _create_metadata_content(cloudinit_data)
|
|
userdata = _create_userdata_content(cloudinit_data)
|
|
|
|
data = [(metadata, "meta-data"), (userdata, "user-data")]
|
|
network_config = _create_network_config_content(cloudinit_data)
|
|
if network_config:
|
|
data.append((network_config, 'network-config'))
|
|
|
|
filepairs = []
|
|
try:
|
|
for content, destfile in data:
|
|
fileobj = tempfile.NamedTemporaryFile(
|
|
prefix="virtinst-", suffix=("-%s" % destfile),
|
|
dir=scratchdir, delete=False)
|
|
filename = fileobj.name
|
|
filepairs.append((filename, destfile))
|
|
|
|
with open(filename, "w+") as f:
|
|
f.write(content)
|
|
except Exception: # pragma: no cover
|
|
for filepair in filepairs:
|
|
os.unlink(filepair[0])
|
|
|
|
return filepairs
|