From 27df1a84bc5310bcd32d90b7157a3ad46ef3ac76 Mon Sep 17 00:00:00 2001
From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org>
Date: Wed, 14 May 2014 17:38:27 +0900
Subject: Kernel.local_variables to list UPVAR as well

---
 mrbgems/mruby-proc-ext/src/proc.c   | 20 +++++++++++++++++++-
 mrbgems/mruby-proc-ext/test/proc.rb |  6 ++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/mrbgems/mruby-proc-ext/src/proc.c b/mrbgems/mruby-proc-ext/src/proc.c
index 341001f05..f0e677045 100644
--- a/mrbgems/mruby-proc-ext/src/proc.c
+++ b/mrbgems/mruby-proc-ext/src/proc.c
@@ -137,10 +137,28 @@ mrb_local_variables(mrb_state *mrb, mrb_value self)
   }
 
   irep = proc->body.irep;
+  if (!irep->lv) {
+    return mrb_ary_new(mrb);
+  }
   ret = mrb_ary_new_capa(mrb, irep->nlocals - 1);
-  for (i = 0; i < (irep->nlocals - 1); ++i) {
+  for (i = 0; i + 1 < irep->nlocals; ++i) {
     mrb_ary_push(mrb, ret, mrb_symbol_value(irep->lv[i].name));
   }
+  if (proc->env) {
+    struct REnv *e = proc->env;
+
+    while (e) {
+      if (!MRB_PROC_CFUNC_P(mrb->c->cibase[e->cioff].proc)) {
+        irep = mrb->c->cibase[e->cioff].proc->body.irep;
+        if (irep->lv) {
+          for (i = 0; i + 1 < irep->nlocals; ++i) {
+            mrb_ary_push(mrb, ret, mrb_symbol_value(irep->lv[i].name));
+          }
+        }
+      }
+      e = (struct REnv*)e->c;
+    }
+  }
 
   return ret;
 }
diff --git a/mrbgems/mruby-proc-ext/test/proc.rb b/mrbgems/mruby-proc-ext/test/proc.rb
index a5611e586..4da98de05 100644
--- a/mrbgems/mruby-proc-ext/test/proc.rb
+++ b/mrbgems/mruby-proc-ext/test/proc.rb
@@ -81,4 +81,10 @@ assert('Kernel.local_variables') do
 
   vars = Kernel.local_variables.sort
   assert_equal [:a, :b, :vars], vars
+
+  proc {
+    c = 2
+    vars = Kernel.local_variables.sort
+    assert_equal [:a, :b, :c, :vars], vars
+  }.call
 end
-- 
cgit v1.2.3