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
|
# File 'lib/newrelic_security/instrumentation-security/sqlite3/chain.rb', line 7
def self.instrument!
::SQLite3::Database.class_eval do
include NewRelic::Security::Instrumentation::SQLite3::Database
alias_method :execute_without_security, :execute
def execute(sql, bind_vars = [], *args, &block)
retval = nil
event = execute_on_enter(sql, bind_vars, *args) { retval = execute_without_security(sql, bind_vars, *args, &block) }
execute_on_exit(event) { return retval }
end
alias_method :execute_batch_without_security, :execute_batch
def execute_batch(sql, bind_vars = [], *args)
retval = nil
event = execute_batch_on_enter(sql, bind_vars, *args) { retval = execute_batch_without_security(sql, bind_vars, *args) }
execute_batch_on_exit(event) { return retval }
end
if ::SQLite3::VERSION >= '1.4'
alias_method :execute_batch2_without_security, :execute_batch2
def execute_batch2(sql, &block)
retval = nil
event = execute_batch2_on_enter(sql) { retval = execute_batch2_without_security(sql, &block) }
execute_batch2_on_exit(event) { return retval }
end
end
end
end
|