Tracer.set_get_line_procs should support block and Proc object

Original Tracer.set_get_line_procs is implemented by
"def set_get_line_procs(p = proc)". It means that original
Tracer.set_get_line_procs supports block and Proc object.

1 files changed, 5 insertions, 3 deletions

diff --git a/lib/tracer.rb b/lib/tracer.rb
index cf44db5fcd..8011ad2399 100644
--- a/lib/tracer.rb
+++ b/lib/tracer.rb
@@ -147,7 +147,8 @@ class Tracer
     @filters.push p
   end
 
-  def set_get_line_procs(file, &p) # :nodoc:
+  def set_get_line_procs(file, p = nil, &b) # :nodoc:
+    p ||= b
     @get_line_procs[file] = p
   end
 
@@ -249,8 +250,9 @@ class Tracer
   #     puts "line number executed is #{line}"
   #   })
 
-  def Tracer.set_get_line_procs(file_name, &p)
-    Single.set_get_line_procs(file_name, &p)
+  def Tracer.set_get_line_procs(file_name, p = nil, &b)
+    p ||= b
+    Single.set_get_line_procs(file_name, p)
   end
 
   ##