summaryrefslogtreecommitdiff
path: root/lib/puppet
diff options
context:
space:
mode:
authorHunter Haugen <hunter@puppetlabs.com>2016-02-18 10:32:25 -0800
committerHunter Haugen <hunter@puppetlabs.com>2016-02-18 10:32:25 -0800
commit69ca8d09c6124e9b5bd98e62fb201e734ae5bd05 (patch)
tree0193401705d5adee95196efc9830240a596992c4 /lib/puppet
parent2c3beace2438882fa7dfe89ad5817ea7fe7cba21 (diff)
parentd85aec41a3b57a13f16086cc1ff7ed2fe09602b8 (diff)
Merge pull request #570 from gfidente/master
Add is_ipv4_address and is_ipv6_address functions
Diffstat (limited to 'lib/puppet')
-rw-r--r--lib/puppet/parser/functions/is_ipv4_address.rb28
-rw-r--r--lib/puppet/parser/functions/is_ipv6_address.rb28
2 files changed, 56 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/is_ipv4_address.rb b/lib/puppet/parser/functions/is_ipv4_address.rb
new file mode 100644
index 0000000..b4861d5
--- /dev/null
+++ b/lib/puppet/parser/functions/is_ipv4_address.rb
@@ -0,0 +1,28 @@
+#
+# is_ipv4_address.rb
+#
+
+module Puppet::Parser::Functions
+ newfunction(:is_ipv4_address, :type => :rvalue, :doc => <<-EOS
+Returns true if the string passed to this function is a valid IPv4 address.
+ EOS
+ ) do |arguments|
+
+ require 'ipaddr'
+
+ if (arguments.size != 1) then
+ raise(Puppet::ParseError, "is_ipv4_address(): Wrong number of arguments "+
+ "given #{arguments.size} for 1")
+ end
+
+ begin
+ ip = IPAddr.new(arguments[0])
+ rescue ArgumentError
+ return false
+ end
+
+ return ip.ipv4?
+ end
+end
+
+# vim: set ts=2 sw=2 et :
diff --git a/lib/puppet/parser/functions/is_ipv6_address.rb b/lib/puppet/parser/functions/is_ipv6_address.rb
new file mode 100644
index 0000000..475ad50
--- /dev/null
+++ b/lib/puppet/parser/functions/is_ipv6_address.rb
@@ -0,0 +1,28 @@
+#
+# is_ipv6_address.rb
+#
+
+module Puppet::Parser::Functions
+ newfunction(:is_ipv6_address, :type => :rvalue, :doc => <<-EOS
+Returns true if the string passed to this function is a valid IPv6 address.
+ EOS
+ ) do |arguments|
+
+ require 'ipaddr'
+
+ if (arguments.size != 1) then
+ raise(Puppet::ParseError, "is_ipv6_address(): Wrong number of arguments "+
+ "given #{arguments.size} for 1")
+ end
+
+ begin
+ ip = IPAddr.new(arguments[0])
+ rescue ArgumentError
+ return false
+ end
+
+ return ip.ipv6?
+ end
+end
+
+# vim: set ts=2 sw=2 et :