1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
module ActiveModel
module MassAssignmentSecurity
class Sanitizer #:nodoc:
# Returns all attributes not denied by the authorizer.
def sanitize(klass, attributes, authorizer)
rejected = []
sanitized_attributes = attributes.reject do |key, value|
rejected << key if authorizer.deny?(key)
end
process_removed_attributes(klass, rejected) unless rejected.empty?
sanitized_attributes
end
protected
def process_removed_attributes(klass, attrs)
raise NotImplementedError, "#process_removed_attributes(klass, attrs) is intended to be overwritten by a subclass"
end
end
class LoggerSanitizer < Sanitizer #:nodoc:
def initialize(target)
@target = target
super()
end
def logger
@target.logger
end
def logger?
@target.respond_to?(:logger) && @target.logger
end
def backtrace
if defined? Rails
Rails.backtrace_cleaner.clean(caller)
else
caller
end
end
def process_removed_attributes(klass, attrs)
if logger?
logger.warn do
"WARNING: Can't mass-assign protected attributes for #{klass.name}: #{attrs.join(', ')}\n" +
backtrace.map { |trace| "\t#{trace}" }.join("\n")
end
end
end
end
class StrictSanitizer < Sanitizer #:nodoc:
def initialize(target = nil)
super()
end
def process_removed_attributes(klass, attrs)
unless (attrs - insensitive_attributes).empty?
raise ActiveModel::MassAssignmentSecurity::Error.new(klass, attrs)
end
end
def insensitive_attributes
['id']
end
end
class Error < StandardError #:nodoc:
def initialize(klass, attrs)
super("Can't mass-assign protected attributes for #{klass.name}: #{attrs.join(', ')}")
end
end
end
end
|
