class Brakeman::LibraryProcessor

Process generic library and stores it in Tracker.libs

Public Class Methods

new(tracker) click to toggle source
Calls superclass method Brakeman::BaseProcessor::new
# File lib/brakeman/processors/library_processor.rb, line 10
def initialize tracker
  super
  @current_file = nil
  @alias_processor = Brakeman::AliasProcessor.new tracker
  @current_module = nil
  @current_class = nil
  @initializer_env = nil
end

Public Instance Methods

process_call(exp) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 55
def process_call exp
  if process_call_defn? exp
    exp
  elsif @current_method.nil? and exp.target.nil? and (@current_class or @current_module)
    # Methods called inside class / module
    case exp.method
    when :include
      module_name = class_name(exp.first_arg)
      (@current_class || @current_module).add_include module_name
    end

    exp
  else
    process_default exp
  end
end
process_class(exp) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 24
def process_class exp
  handle_class exp, @tracker.libs, Brakeman::Library
end
process_defn(exp) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 32
def process_defn exp
  if exp.method_name == :initialize
    @alias_processor.process_safely exp.body_list
    @initializer_env = @alias_processor.only_ivars
  elsif node_type? exp, :defn
    exp = @alias_processor.process_safely exp, @initializer_env
  else
    exp = @alias_processor.process exp
  end

  if @current_class
    exp.body = process_all! exp.body
    @current_class.add_method :public, exp.method_name, exp, @current_file
  elsif @current_module
    exp.body = process_all! exp.body
    @current_module.add_method :public, exp.method_name, exp, @current_file
  end

  exp
end
Also aliased as: process_defs
process_defs(exp)
Alias for: process_defn
process_iter(exp) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 72
def process_iter exp
  res = process_default exp

  if node_type? res, :iter and call? exp.block_call # sometimes this changes after processing
    if exp.block_call.method == :included and (@current_module or @current_class)
      (@current_module || @current_class).options[:included] = res.block
    end
  end

  res
end
process_library(src, current_file = @current_file) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 19
def process_library src, current_file = @current_file
  @current_file = current_file
  process src
end
process_module(exp) click to toggle source
# File lib/brakeman/processors/library_processor.rb, line 28
def process_module exp
  handle_module exp, Brakeman::Library
end