Class: Brakeman::CheckDeserialize

Inherits:
BaseCheck show all
Defined in:
lib/brakeman/checks/check_deserialize.rb

Constant Summary

Constants inherited from BaseCheck

BaseCheck::CONFIDENCE

Constants included from Util

Util::ALL_PARAMETERS, Util::COOKIES, Util::COOKIES_SEXP, Util::PARAMETERS, Util::PARAMS_SEXP, Util::PATH_PARAMETERS, Util::QUERY_PARAMETERS, Util::REQUEST_ENV, Util::REQUEST_PARAMETERS, Util::REQUEST_PARAMS, Util::SESSION, Util::SESSION_SEXP

Constants inherited from SexpProcessor

SexpProcessor::VERSION

Instance Attribute Summary

Attributes inherited from BaseCheck

#tracker, #warnings

Attributes inherited from SexpProcessor

#context, #env, #expected

Instance Method Summary collapse

Methods inherited from BaseCheck

#add_result, inherited, #initialize, #process_call, #process_cookies, #process_default, #process_if, #process_params, #process_string_interp

Methods included from Util

#array?, #block?, #call?, #camelize, #contains_class?, #context_for, #cookies?, #false?, #file_by_name, #file_for, #github_url, #hash?, #hash_access, #hash_insert, #hash_iterate, #integer?, #make_call, #node_type?, #number?, #params?, #pluralize, #regexp?, #relative_path, #request_env?, #request_value?, #result?, #set_env_defaults, #sexp?, #string?, #symbol?, #table_to_csv, #template_path_to_name, #true?, #truncate_table, #underscore

Methods included from ProcessorHelper

#class_name, #process_all, #process_all!, #process_call_args, #process_class, #process_module

Methods inherited from SexpProcessor

#error_handler, #in_context, #initialize, #process, #process_dummy, #scope

Constructor Details

This class inherits a constructor from Brakeman::BaseCheck

Instance Method Details

#check_csvObject


18
19
20
# File 'lib/brakeman/checks/check_deserialize.rb', line 18

def check_csv
  check_methods :CSV, :load
end

#check_deserialize(result, target, arg = nil) ⇒ Object


32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# File 'lib/brakeman/checks/check_deserialize.rb', line 32

def check_deserialize result, target, arg = nil
  return if duplicate? result
  add_result result

  arg ||= result[:call].first_arg
  method = result[:call].method

  if input = has_immediate_user_input?(arg)
    confidence = CONFIDENCE[:high]
  elsif input = include_user_input?(arg)
    confidence = CONFIDENCE[:med]
  end

  if confidence
    message = "#{target}.#{method} called with #{friendly_type_of input}"

    warn :result => result,
      :warning_type => "Remote Code Execution",
      :warning_code => :unsafe_deserialize,
      :message => message,
      :user_input => input.match,
      :confidence => confidence,
      :link_path => "unsafe_deserialization"
  end
end

#check_marshalObject


22
23
24
# File 'lib/brakeman/checks/check_deserialize.rb', line 22

def check_marshal
  check_methods :Marshal, :load, :restore
end

#check_methods(target, *methods) ⇒ Object


26
27
28
29
30
# File 'lib/brakeman/checks/check_deserialize.rb', line 26

def check_methods target, *methods
  tracker.find_call(:target => target, :methods => methods ).each do |result|
    check_deserialize result, target
  end
end

#check_yamlObject


14
15
16
# File 'lib/brakeman/checks/check_deserialize.rb', line 14

def check_yaml
  check_methods :YAML, :load, :load_documents, :load_stream, :parse_documents, :parse_stream
end

#run_checkObject


8
9
10
11
12
# File 'lib/brakeman/checks/check_deserialize.rb', line 8

def run_check
  check_yaml
  check_csv
  check_marshal
end