From 7b2fc4e014e6b7a43e6b8fc719d82c3d0b0a8bed Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@ubuntu.com>
Date: Wed, 4 Apr 2012 21:35:32 -0400
Subject: initial checkin of kernel commandline cloud-config url support

---
 cloudinit/__init__.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
 cloudinit/util.py     | 13 +++++++++++++
 2 files changed, 57 insertions(+)

(limited to 'cloudinit')

diff --git a/cloudinit/__init__.py b/cloudinit/__init__.py
index 6d276d84..0b46dde9 100644
--- a/cloudinit/__init__.py
+++ b/cloudinit/__init__.py
@@ -137,7 +137,27 @@ class CloudInit:
 
         if ds_deps != None:
             self.ds_deps = ds_deps
+
         self.sysconfig = sysconfig
+
+        if DataSource.DEP_NETWORK in self.ds_deps:
+            target = "%s.d/%s" % (self.sysconfig, "91_kernel_cmdline_url.cfg")
+            if os.path.exists(target):
+                log.debug("cmdline: %s existed" % target)
+            else:
+                try:
+                    (key, url, content) = get_cmdline_url()
+                    if key and content:
+                        util.write_file(target, content, mode=0600)
+                        log.debug("cmdline: wrote %s from %s, %s" %
+                            (target, key, url))
+                    elif key:
+                        log.debug("cmdline: %s, %s had no cloud-config" %
+                            (key, url))
+                except Exception:
+                    util.logexc(log)
+                    log.warn("cmdline: exception occurred while reading")
+
         self.cfg = self.read_cfg()
 
     def read_cfg(self):
@@ -639,3 +659,27 @@ class InternalPartHandler:
 
     def handle_part(self, data, ctype, filename, payload, frequency):
         return(self.handler(data, ctype, filename, payload, frequency))
+
+
+def get_cmdline_url(names=('cloud-config-url', 'url'),
+                    starts="#cloud-config", cmdline=None):
+
+    if cmdline == None:
+        cmdline = util.get_cmdline()
+
+    data = util.keyval_str_to_dict(cmdline)
+    url = None
+    key = None
+    for key in names:
+        if key in data:
+            url = data[key]
+            break
+    if url == None:
+        return (None, None, None)
+
+    contents = util.readurl(url)
+
+    if contents.startswith(starts):
+        return (key, url, contents)
+
+    return (key, url, None)
diff --git a/cloudinit/util.py b/cloudinit/util.py
index 9133426c..47397418 100644
--- a/cloudinit/util.py
+++ b/cloudinit/util.py
@@ -840,3 +840,16 @@ def wait_for_url(urls, max_wait=None, timeout=None,
         time.sleep(sleeptime)
 
     return False
+
+
+def keyval_str_to_dict(kvstring):
+    ret = {}
+    for tok in kvstring.split():
+        try:
+            (key, val) = tok.split("=", 1)
+        except ValueError:
+            key = tok
+            val = True
+        ret[key] = val
+
+    return(ret)
-- 
cgit v1.2.3