149 lines
4.8 KiB
Python
149 lines
4.8 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2009-2010 Canonical Ltd.
|
|
#
|
|
# Author: Scott Moser <scott.moser@canonical.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/>.
|
|
import cloudinit.util as util
|
|
import os
|
|
import re
|
|
import string
|
|
|
|
def handle(name,cfg,cloud,log,args):
|
|
# these are our default set of mounts
|
|
defmnts = [ [ "ephemeral0", "/mnt", "auto", "defaults", "0", "0" ],
|
|
[ "swap", "none", "swap", "sw", "0", "0" ] ]
|
|
|
|
# fs_spec, fs_file, fs_vfstype, fs_mntops, fs-freq, fs_passno
|
|
defvals = [ None, None, "auto", "defaults", "0", "0" ]
|
|
|
|
cfgmnt = [ ]
|
|
if cfg.has_key("mounts"):
|
|
cfgmnt = cfg["mounts"]
|
|
|
|
for i in range(len(cfgmnt)):
|
|
# skip something that wasn't a list
|
|
if not isinstance(cfgmnt[i],list): continue
|
|
|
|
# workaround, allow user to specify 'ephemeral'
|
|
# rather than more ec2 correct 'ephemeral0'
|
|
if cfgmnt[i][0] == "ephemeral":
|
|
cfgmnt[i][0] = "ephemeral0"
|
|
|
|
newname = cfgmnt[i][0]
|
|
if not newname.startswith("/"):
|
|
newname = cloud.device_name_to_device(cfgmnt[i][0])
|
|
if newname is not None:
|
|
cfgmnt[i][0] = newname
|
|
else:
|
|
# there is no good way of differenciating between
|
|
# a name that *couldn't* exist in the md service and
|
|
# one that merely didnt
|
|
# in order to allow user to specify 'sda3' rather
|
|
# than '/dev/sda3', go through some hoops
|
|
ok = False
|
|
for f in [ "/", "sd", "hd", "vd", "xvd" ]:
|
|
if cfgmnt[i][0].startswith(f):
|
|
ok = True
|
|
break
|
|
if not ok:
|
|
cfgmnt[i][1] = None
|
|
|
|
for i in range(len(cfgmnt)):
|
|
# fill in values with
|
|
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 not cfgmnt[i][0].startswith("/"):
|
|
cfgmnt[i][0]="/dev/%s" % cfgmnt[i][0]
|
|
|
|
# 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:
|
|
devname = cloud.device_name_to_device(defmnt[0])
|
|
if devname is None: continue
|
|
if devname.startswith("/"):
|
|
defmnt[0] = devname
|
|
else:
|
|
defmnt[0] = "/dev/%s" % devname
|
|
|
|
cfgmnt_has = False
|
|
for cfgm in cfgmnt:
|
|
if cfgm[0] == defmnt[0]:
|
|
cfgmnt_has = True
|
|
break
|
|
|
|
if cfgmnt_has: 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 = filter(lambda x: x[1] is not None, cfgmnt)
|
|
|
|
if len(actlist) == 0: 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,comment=cloudconfig" % line[3]
|
|
if line[2] == "swap": needswap = True
|
|
if line[1].startswith("/"): dirs.append(line[1])
|
|
cc_lines.append('\t'.join(line))
|
|
|
|
fstab_lines = [ ]
|
|
fstab=open("/etc/fstab","r+")
|
|
ws = re.compile("[%s]+" % string.whitespace)
|
|
for line in fstab.read().splitlines():
|
|
try:
|
|
toks = ws.split(line)
|
|
if toks[3].find(comment) != -1: continue
|
|
except:
|
|
pass
|
|
fstab_lines.append(line)
|
|
|
|
fstab_lines.extend(cc_lines)
|
|
|
|
fstab.seek(0)
|
|
fstab.write("%s\n" % '\n'.join(fstab_lines))
|
|
fstab.truncate()
|
|
fstab.close()
|
|
|
|
if needswap:
|
|
try: util.subp(("swapon", "-a"))
|
|
except: log.warn("Failed to enable swap")
|
|
|
|
for d in dirs:
|
|
if os.path.exists(d): continue
|
|
try: os.makedirs(d)
|
|
except: log.warn("Failed to make '%s' config-mount\n",d)
|
|
|
|
try: util.subp(("mount","-a"))
|
|
except: pass
|