class ForemanScapClient::Client

Attributes

config[R]
policy_id[R]
tailored[R]

Public Instance Methods

run(policy_id) click to toggle source
# File lib/foreman_scap_client/client.rb, line 14
def run(policy_id)
  @policy_id = policy_id
  load_config
  ensure_scan_file
  ensure_tailoring_file
  Dir.mktmpdir do |dir|
    @tmp_dir = dir
    scan
    bzip
    upload
  end
end

Private Instance Methods

bzip() click to toggle source
# File lib/foreman_scap_client/client.rb, line 76
def bzip
  puts 'DEBUG: running: ' + bzip_command
  result = %x#{bzip_command}`
  if !$?.success?
    puts 'bzip failed'
    puts results
    exit(2)
  end
end
bzip_command() click to toggle source
# File lib/foreman_scap_client/client.rb, line 72
def bzip_command
  "/usr/bin/bzip2 #{results_path}"
end
download_uri(download_path) click to toggle source
# File lib/foreman_scap_client/client.rb, line 165
def download_uri(download_path)
  foreman_proxy_uri + "#{download_path}"
end
ensure_file(dir, download_path, type_humanized) click to toggle source
# File lib/foreman_scap_client/client.rb, line 137
def ensure_file(dir, download_path, type_humanized)
  return if File.exist?(config[policy_id][dir])
  puts "File #{config[policy_id][dir]} is missing. Downloading it from proxy."
  begin
    FileUtils.mkdir_p(File.dirname(config[policy_id][dir]))
    uri = URI.parse(download_uri(config[policy_id][download_path]))
    puts "Download #{type_humanized} xml from: #{uri}"
    request = generate_https_object(uri).get(uri.path)
    request.value
    ds_content_xml = request.body
    open(config[policy_id][dir], 'wb') do |file|
      file << ds_content_xml
    end
  rescue StandardError => e
    puts "#{type_humanized} is missing and download failed with error: #{e.message}"
    exit(5)
  end
end
ensure_policy_exist() click to toggle source
# File lib/foreman_scap_client/client.rb, line 130
def ensure_policy_exist
  if config[@policy_id].nil?
    puts "Policy id #{@policy_id} not found."
    exit(1)
  end
end
ensure_scan_file() click to toggle source
# File lib/foreman_scap_client/client.rb, line 156
def ensure_scan_file
  ensure_file :content_path, :download_path, "SCAP content"
end
ensure_tailoring_file() click to toggle source
# File lib/foreman_scap_client/client.rb, line 160
def ensure_tailoring_file
  return unless tailored
  ensure_file :tailoring_path, :tailoring_download_path, "Tailoring file"
end
foreman_proxy_uri() click to toggle source
# File lib/foreman_scap_client/client.rb, line 108
def foreman_proxy_uri
  foreman_proxy_fqdn = config[:server]
  foreman_proxy_port = config[:port]
  "https://#{foreman_proxy_fqdn}:#{foreman_proxy_port}"
end
generate_https_object(uri) click to toggle source
# File lib/foreman_scap_client/client.rb, line 114
def generate_https_object(uri)
  https = Net::HTTP.new(uri.host, uri.port)
  https.use_ssl = true
  https.verify_mode = OpenSSL::SSL::VERIFY_PEER
  https.ca_file = config[:ca_file]
  begin
    https.cert = OpenSSL::X509::Certificate.new File.read(config[:host_certificate])
    https.key = OpenSSL::PKey::RSA.new File.read(config[:host_private_key])
  rescue StandardError => e
    puts 'Unable to load certs'
    puts e.message
    exit(3)
  end
  https
end
load_config() click to toggle source
# File lib/foreman_scap_client/client.rb, line 29
def load_config
  @config ||= YAML.load_file(CONFIG_FILE)
  ensure_policy_exist
  @tailored = @config[policy_id][:tailoring_path] && !@config[policy_id][:tailoring_path].empty?
rescue => e
  puts 'Config file could not be loaded'
  puts e.message
  exit(1)
end
results_bzip_path() click to toggle source
# File lib/foreman_scap_client/client.rb, line 55
def results_bzip_path
  "#{results_path}.bz2"
end
results_path() click to toggle source
# File lib/foreman_scap_client/client.rb, line 51
def results_path
  "#{@tmp_dir}/results.xml"
end
scan() click to toggle source
# File lib/foreman_scap_client/client.rb, line 39
def scan
  puts "DEBUG: running: " + scan_command
  result = %x#{scan_command}`
  if $?.success? || $?.exitstatus == 2
    @report = results_path
  else
    puts 'Scan failed'
    puts result
    exit(2)
  end
end
scan_command() click to toggle source
# File lib/foreman_scap_client/client.rb, line 59
def scan_command
  if config[@policy_id] && config[@policy_id][:profile] && !config[@policy_id][:profile].empty?
    profile = "--profile #{config[@policy_id][:profile]}"
  else
    profile = ''
  end
  "oscap xccdf eval #{profile} #{tailoring_subcommand} --results-arf #{results_path} #{config[@policy_id][:content_path]}"
end
tailoring_subcommand() click to toggle source
# File lib/foreman_scap_client/client.rb, line 68
def tailoring_subcommand
  tailored ? "--tailoring-file #{config[policy_id][:tailoring_path]}" : ""
end
upload() click to toggle source
# File lib/foreman_scap_client/client.rb, line 86
def upload
  uri = URI.parse(upload_uri)
  puts "Uploading results to #{uri}"
  https = generate_https_object(uri)
  request = Net::HTTP::Post.new uri.path
  request.body = File.read(results_bzip_path)
  request['Content-Type'] = 'text/xml'
  request['Content-Encoding'] = 'x-bzip2'
  begin
    res = https.request(request)
    res.value
  rescue StandardError => e
    puts res.body if res
    puts "Upload failed: #{e.message}"
    exit(4)
  end
end
upload_uri() click to toggle source
# File lib/foreman_scap_client/client.rb, line 104
def upload_uri
  foreman_proxy_uri + "/compliance/arf/#{@policy_id}"
end