1a0f8e9b70d712a46326d2ce2e07cbabb6659d5e
[mirror/dsa-puppet.git] / 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: