Class: Puppet::Network::HTTP::WEBrick

Inherits:
Object
  • Object
show all
Defined in:
lib/puppet/network/http/webrick.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(args = {}) ⇒ WEBrick

Returns a new instance of WEBrick.



11
12
13
14
 
# File 'lib/puppet/network/http/webrick.rb', line 11

def initialize(args = {})
  @listening = false
  @mutex = Mutex.new
end

Class Method Details

.class_for_protocol(protocol) ⇒ Object 



16
17
18
19
 
# File 'lib/puppet/network/http/webrick.rb', line 16

def self.class_for_protocol(protocol)
  return Puppet::Network::HTTP::WEBrickREST if protocol.to_sym == :rest
  raise "Unknown protocol [#{protocol}]."
end

Instance Method Details

#listen(args = {}) ⇒ Object

Raises:

  • (ArgumentError) 


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
 
# File 'lib/puppet/network/http/webrick.rb', line 21

def listen(args = {})
  raise ArgumentError, ":protocols must be specified." if !args[:protocols] or args[:protocols].empty?
  raise ArgumentError, ":address must be specified." unless args[:address]
  raise ArgumentError, ":port must be specified." unless args[:port]

  @protocols = args[:protocols]
  @xmlrpc_handlers = args[:xmlrpc_handlers]

  arguments = {:BindAddress => args[:address], :Port => args[:port]}
  arguments.merge!(setup_logger)
  arguments.merge!(setup_ssl)

  @server = WEBrick::HTTPServer.new(arguments)
  @server.listeners.each { |l| l.start_immediately = false }

  setup_handlers

  @mutex.synchronize do
    raise "WEBrick server is already listening" if @listening
    @listening = true
    @thread = Thread.new {
      @server.start { |sock|
        raise "Client disconnected before connection could be established" unless IO.select([sock],nil,nil,0.1)
        sock.accept
        @server.run(sock)
      }
    }
    sleep 0.1 until @server.status == :Running
  end
end

#listening? ⇒ Boolean

Returns:

  • (Boolean) 


62
63
64
65
66
 
# File 'lib/puppet/network/http/webrick.rb', line 62

def listening?
  @mutex.synchronize do
    @listening
  end
end

#setup_logger ⇒ Object

Configure our http log file.



69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/puppet/network/http/webrick.rb', line 69

def setup_logger
  # Make sure the settings are all ready for us.
  Puppet.settings.use(:main, :ssl, Puppet[:name])

  if Puppet.run_mode.master?
    file = Puppet[:masterhttplog]
  else
    file = Puppet[:httplog]
  end

  # open the log manually to prevent file descriptor leak
  file_io = ::File.open(file, "a+")
  file_io.sync
  file_io.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC)

  args = [file_io]
  args << WEBrick::Log::DEBUG if Puppet::Util::Log.level == :debug

  logger = WEBrick::Log.new(*args)
  return :Logger => logger, :AccessLog => [
    [logger, WEBrick::AccessLog::COMMON_LOG_FORMAT ],
    [logger, WEBrick::AccessLog::REFERER_LOG_FORMAT ]
  ]
end

#setup_ssl ⇒ Object

Add all of the ssl cert information.

Raises:



95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/puppet/network/http/webrick.rb', line 95

def setup_ssl
  results = {}

  # Get the cached copy.  We know it's been generated, too.
  host = Puppet::SSL::Host.localhost

  raise Puppet::Error, "Could not retrieve certificate for #{host.name} and not running on a valid certificate authority" unless host.certificate

  results[:SSLPrivateKey] = host.key.content
  results[:SSLCertificate] = host.certificate.content
  results[:SSLStartImmediately] = true
  results[:SSLEnable] = true

  raise Puppet::Error, "Could not find CA certificate" unless Puppet::SSL::Certificate.find(Puppet::SSL::CA_NAME)

  results[:SSLCACertificateFile] = Puppet[:localcacert]
  results[:SSLVerifyClient] = OpenSSL::SSL::VERIFY_PEER

  results[:SSLCertificateStore] = host.ssl_store

  results
end

#unlisten ⇒ Object 



52
53
54
55
56
57
58
59
60
 
# File 'lib/puppet/network/http/webrick.rb', line 52

def unlisten
  @mutex.synchronize do
    raise "WEBrick server is not listening" unless @listening
    @server.shutdown
    @thread.join
    @server = nil
    @listening = false
  end
end