class SSLTest::Runner

# File lib/ssl_test/runner.rb, line 8
def initialize(args, stdin, stdout)
  options = RunnerOptions.parse(args)
  @test_list = options.args
  @stdin = stdin
  @stdout = stdout

  @config = Config.new options.options
  cert_manager = CertificateManager.new(@config.certs)
  @logger = Logger.new(@config.logfile)
  @logger.level = @config.loglevel
  @logger.datetime_format = "%Y-%m-%d %H:%M:%S %Z"
  @logger.formatter = proc do |severity, datetime, progname, msg|
      "#{datetime}:#{severity}: #{msg}\n"
  end
  @app_context = AppContext.new(@config, cert_manager, @logger)

  @report = SSLTestReport.new
  init_packetthief
end

# File lib/ssl_test/runner.rb, line 103
def display_test(test)
  @stdout.printf "%s: %s\n", test.id, test.description
  @stdout.printf "  %s\n", test.certchainalias.inspect
  @stdout.puts ''
end

# File lib/ssl_test/runner.rb, line 73
def init_packetthief
  PacketThief.logger = @logger
  if @config.packetthief.has_key? 'implementation'
    impl = @config.packetthief['implementation']
    case impl
    when %rmanual\(/
      PacketThief.implementation = :manual
      host = %rmanual\((.*)\)/.match(impl)[1]
      PacketThief.set_dest(host, @config.packetthief.fetch('dest_port',443))
    when %rexternal\(/
      real_impl = %rexternal\((.*)\)/.match(impl)[1]
      PacketThief.implementation = real_impl.strip.downcase.to_sym
    else
      PacketThief.implementation = impl
    end
  end
end

# File lib/ssl_test/runner.rb, line 28
def run
  case @config.action
  when :list
    @stdout.puts "These are the test I will perform and their descriptions:"
    @stdout.puts ''
    SSLTestCase.factory(@app_context, @config.tests, @test_list).each do |test|
      display_test test
    end
  when :runtests
    @stdout.puts "Press spacebar to skip a test, or 'q' to stop testing."
    loginfo "Hostname being tested (assuming certs are up to date): #{@config.hosttotest}"

    @tests = SSLTestCase.factory(@app_context, @config.tests, @test_list)
    loginfo "Running #{@tests.length} tests"

    start_packetthief
    run_tests(@tests)
    stop_packetthief

    @report.print_results(@stdout)
  else
    raise "Unknown action: #{opts[:action]}"
  end
end

# File lib/ssl_test/runner.rb, line 53
def run_tests(testlist)
  test_manager = TestManager.new(@app_context, testlist, @report, @logger)
  EM.run do
    # @listener handles the initial server socket, not the accepted connections.
    # h in the code block is for each accepted connection.
    @listener = TestListener.start('', @config.listener_port, test_manager)
    @listener.logger = @logger
    @keyboard = EM.open_keyboard InputHandler do |h|
      h.on(' ') { test_manager.test_completed(test_manager.current_test, :skipped) }
      h.on('q') { test_manager.stop_testing }
      h.on("\n") { test_manager.unpause }
    end
    EM.add_periodic_timer(5) { logdebug "EM connection count: #{EM.connection_count}" }
  end
end

# File lib/ssl_test/runner.rb, line 91
def start_packetthief
  ptconf = @config.packetthief
  unless ptconf.has_key? 'implementation' and ptconf['implementation'].match(%rexternal/)
    PacketThief.redirect(:to_ports => @config.listener_port).where(ptconf).run
  end
  at_exit { PacketThief.revert }
end

# File lib/ssl_test/runner.rb, line 99
def stop_packetthief
  PacketThief.revert
end