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, 92 insertions, 0 deletions

diff --git a/spec/unit/parser/functions/shellquote.rb b/spec/unit/parser/functions/shellquote.rb
new file mode 100755
index 000000000..283a4de1e
--- /dev/null
+++ b/spec/unit/parser/functions/shellquote.rb
@@ -0,0 +1,92 @@
+#! /usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe "the shellquote function" do
+
+    before :each do
+        @scope = Puppet::Parser::Scope.new()
+    end
+
+    it "should exist" do
+        Puppet::Parser::Functions.function("shellquote").should == "function_shellquote"
+    end
+
+
+    it "should handle no arguments" do
+        result = @scope.function_shellquote([])
+        result.should(eql(""))
+    end
+
+    it "should handle several simple arguments" do
+        result = @scope.function_shellquote(
+            ['foo', 'bar@example.com', 'localhost:/dev/null', 'xyzzy+-4711,23'])
+        result.should(eql(
+            'foo bar@example.com localhost:/dev/null xyzzy+-4711,23'))
+    end
+
+    it "should handle array arguments" do
+        result = @scope.function_shellquote(
+            ['foo', ['bar@example.com', 'localhost:/dev/null'],
+             'xyzzy+-4711,23'])
+        result.should(eql(
+            'foo bar@example.com localhost:/dev/null xyzzy+-4711,23'))
+    end
+
+    it "should quote unsafe characters" do
+        result = @scope.function_shellquote(
+            ['/etc/passwd ', '(ls)', '*', '[?]', "'&'"])
+        result.should(eql(
+            '"/etc/passwd " "(ls)" "*" "[?]" "\'&\'"'))
+    end
+
+    it "should deal with double quotes" do
+        result = @scope.function_shellquote(
+           ['"foo"bar"'])
+        result.should(eql(
+            '\'"foo"bar"\''))
+    end
+
+    it "should cope with dollar signs" do
+        result = @scope.function_shellquote(
+           ['$PATH', 'foo$bar', '"x$"'])
+        result.should(eql(
+            "'$PATH' 'foo$bar' '\"x$\"'"))
+    end
+
+    it "should deal with apostrophes (single quotes)" do
+        result = @scope.function_shellquote(
+           ["'foo'bar'", "`$'EDITOR'`"])
+        result.should(eql(
+            '"\'foo\'bar\'" "\\`\\$\'EDITOR\'\\`"'))
+    end
+
+    it "should cope with grave accents (backquotes)" do
+        result = @scope.function_shellquote(
+           ['`echo *`', '`ls "$MAILPATH"`'])
+        result.should(eql(
+            "'`echo *`' '`ls \"$MAILPATH\"`'"))
+    end
+
+    it "should deal with both single and double quotes" do
+        result = @scope.function_shellquote(
+           ['\'foo"bar"xyzzy\'', '"foo\'bar\'xyzzy"'])
+        result.should(eql(
+            '"\'foo\\"bar\\"xyzzy\'" "\\"foo\'bar\'xyzzy\\""'))
+    end
+
+    it "should handle multiple quotes *and* dollars and backquotes" do
+        result = @scope.function_shellquote(
+           ['\'foo"$x`bar`"xyzzy\''])
+        result.should(eql(
+            '"\'foo\\"\\$x\\`bar\\`\\"xyzzy\'"'))
+    end
+
+    it "should handle linefeeds" do
+        result = @scope.function_shellquote(
+           ["foo \n bar"])
+        result.should(eql(
+            "\"foo \n bar\""))
+    end
+
+end