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

   1 module Puppet::Parser::Functions
   2   newfunction(:entropy_provider, :type => :rvalue) do |args|
   3     begin
   4       require '/var/lib/puppet/lib/puppet/parser/functions/whohosts.rb'
   5
   6       fqdn = args[0]
   7       nodeinfo = args[1]
   8
   9       localinfo = lookupvar('localinfo')
  10       allnodeinfo = lookupvar('allnodeinfo')
  11
  12       raise Puppet::ParseError, "entropy_provider: Cannot learn fqdn" unless fqdn
  13       raise Puppet::ParseError, "entropy_provider: Cannot learn nodeinfo" unless nodeinfo
  14       raise Puppet::ParseError, "entropy_provider: Cannot learn localinfo" unless localinfo
  15       raise Puppet::ParseError, "entropy_provider: Cannot learn allnodeinfo" unless allnodeinfo
  16
  17       # find where all the entropy keys are
  18       provider = []
  19       hoster = {}
  20       localinfo.keys.sort.each do |node|
  21         next unless  localinfo[node]['entropy_key']
  22
  23         addresses = allnodeinfo[node]['ipHostNumber']
  24         thishoster = function_whohosts(addresses, "/etc/puppet/modules/debian-org/misc/hoster.yaml")
  25         name = thishoster['name']
  26
  27         provider << node
  28
  29         hoster[name] = [] unless hoster[name]
  30         hoster[name] << node
  31       end
  32
  33       # figure out which entropy provider to use
  34       consumer_hoster = nodeinfo['hoster']['name']
  35       if provider.include?(fqdn) # if the host has an ekeyd
  36         ans = 'local'
  37       elsif hoster[consumer_hoster] # if there are any at the same hoster, use one of them
  38         # if there are more than one ekeys at this hoster pick an arbitrary
  39         # one, but the same every time
  40         index = fqdn.hash % hoster[consumer_hoster].length
  41         ans = hoster[consumer_hoster][index]
  42       else # pick an arbitrary provider from all providers
  43         index = fqdn.hash % provider.size
  44         ans = provider[index]
  45       end
  46
  47       return ans
  48     rescue => e
  49       raise Puppet::ParseError, "Error in entropy_provider: #{e.message}\n#{e.backtrace}"
  50     end
  51   end
  52 end
  53 # vim:set ts=2:
  54 # vim:set et:
  55 # vim:set shiftwidth=2: