return nil on attribute_dictionary call from materials or definitionl…

…ist entities (#14) Co-authored-by: Anton Chalakov <[email protected]>
ChaosGroup · Nov 7, 2024 · 79e1fa7 · 79e1fa7
1 parent 679ebac
commit 79e1fa7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ext/entity.c b/ext/entity.c
@@ -159,6 +159,10 @@ void Sketchup_Entity_attribute_dictionary_Iterator(SUAttributeDictionaryRef dict
 
 static VALUE Sketchup_Entity_attribute_dictionary(int argc, VALUE* argv, VALUE self)
 {
+	if (!rb_obj_is_kind_of(self, rb_path2class(SKETCHUP_MATERIALS)))
+		return Qnil;
+	if (!rb_obj_is_kind_of(self, rb_path2class(SKETCHUP_DEFINITIONLIST)))
+		return Qnil;
 	SUEntityRef entity = {DATA_PTR(self)};
 	struct AttributeDictionaryIterationContext attribute_dictionary_struct = {argv[0], SU_INVALID};
 	FOREACH(SUEntityGetNumAttributeDictionaries, SUEntityGetAttributeDictionaries, SUAttributeDictionaryRef, entity, Sketchup_Entity_attribute_dictionary_Iterator, &attribute_dictionary_struct);