presidentbeef/brakeman

require 'brakeman/checks/base_check'

class Brakeman::CheckHeaderDoS < Brakeman::BaseCheck
  Brakeman::Checks.add self

  @description = "Checks for header DoS (CVE-2013-6414)"

  def run_check
    if (version_between? "3.0.0", "3.2.15" or version_between? "4.0.0", "4.0.1") and not has_workaround?
      message = msg(msg_version(rails_version), " has a denial of service vulnerability ", msg_cve("CVE-2013-6414"), ". Upgrade to ")

      if version_between? "3.0.0", "3.2.15"
        message << msg_version("3.2.16")
      else
        message << msg_version("4.0.2")
      end

      warn :warning_type => "Denial of Service",
        :warning_code => :CVE_2013_6414,
        :message => message,
        :confidence => :medium,
        :gem_info => gemfile_or_environment,
        :link_path => "https://groups.google.com/d/msg/ruby-security-ann/A-ebV4WxzKg/KNPTbX8XAQUJ",
        :cwe_id => [20]
    end
  end

  def has_workaround?
    tracker.find_call(target: :ActiveSupport, method: :on_load).any? and
      tracker.find_call(target: :"ActionView::LookupContext::DetailsKey", method: :class_eval).any?
  end
end