diff options
Diffstat (limited to 'cloud-init.py')
| -rwxr-xr-x | cloud-init.py | 154 |
1 files changed, 49 insertions, 105 deletions
diff --git a/cloud-init.py b/cloud-init.py index 11bf89af..adac1874 100755 --- a/cloud-init.py +++ b/cloud-init.py @@ -22,12 +22,13 @@ import sys import cloudinit import cloudinit.util as util +import cloudinit.CloudConfig as CC import time import logging import errno -def warn(str): - sys.stderr.write(str) +def warn(wstr): + sys.stderr.write(wstr) def main(): cmds = ( "start", "start-local" ) @@ -39,7 +40,7 @@ def main(): sys.stderr.write("bad command %s. use one of %s\n" % (cmd, cmds)) sys.exit(1) - now = time.strftime("%a, %d %b %Y %H:%M:%S %z") + now = time.strftime("%a, %d %b %Y %H:%M:%S %z",time.gmtime()) try: uptimef=open("/proc/uptime") uptime=uptimef.read().split(" ")[0] @@ -48,18 +49,31 @@ def main(): warn("unable to open /proc/uptime\n") uptime = "na" - msg = "cloud-init %s running: %s. up %s seconds" % (cmd, now, uptime) - sys.stderr.write(msg + "\n") - sys.stderr.flush() - source_type = "all" if cmd == "start-local": source_type = "local" - cloudinit.logging_set_from_cfg_file() + try: + cfg = cloudinit.get_base_cfg() + (outfmt, errfmt) = CC.get_output_cfg(cfg,"init") + CC.redirect_output(outfmt, errfmt) + except Exception, e: + warn("Failed to get and set output config: %s\n" % e) + + msg = "cloud-init %s running: %s. up %s seconds" % (cmd, now, uptime) + sys.stderr.write(msg + "\n") + sys.stderr.flush() + + cloudinit.logging_set_from_cfg(cfg) log = logging.getLogger() log.info(msg) + try: + cloudinit.initfs() + except Exception, e: + warn("failed to initfs, likely bad things to come: %s\n" % str(e)) + + # cache is not instance specific, so it has to be purged # but we want 'start' to benefit from a cache if # a previous start-local populated one @@ -74,6 +88,9 @@ def main(): sys.stderr.write("no instance data found in %s\n" % cmd) sys.exit(1) + # set this as the current instance + cloud.set_cur_instance() + # store the metadata cloud.update_cache() @@ -89,108 +106,35 @@ def main(): warn("consuming user data failed!\n") raise - try: - if util.get_cfg_option_bool(cloud.cfg,"preserve_hostname",False): - log.debug("preserve_hostname is set. not managing hostname") - else: - hostname = cloud.get_hostname() - cloud.sem_and_run("set_hostname", "once-per-instance", - set_hostname, [ hostname, log ], False) - cloud.sem_and_run("update_hostname", "always", - update_hostname, [ hostname, log ], False) - except: - warn("failed to set hostname\n") - - #print "user data is:" + cloud.get_user_data() - - # set the defaults (like what ec2-set-defaults.py did) - try: - cloud.sem_and_run("set_defaults", "once-per-instance", - set_defaults,[ cloud ],False) - except: - warn("failed to set defaults\n") - # finish, send the cloud-config event cloud.initctl_emit() - sys.exit(0) - -def set_defaults(cloud): - apply_locale(cloud.get_locale()) - -def apply_locale(locale): - subprocess.Popen(['locale-gen', locale]).communicate() - subprocess.Popen(['update-locale', locale]).communicate() - - util.render_to_file('default-locale', '/etc/default/locale', \ - { 'locale' : locale }) - -# read hostname from a 'hostname' file -# allow for comments and stripping line endings. -# if file doesn't exist, or no contents, return default -def read_hostname(filename, default=None): - try: - fp = open(filename,"r") - lines = fp.readlines() - fp.close() - for line in lines: - hpos = line.find("#") - if hpos != -1: - line = line[0:hpos] - line = line.rstrip() - if line: - return line - except IOError, e: - if e.errno == errno.ENOENT: pass - return default - -def set_hostname(hostname, log): - try: - subprocess.Popen(['hostname', hostname]).communicate() - util.write_file("/etc/hostname","%s\n" % hostname, 0644) - log.debug("populated /etc/hostname with %s on first boot", hostname) - except: - log.error("failed to set_hostname") - -def update_hostname(hostname, log): - prev_file="%s/%s" % (cloudinit.datadir,"previous-hostname") - etc_file = "/etc/hostname" - - hostname_prev = None - hostname_in_etc = None + cfg_path = cloudinit.get_ipath_cur("cloud_config") + cc = CC.CloudConfig(cfg_path, cloud) + # if the output config changed, update output and err try: - hostname_prev = read_hostname(prev_file) - except: - log.warn("Failed to open %s" % prev_file) - - try: - hostname_in_etc = read_hostname(etc_file) - except: - log.warn("Failed to open %s" % etc_file) - - update_files = [] - if not hostname_prev or hostname_prev != hostname: - update_files.append(prev_file) - - if (not hostname_in_etc or - (hostname_in_etc == hostname_prev and hostname_in_etc != hostname)): - update_files.append(etc_file) - - try: - for fname in update_files: - util.write_file(fname,"%s\n" % hostname, 0644) - log.debug("wrote %s to %s" % (hostname,fname)) - except: - log.warn("failed to write hostname to %s" % fname) - - if hostname_in_etc and hostname_prev and hostname_in_etc != hostname_prev: - log.debug("%s differs from %s. assuming user maintained" % - (prev_file,etc_file)) - - if etc_file in update_files: - log.debug("setting hostname to %s" % hostname) - subprocess.Popen(['hostname', hostname]).communicate() + outfmt_orig = outfmt + errfmt_orig = errfmt + (outfmt, errfmt) = CC.get_output_cfg(cc.cfg,"init") + if outfmt_orig != outfmt or errfmt_orig != errfmt: + warn("stdout, stderr changing to (%s,%s)" % (outfmt,errfmt)) + CC.redirect_output(outfmt, errfmt) + except Exception, e: + warn("Failed to get and set output config: %s\n" % e) + + module_list = CC.read_cc_modules(cc.cfg,"cloud_init_modules") + + failures = [] + if len(module_list): + failures = CC.run_cc_modules(cc,module_list,log) + else: + msg = "no cloud_init_modules to run" + sys.stderr.write(msg + "\n") + log.debug(msg) + sys.exit(0) + + sys.exit(len(failures)) if __name__ == '__main__': main() |