738a3472f1
Modified cc_mounts to identify whether ephermalX is partitioned. Changed datasources for Azure and SmartOS to use 'ephemeralX.Y' format. Added disk remove functionally
234 lines
7.7 KiB
Python
234 lines
7.7 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2009-2010 Canonical Ltd.
|
|
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Author: Scott Moser <scott.moser@canonical.com>
|
|
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 3, as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from string import whitespace # pylint: disable=W0402
|
|
|
|
import os.path
|
|
import re
|
|
|
|
from cloudinit import type_utils
|
|
from cloudinit import util
|
|
|
|
# Shortname matches 'sda', 'sda1', 'xvda', 'hda', 'sdb', xvdb, vda, vdd1, sr0
|
|
SHORTNAME_FILTER = r"^([x]{0,1}[shv]d[a-z][0-9]*|sr[0-9]+)$"
|
|
SHORTNAME = re.compile(SHORTNAME_FILTER)
|
|
WS = re.compile("[%s]+" % (whitespace))
|
|
FSTAB_PATH = "/etc/fstab"
|
|
|
|
|
|
def is_mdname(name):
|
|
# return true if this is a metadata service name
|
|
if name in ["ami", "root", "swap"]:
|
|
return True
|
|
# names 'ephemeral0' or 'ephemeral1'
|
|
# 'ebs[0-9]' appears when '--block-device-mapping sdf=snap-d4d90bbc'
|
|
for enumname in ("ephemeral", "ebs"):
|
|
if name.startswith(enumname) and name.find(":") == -1:
|
|
return True
|
|
return False
|
|
|
|
|
|
def handle(_name, cfg, cloud, log, _args):
|
|
# fs_spec, fs_file, fs_vfstype, fs_mntops, fs-freq, fs_passno
|
|
defvals = [None, None, "auto", "defaults,nobootwait", "0", "2"]
|
|
defvals = cfg.get("mount_default_fields", defvals)
|
|
|
|
# these are our default set of mounts
|
|
defmnts = [["ephemeral0", "/mnt", "auto", defvals[3], "0", "2"],
|
|
["swap", "none", "swap", "sw", "0", "0"]]
|
|
|
|
cfgmnt = []
|
|
if "mounts" in cfg:
|
|
cfgmnt = cfg["mounts"]
|
|
|
|
for i in range(len(cfgmnt)):
|
|
# skip something that wasn't a list
|
|
if not isinstance(cfgmnt[i], list):
|
|
log.warn("Mount option %s not a list, got a %s instead",
|
|
(i + 1), type_utils.obj_name(cfgmnt[i]))
|
|
continue
|
|
|
|
startname = str(cfgmnt[i][0])
|
|
log.debug("Attempting to determine the real name of %s", startname)
|
|
|
|
# workaround, allow user to specify 'ephemeral'
|
|
# rather than more ec2 correct 'ephemeral0'
|
|
if startname == "ephemeral":
|
|
cfgmnt[i][0] = "ephemeral0"
|
|
log.debug(("Adjusted mount option %s "
|
|
"name from ephemeral to ephemeral0"), (i + 1))
|
|
|
|
if is_mdname(startname):
|
|
candidate_name = cloud.device_name_to_device(startname)
|
|
newname = disk_or_part(candidate_name)
|
|
|
|
if not newname:
|
|
log.debug("Ignoring nonexistant named mount %s", startname)
|
|
cfgmnt[i][1] = None
|
|
else:
|
|
renamed = newname
|
|
if not newname.startswith("/"):
|
|
renamed = "/dev/%s" % newname
|
|
cfgmnt[i][0] = renamed
|
|
log.debug("Mapped metadata name %s to %s", startname, renamed)
|
|
else:
|
|
if SHORTNAME.match(startname):
|
|
renamed = "/dev/%s" % startname
|
|
log.debug("Mapped shortname name %s to %s", startname, renamed)
|
|
cfgmnt[i][0] = renamed
|
|
|
|
# in case the user did not quote a field (likely fs-freq, fs_passno)
|
|
# but do not convert None to 'None' (LP: #898365)
|
|
for j in range(len(cfgmnt[i])):
|
|
if cfgmnt[i][j] is None:
|
|
continue
|
|
else:
|
|
cfgmnt[i][j] = str(cfgmnt[i][j])
|
|
|
|
for i in range(len(cfgmnt)):
|
|
# fill in values with defaults from defvals above
|
|
for j in range(len(defvals)):
|
|
if len(cfgmnt[i]) <= j:
|
|
cfgmnt[i].append(defvals[j])
|
|
elif cfgmnt[i][j] is None:
|
|
cfgmnt[i][j] = defvals[j]
|
|
|
|
# if the second entry in the list is 'None' this
|
|
# clears all previous entries of that same 'fs_spec'
|
|
# (fs_spec is the first field in /etc/fstab, ie, that device)
|
|
if cfgmnt[i][1] is None:
|
|
for j in range(i):
|
|
if cfgmnt[j][0] == cfgmnt[i][0]:
|
|
cfgmnt[j][1] = None
|
|
|
|
# for each of the "default" mounts, add them only if no other
|
|
# entry has the same device name
|
|
for defmnt in defmnts:
|
|
startname = defmnt[0]
|
|
candidate_name = cloud.device_name_to_device(startname)
|
|
devname = disk_or_part(candidate_name)
|
|
if devname is None:
|
|
log.debug("Ignoring nonexistant named default mount %s", startname)
|
|
continue
|
|
if devname.startswith("/"):
|
|
defmnt[0] = devname
|
|
else:
|
|
defmnt[0] = "/dev/%s" % devname
|
|
|
|
log.debug("Mapped default device %s to %s", startname, defmnt[0])
|
|
|
|
cfgmnt_has = False
|
|
for cfgm in cfgmnt:
|
|
if cfgm[0] == defmnt[0]:
|
|
cfgmnt_has = True
|
|
break
|
|
|
|
if cfgmnt_has:
|
|
log.debug(("Not including %s, already"
|
|
" previously included"), startname)
|
|
continue
|
|
cfgmnt.append(defmnt)
|
|
|
|
# now, each entry in the cfgmnt list has all fstab values
|
|
# if the second field is None (not the string, the value) we skip it
|
|
actlist = []
|
|
for x in cfgmnt:
|
|
if x[1] is None:
|
|
log.debug("Skipping non-existent device named %s", x[0])
|
|
else:
|
|
actlist.append(x)
|
|
|
|
if len(actlist) == 0:
|
|
log.debug("No modifications to fstab needed.")
|
|
return
|
|
|
|
comment = "comment=cloudconfig"
|
|
cc_lines = []
|
|
needswap = False
|
|
dirs = []
|
|
for line in actlist:
|
|
# write 'comment' in the fs_mntops, entry, claiming this
|
|
line[3] = "%s,%s" % (line[3], comment)
|
|
if line[2] == "swap":
|
|
needswap = True
|
|
if line[1].startswith("/"):
|
|
dirs.append(line[1])
|
|
cc_lines.append('\t'.join(line))
|
|
|
|
fstab_lines = []
|
|
for line in util.load_file(FSTAB_PATH).splitlines():
|
|
try:
|
|
toks = WS.split(line)
|
|
if toks[3].find(comment) != -1:
|
|
continue
|
|
except:
|
|
pass
|
|
fstab_lines.append(line)
|
|
|
|
fstab_lines.extend(cc_lines)
|
|
contents = "%s\n" % ('\n'.join(fstab_lines))
|
|
util.write_file(FSTAB_PATH, contents)
|
|
|
|
if needswap:
|
|
try:
|
|
util.subp(("swapon", "-a"))
|
|
except:
|
|
util.logexc(log, "Activating swap via 'swapon -a' failed")
|
|
|
|
for d in dirs:
|
|
try:
|
|
util.ensure_dir(d)
|
|
except:
|
|
util.logexc(log, "Failed to make '%s' config-mount", d)
|
|
|
|
try:
|
|
util.subp(("mount", "-a"))
|
|
except:
|
|
util.logexc(log, "Activating mounts via 'mount -a' failed")
|
|
|
|
|
|
def disk_or_part(device):
|
|
"""
|
|
Find where the file system is on the disk, either on
|
|
the disk itself or on the first partition. We don't go
|
|
any deeper than partition 1 though.
|
|
"""
|
|
|
|
if not device:
|
|
return None
|
|
|
|
short_name = device.split('/')[-1]
|
|
sys_path = "/sys/block/%s" % short_name
|
|
|
|
if not os.path.exists(sys_path):
|
|
LOG.warn("Device %s does not exist in sysfs" % device)
|
|
return None
|
|
|
|
sys_long_path = sys_path + "/" + short_name + "%s"
|
|
valid_mappings = [ sys_long_path % "1",
|
|
sys_long_path % "p1",
|
|
sys_path ]
|
|
|
|
for cdisk in valid_mappings:
|
|
if not os.path.exists(cdisk):
|
|
continue
|
|
return "/dev/%s" % cdisk.split('/')[-1]
|
|
return None
|