29 lines
919 B
Ruby
29 lines
919 B
Ruby
#
|
|
# cidr_to_netmask.rb
|
|
#
|
|
require 'ipaddr'
|
|
begin
|
|
require 'puppet/parser/functions/lib/prepare_cidr.rb'
|
|
rescue LoadError => e
|
|
# puppet apply does not add module lib directories to the $LOAD_PATH (See
|
|
# #4248). It should (in the future) but for the time being we need to be
|
|
# defensive which is what this rescue block is doing.
|
|
rb_file = File.join(File.dirname(__FILE__),'lib','prepare_cidr.rb')
|
|
load rb_file if File.exists?(rb_file) or raise e
|
|
end
|
|
|
|
module Puppet::Parser::Functions
|
|
newfunction(:cidr_to_netmask, :type => :rvalue, :doc => <<-EOS
|
|
This function get cidr-notated IP addresses and return netmask.
|
|
EOS
|
|
) do |arguments|
|
|
if arguments.size != 1
|
|
raise(Puppet::ParseError, "cidr_to_netmask(): Wrong number of arguments " +
|
|
"given (#{arguments.size} for 1)")
|
|
end
|
|
|
|
masklen = prepare_cidr(arguments[0])[1]
|
|
return IPAddr.new('255.255.255.255').mask(masklen).to_s
|
|
end
|
|
end
|