From 6b302d80a364c9433af68a0d07a1a524c020e017 Mon Sep 17 00:00:00 2001
From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org>
Date: Thu, 4 Sep 2014 18:36:28 +0900
Subject: instance variable name validation based on <ctype.h>; fix #2584

---
 src/variable.c | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

(limited to 'src/variable.c')

diff --git a/src/variable.c b/src/variable.c
index 948bd1871..440a33948 100644
--- a/src/variable.c
+++ b/src/variable.c
@@ -542,23 +542,20 @@ mrb_iv_defined(mrb_state *mrb, mrb_value obj, mrb_sym sym)
   return mrb_obj_iv_defined(mrb, mrb_obj_ptr(obj), sym);
 }
 
+#define identchar(c) (ISALNUM(c) || (c) == '_' || !ISASCII(c))
+
 MRB_API mrb_bool
 mrb_iv_p(mrb_state *mrb, mrb_sym iv_name)
 {
   const char *s;
   mrb_int i, len;
-  size_t j;
-  const char *invalid = "@$!? ";
 
   s = mrb_sym2name_len(mrb, iv_name, &len);
   if (len < 2) return FALSE;
   if (s[0] != '@') return FALSE;
   if (s[1] == '@') return FALSE;
   for (i=1; i<len; i++) {
-    char c = s[i];
-    for (j=0; j<sizeof(invalid); j++) {
-      if (c == invalid[j]) return FALSE;
-    }
+    if (!identchar(s[i])) return FALSE;
   }
   return TRUE;
 }
-- 
cgit v1.2.3