class Sqreen::Rules::NotFoundCB

Constants

IGNORED_EXTENSIONS

Public Instance Methods

extension?(path, extensions) click to toggle source
# File lib/sqreen/rules/not_found_cb.rb, line 72
def extension?(path, extensions)
  return false if path.nil?

  candidate = File.extname(path).downcase

  extensions.include?(candidate)
end
path_from_variables(script_name, path_info, override) click to toggle source
# File lib/sqreen/rules/not_found_cb.rb, line 58
def path_from_variables(script_name, path_info, override)
  path = script_name

  if path.nil?
    path = override || path_info
  elsif override
    path += override
  elsif path_info
    path += path_info
  end

  path
end
post(rv, _inst, args, _budget = nil, &_block) click to toggle source
# File lib/sqreen/rules/not_found_cb.rb, line 14
def post(rv, _inst, args, _budget = nil, &_block)
  return if rv[0].to_i != 404

  env         = args[0]
  ua          = env['HTTP_USER_AGENT']
  script_name = env['SCRIPT_NAME']
  path_info   = env['PATH_INFO']
  verb        = env['REQUEST_METHOD']
  host        = env['SERVER_NAME']
  override    = env['action_dispatch.original_path']
  exception   = env['action_dispatch.exception']

  record_from_env(ua, script_name, path_info, verb, override, host, exception)

  nil
end
record_from_env(ua, script_name, path_info, verb, override, host, exception) click to toggle source
# File lib/sqreen/rules/not_found_cb.rb, line 31
def record_from_env(ua, script_name, path_info, verb, override, host, exception)
  path = path_from_variables(script_name, path_info, override)

  return if extension?(path, IGNORED_EXTENSIONS)

  if !override && exception && !exception.to_s.empty?
    record_from_exception({ 'ua' => ua, 'verb' => verb, 'host' => host, 'script_name' => script_name, 'path_info' => path_info }, exception.exception)
  end

  record_event({ 'path' => path, 'ua' => ua, 'verb' => verb, 'host' => host })
end
record_from_exception(payload, exception) click to toggle source
# File lib/sqreen/rules/not_found_cb.rb, line 43
def record_from_exception(payload, exception)
  message = exception.to_s

  if message && !message.empty?
    override = message =~ /No route matches\s+\[[a-z]+\]\s+"(.*)"/i && $1
  end
  payload['path'] = path_from_variables(payload['script_name'], payload['path_info'], override)
  return if extension?(payload['path'], IGNORED_EXTENSIONS)

  record = payload.reject { |k, v| v.nil? || ['path_info', 'script_name'].include?(k) }
  payload.delete('path') # remove added claim

  record_event(record)
end