summaryrefslogtreecommitdiffstats
path: root/lib/puppet/network/handler/master.rb
blob: 62aab539e8ef185412bb137ee3e7a4dfa20056fa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
require 'openssl'
require 'puppet'
require 'puppet/sslcertificates'
require 'xmlrpc/server'
require 'yaml'

class Puppet::Network::Handler
  class MasterError < Puppet::Error; end
  class Master < Handler
    desc "Puppet's configuration interface.  Used for all interactions related to
    generating client configurations."

    include Puppet::Util

    attr_accessor :ast
    attr_reader :ca

    @interface = XMLRPC::Service::Interface.new("puppetmaster") { |iface|
        iface.add_method("string getconfig(string)")
        iface.add_method("int freshness()")
    }

    # Tell a client whether there's a fresh config for it
    def freshness(client = nil, clientip = nil)
      # Always force a recompile.  Newer clients shouldn't do this (as of April 2008).
      Time.now.to_i
    end

    def initialize(hash = {})
      args = {}

      @local = hash[:Local]

      args[:Local] = true

      @ca = (hash.include?(:CA) and hash[:CA]) ? Puppet::SSLCertificates::CA.new : nil

      # This is only used by the cfengine module, or if --loadclasses was
      # specified in +puppet+.
      args[:Classes] = hash[:Classes] if hash.include?(:Classes)
    end

    # Call our various handlers; this handler is getting deprecated.
    def getconfig(facts, format = "marshal", client = nil, clientip = nil)
      facts = decode_facts(facts)

      client ||= facts["hostname"]

      # Pass the facts to the fact handler
      Puppet::Node::Facts.indirection.save(Puppet::Node::Facts.new(client, facts)) unless local?

      catalog = Puppet::Resource::Catalog.indirection.find(client)

      case format
      when "yaml"
        return CGI.escape(catalog.extract.to_yaml)
      when "marshal"
        return CGI.escape(Marshal.dump(catalog.extract))
      else
        raise "Invalid markup format '#{format}'"
      end
    end

    #
    def decode_facts(facts)
      if @local
        # we don't need to do anything, since we should already
        # have raw objects
        Puppet.debug "Our client is local"
      else
        Puppet.debug "Our client is remote"

        begin
          facts = YAML.load(CGI.unescape(facts))
        rescue => detail
          raise XMLRPC::FaultException.new(
            1, "Could not rebuild facts"
          )
        end
      end

      facts
    end

    # Translate our configuration appropriately for sending back to a client.
    def translate(config)
    end
  end
end