Make sure nodeinfo['misc']['v[46]addrs'] always exists, possibly empty.
[mirror/dsa-puppet.git] / modules / puppetmaster / lib / puppet / parser / functions / nodeinfo.rb
1 module Puppet::Parser::Functions
2   newfunction(:nodeinfo, :type => :rvalue) do |args|
3     host = args[0]
4     yamlfile = args[1]
5     begin
6       nodeinfo         = function_yamlinfo([host, yamlfile])
7       nodeinfo['ldap'] = function_ldapinfo([host, '*'])
8       unless nodeinfo['ldap']['ipHostNumber']
9         raise Puppet::ParseError, "Host #{host} does not have ipHostNumber values in ldap"
10       end
11       nodeinfo['hoster'] = function_whohosts([nodeinfo['ldap']['ipHostNumber']])
12       nodeinfo['buildd'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('buildd'))
13       nodeinfo['timeserver'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('timeserver'))
14       nodeinfo['porterbox'] = (nodeinfo['ldap']['purpose'].respond_to?('include?') && nodeinfo['ldap']['purpose'].include?('porterbox'))
15
16       if lookupvar('::mta') == 'exim4'
17         unless nodeinfo['heavy_exim']
18           nodeinfo['smarthost'] = 'mailout.debian.org'
19         end
20       elsif lookupvar('::mta') == 'postfix'
21         unless nodeinfo['heavy_postfix']
22           nodeinfo['smarthost'] = 'mailout.debian.org'
23         end
24       end
25
26       nodeinfo['misc'] = {}
27       fqdn = lookupvar('::fqdn')
28       if fqdn and fqdn == host
29         v4ips = lookupvar('::v4ips')
30         if v4ips and v4ips.to_s != "" and v4ips.to_s != 'undefined'
31           nodeinfo['misc']['v4addrs'] = v4ips.split(',')
32
33           # find out if we are behind nat
34           intersection = nodeinfo['misc']['v4addrs'] & nodeinfo['ldap']['ipHostNumber']
35           nodeinfo['misc']['natted'] = intersection.empty?
36         else
37           nodeinfo['misc']['v4addrs'] = []
38         end
39
40         v6ips = lookupvar('::v6ips')
41         if v6ips and v6ips.to_s != "" and v6ips.to_s != 'undefined'
42           nodeinfo['misc']['v6addrs'] = v6ips.split(',')
43         else
44           nodeinfo['misc']['v6addrs'] = []
45         end
46       end
47
48       ns = call_function('hiera',['nameservers'])
49       allow_dns_q = call_function('hiera',['allow_dns_query'])
50       if ns.empty?
51         # no nameservers known for this hoster
52         nodeinfo['misc']['resolver-recursive'] = true
53
54         if not allow_dns_q.empty?
55           raise Puppet::ParseError, "No nameservers listed for #{nodeinfo['hoster']['name']} yet we should answer somebody's queries?  That makes no sense.  allow_dns_q: #{allow_dns_q}."
56         end
57       elsif (nodeinfo['misc']['v4addrs'] and (ns & nodeinfo['misc']['v4addrs']).size > 0) or
58             (nodeinfo['misc']['v6addrs'] and (ns & nodeinfo['misc']['v6addrs']).size > 0)
59         # this host is listed as a nameserver at this location
60         nodeinfo['misc']['resolver-recursive'] = true
61
62         if allow_dns_q.empty?
63           raise Puppet::ParseError, "Host #{host} is listed as a nameserver for #{nodeinfo['hoster']['name']} but no allow_dns_query networks are defined for this location"
64         end
65       else
66         nodeinfo['misc']['resolver-recursive'] = false
67       end
68
69       return(nodeinfo)
70     rescue => e
71       raise Puppet::ParseError, "Error in nodeinfo for node #{host}, yamlfile #{yamlfile}: #{e.message}\n#{e.backtrace}"
72     end
73   end
74 end
75
76 # vim: set fdm=marker ts=2 sw=2 et: