diff options
author | knu <knu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-06-03 04:03:15 +0000 |
---|---|---|
committer | knu <knu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-06-03 04:03:15 +0000 |
commit | 1434fc3fa86b51a344c0d6176229229b50d87a95 (patch) | |
tree | 094306fe910429085fb25745ffa202de9243ad04 /enumerator.c | |
parent | 0eb420cc78e5027a2611057a535fd255e87723fb (diff) |
* enumerator.c (enumerator_allocate, enumerator_ptr): Properly
detect if the object is initialized and raise error when
appropriate.
(enumerator_initialize): Fix a typo in rdoc. [ruby-core:17052]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8_7@16769 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'enumerator.c')
-rw-r--r-- | enumerator.c | 22 |
1 files changed, 16 insertions, 6 deletions
diff --git a/enumerator.c b/enumerator.c index e0365fc5bc..8d5061d308 100644 --- a/enumerator.c +++ b/enumerator.c @@ -55,7 +55,7 @@ enumerator_ptr(obj) "wrong argument type %s (expected Enumerable::Enumerator)", rb_obj_classname(obj)); } - if (!ptr) { + if (!ptr || ptr->obj == Qundef) { rb_raise(rb_eArgError, "uninitialized enumerator"); } return ptr; @@ -214,8 +214,13 @@ enumerator_allocate(klass) VALUE klass; { struct enumerator *ptr; - return Data_Make_Struct(klass, struct enumerator, - enumerator_mark, -1, ptr); + VALUE enum_obj; + + enum_obj = Data_Make_Struct(klass, struct enumerator, + enumerator_mark, -1, ptr); + ptr->obj = Qundef; + + return enum_obj; } static VALUE enumerator_each_i _((VALUE, VALUE)); @@ -235,7 +240,13 @@ enumerator_init(enum_obj, obj, meth, argc, argv) int argc; VALUE *argv; { - struct enumerator *ptr = enumerator_ptr(enum_obj); + struct enumerator *ptr; + + Data_Get_Struct(enum_obj, struct enumerator, ptr); + + if (!ptr) { + rb_raise(rb_eArgError, "unallocated enumerator"); + } ptr->obj = obj; ptr->meth = rb_to_id(meth); @@ -253,8 +264,7 @@ enumerator_init(enum_obj, obj, meth, argc, argv) * used as an Enumerable object using the given object's given * method with the given arguments. * - * Use of this method is not discouraged. Use Kernel#enum_for() - * instead. + * Use of this method is discouraged. Use Kernel#enum_for() instead. */ static VALUE enumerator_initialize(argc, argv, obj) |