summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions/is_valid_ip_address.rb
blob: 4f45890a960d7c47a1bf96f1429809b37c6684f0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#
# is_valid_ip_address.rb
#

module Puppet::Parser::Functions
  newfunction(:is_valid_ip_address, :type => :rvalue, :doc => <<-EOS
    EOS
  ) do |arguments|

    require 'ipaddr'

    if (arguments.size != 1) then
      raise(Puppet::ParseError, "is_valid_ip_address(): Wrong number of arguments "+
        "given #{arguments.size} for 1")
    end

    begin 
      ip = IPAddr.new(arguments[0])
    rescue ArgumentError
      return false
    end

    if ip.ipv4? or ip.ipv6? then
      return true
    else
      return false
    end
  end
end

# vim: set ts=2 sw=2 et :