modules/puppetmaster/lib/puppet/parser/functions/whohosts.rb

   1 module Puppet::Parser::Functions
   2   newfunction(:whohosts, :type => :rvalue) do |args|
   3     require 'ipaddr'
   4     require 'yaml'
   5
   6     nodeinfo = args[0]
   7     yamlfile = args[1]
   8     parser.watch_file(yamlfile)
   9
  10     $KCODE = 'utf-8'
  11
  12     ans = "unknown"
  13     yaml = YAML.load_file(yamlfile)
  14
  15     if (nodeinfo['ldap'].has_key?('ipHostNumber'))
  16       nodeinfo['ldap']['ipHostNumber'].each do |addr|
  17         yaml.keys.each do |hoster|
  18           if yaml[hoster].kind_of?(Array)
  19             netrange = yaml[hoster]
  20           elsif yaml[hoster].kind_of?(Hash) and yaml[hoster].has_key?('netrange')
  21             netrange = yaml[hoster]['netrange']
  22           else
  23             next
  24           end
  25           netrange.each do |net|
  26             begin
  27               if IPAddr.new(net).include?(addr)
  28                 ans = hoster
  29               end
  30             rescue Exception => e
  31               raise "Error while trying to match addr #{addr} for net #{net}: #{e.message}\n#{e.backtrace}"
  32             end
  33           end
  34         end
  35       end
  36     end
  37     return ans
  38   end
  39 end
  40 # vim:set ts=2:
  41 # vim:set et:
  42 # vim:set shiftwidth=2: