From a716d8a6a82231d71c80e43e4c7734513de8ba2b Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Fri, 22 Jun 2012 18:07:05 -0700
Subject: 1. Fix the cfgobj to make sure it handles the quoting    of the keys
 and values in sysconfig specially by    ensuring that it does the quoting so
 that the files    written out can be sourced into bash scripts, which    is
 typically what these files are used for.

---
 cloudinit/distros/rhel.py | 61 +++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 57 insertions(+), 4 deletions(-)

(limited to 'cloudinit/distros')

diff --git a/cloudinit/distros/rhel.py b/cloudinit/distros/rhel.py
index 4ed9d43f..2c5bcab9 100644
--- a/cloudinit/distros/rhel.py
+++ b/cloudinit/distros/rhel.py
@@ -36,9 +36,24 @@ NETWORK_FN_TPL = '/etc/sysconfig/network-scripts/ifcfg-%s'
 
 # This library is used to parse/write
 # out the various sysconfig files edited
+#
+# It has to be slightly modified though
+# to ensure that all values are quoted
+# since these configs are usually sourced into
+# bash scripts...
 from configobj import ConfigObj
 
 
+# See: http://tiny.cc/oezbgw
+D_QUOTE_CHARS = {
+    "\"": "\\\"",
+    "(": "\\(",
+    ")": "\\)",
+    "$": '\$',
+    '`': '\`',
+}
+
+
 class Distro(distros.Distro):
 
     def __init__(self, name, cfg, paths):
@@ -144,14 +159,14 @@ class Distro(distros.Distro):
         else:
             return default
 
-    def _read_conf(self, filename):
+    def _read_conf(self, fn):
         exists = False
-        if os.path.isfile(filename):
-            contents = util.load_file(filename).splitlines()
+        if os.path.isfile(fn):
+            contents = util.load_file(fn).splitlines()
             exists = True
         else:
             contents = []
-        return (exists, ConfigObj(contents))
+        return (exists, QuotingConfigObj(contents))
 
     def set_timezone(self, tz):
         tz_file = os.path.join("/usr/share/zoneinfo", tz)
@@ -186,6 +201,44 @@ class Distro(distros.Distro):
         util.subp(cmd, capture=False)
 
 
+# This class helps adjust the configobj
+# writing to ensure that when writing a k/v
+# on a line, that they are properly quoted
+# and have no spaces between the '=' sign.
+# - This is mainly due to the fact that
+# the sysconfig scripts are often sourced
+# directly into bash/shell scripts so ensure
+# that it works for those types of use cases.
+class QuotingConfigObj(ConfigObj):
+    def __init__(self, lines):
+        ConfigObj.__init__(self, lines,
+                           interpolation=False,
+                           write_empty_values=True)
+
+    def _quote_posix(self, text):
+        if not text:
+            return '""'
+        for (k, v) in D_QUOTE_CHARS.iteritems():
+            text = text.replace(k, v)
+        return '"%s"' % (text)
+
+    def _write_line(self, indent_string, entry, this_entry, comment):
+        # Ensure it is formatted fine for
+        # how these sysconfig scripts are used
+        val = self._decode_element(self._quote(this_entry))
+        if not val.startswith("'"):
+            # Not already quoted, double quote
+            # it for safety
+            val = self._quote_posix(val)
+        key = self._decode_element(self._quote(entry, multiline=False))
+        cmnt = self._decode_element(comment)
+        return '%s%s%s%s%s' % (indent_string,
+                               key,
+                               "=",
+                               val,
+                               cmnt)
+
+
 # This is a util function to translate a ubuntu /etc/network/interfaces 'blob'
 # to a rhel equiv. that can then be written to /etc/sysconfig/network-scripts/
 # TODO remove when we have python-netcf active...
-- 
cgit v1.2.3