Added rb_struct_s_members()

Yorick Peterse · Yorick Peterse · commit 2642a30c5797 · 2015-05-14T00:11:07.000+02:00
This function simply returns the members of a Struct, although it
segfaults on MRI when given an instance of the return value of
Struct.new(...).
diff --git a/spec/ruby/optional/capi/ext/rubyspec.h b/spec/ruby/optional/capi/ext/rubyspec.h
@@ -599,6 +599,7 @@
 #define HAVE_RB_STRUCT_DEFINE              1
 #define HAVE_RB_STRUCT_NEW                 1
 #define HAVE_RB_STRUCT_GETMEMBER           1
+#define HAVE_RB_STRUCT_S_MEMBERS           1
 
 /* Symbol */
 #define HAVE_RB_ID2NAME                    1
diff --git a/spec/ruby/optional/capi/ext/struct_spec.c b/spec/ruby/optional/capi/ext/struct_spec.c
@@ -55,6 +55,12 @@ static VALUE struct_spec_rb_struct_new(VALUE self, VALUE klass,
 }
 #endif
 
+#ifdef HAVE_RB_STRUCT_S_MEMBERS
+static VALUE struct_spec_rb_struct_s_members(VALUE self, VALUE obj) {
+  return rb_struct_s_members(obj);
+}
+#endif
+
 void Init_struct_spec() {
   VALUE cls;
   cls = rb_define_class("CApiStructSpecs", rb_cObject);
@@ -78,6 +84,10 @@ void Init_struct_spec() {
 #ifdef HAVE_RB_STRUCT_NEW
   rb_define_method(cls, "rb_struct_new", struct_spec_rb_struct_new, 4);
 #endif
+
+#ifdef HAVE_RB_STRUCT_S_MEMBERS
+  rb_define_method(cls, "rb_struct_s_members", struct_spec_rb_struct_s_members, 1);
+#endif
 }
 
 #ifdef __cplusplus
diff --git a/spec/ruby/optional/capi/struct_spec.rb b/spec/ruby/optional/capi/struct_spec.rb
@@ -126,4 +126,18 @@
       i.c.should == 3
     end
   end
+
+  describe "rb_struct_s_members" do
+    it "returns the members of a Struct as an Array" do
+      struct = Struct.new(:foo, :bar)
+
+      @s.rb_struct_s_members(struct).should == [:foo, :bar]
+    end
+
+    it "returns the members of a Struct instance as an Array" do
+      struct = Struct.new(:foo, :bar).new
+
+      @s.rb_struct_s_members(struct).should == [:foo, :bar]
+    end
+  end
 end
diff --git a/vm/capi/struct.cpp b/vm/capi/struct.cpp
@@ -61,4 +61,8 @@ extern "C" {
   VALUE rb_struct_aset(VALUE struct_handle, VALUE key, VALUE value) {
     return rb_funcall(struct_handle, rb_intern("[]="), 2, key, value);
   }
+
+  VALUE rb_struct_s_members(VALUE obj) {
+    return rb_funcall(obj, rb_intern("members"), 0);
+  }
 }
diff --git a/vm/include/capi/ruby/ruby.h b/vm/include/capi/ruby/ruby.h
@@ -1849,6 +1849,8 @@ struct RTypedData {
   /** Sets the value of the key. */
   VALUE rb_struct_aset(VALUE s, VALUE key, VALUE value);
 
+  VALUE rb_struct_s_members(VALUE obj);
+
   /** Returns a String in locale encoding. */
   VALUE rb_locale_str_new_cstr(const char* string);
 

Original file line number	Diff line number	Diff line change
`@@ -61,4 +61,8 @@ extern "C" {`
`61`	`61`	`VALUE rb_struct_aset(VALUE struct_handle, VALUE key, VALUE value) {`
`62`	`62`	`return rb_funcall(struct_handle, rb_intern("[]="), 2, key, value);`
`63`	`63`	`}`
	`64`	`+`
	`65`	`+ VALUE rb_struct_s_members(VALUE obj) {`
	`66`	`+ return rb_funcall(obj, rb_intern("members"), 0);`
	`67`	`+ }`
`64`	`68`	`}`