Add shellquote() function.

This adds a new function shellquote() which can be used for quoting
arguments in shell commands used in the exec type.

This only supports Unixoid operating systems.  Other systems would
likely require some other quoting.

Signed-off-by: Thomas Bellman <bellman@nsc.liu.se>

1 files changed, 41 insertions, 0 deletions

diff --git a/lib/puppet/parser/functions/shellquote.rb b/lib/puppet/parser/functions/shellquote.rb
new file mode 100644
index 000000000..6c994520c
--- /dev/null
+++ b/lib/puppet/parser/functions/shellquote.rb
@@ -0,0 +1,41 @@
+module Puppet::Parser::Functions
+
+    Safe = 'a-zA-Z0-9@%_+=:,./-'    # Safe unquoted
+    Dangerous = '!"`$\\'            # Unsafe inside double quotes
+
+    newfunction(:shellquote, :type => :rvalue, :doc => "\
+        Quote and concatenate arguments for use in Bourne shell.
+
+        Each argument is quoted separately, and then all are concatenated
+        with spaces.  If an argument is an array, the elements of that
+        array is interpolated within the rest of the arguments; this makes
+        it possible to have an array of arguments and pass that array to
+        shellquote() instead of having to specify specify each argument
+        individually in the call.
+        ") \
+    do |args|
+
+        result = []
+        args.flatten.each do |word|
+            if word.length != 0 and word.count(Safe) == word.length
+                result << word
+            elsif word.count(Dangerous) == 0
+                result << ('"' + word + '"')
+            elsif word.count("'") == 0
+                result << ("'" + word + "'")
+            else
+                r = '"'
+                word.each_byte() do |c|
+                    if Dangerous.include?(c)
+                        r += "\\"
+                    end
+                    r += c.chr()
+                end
+                r += '"'
+                result << r
+            end
+        end
+
+        return result.join(" ")
+    end
+end