summaryrefslogtreecommitdiffhomepage
path: root/src/class.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/class.c')
-rw-r--r--src/class.c11
1 files changed, 10 insertions, 1 deletions
diff --git a/src/class.c b/src/class.c
index f00d3c782..26366aeb9 100644
--- a/src/class.c
+++ b/src/class.c
@@ -230,7 +230,16 @@ mrb_mt_foreach(mrb_state *mrb, struct RClass *c, mrb_mt_foreach_func *fn, void *
struct mt_elem *slot = &t->table[i];
if (slot->key) {
- if (fn(mrb, slot->key, slot, p) != 0)
+ mrb_method_t m;
+
+ if (slot->func_p) {
+ MRB_METHOD_FROM_FUNC(m, slot->ptr.func);
+ }
+ else {
+ MRB_METHOD_FROM_PROC(m, slot->ptr.proc);
+ }
+
+ if (fn(mrb, slot->key, m, p) != 0)
return;
}
}