add sym_validate_len() to validate symbol length

This also fixes an off-by-one in mrb_check_intern().

1 files changed, 11 insertions, 6 deletions

diff --git a/src/symbol.c b/src/symbol.c
index 69f9c90ee..5c55e5b65 100644
--- a/src/symbol.c
+++ b/src/symbol.c
@@ -34,6 +34,15 @@ sym_hash_func(mrb_state *mrb, const symbol_name s)
 KHASH_DECLARE(n2s, symbol_name, mrb_sym, TRUE)
 KHASH_DEFINE (n2s, symbol_name, mrb_sym, TRUE, sym_hash_func, sym_hash_equal)
 /* ------------------------------------------------------ */
+
+static void
+sym_validate_len(mrb_state *mrb, size_t len)
+{
+  if (len > (UINT16_MAX-1)) {   /* UINT16_MAX is reverved */
+    mrb_raise(mrb, E_ARGUMENT_ERROR, "symbol length too long");
+  }
+}
+
 static mrb_sym
 sym_intern(mrb_state *mrb, const char *name, size_t len, mrb_bool lit)
 {
@@ -43,9 +52,7 @@ sym_intern(mrb_state *mrb, const char *name, size_t len, mrb_bool lit)
   mrb_sym sym;
   char *p;
 
-  if (len > (UINT16_MAX-1)) {   /* UINT16_MAX is reverved */
-    mrb_raise(mrb, E_ARGUMENT_ERROR, "symbol length too long");
-  }
+  sym_validate_len(mrb, len);
   sname.lit = lit;
   sname.len = (uint16_t)len;
   sname.name = name;
@@ -100,9 +107,7 @@ mrb_check_intern(mrb_state *mrb, const char *name, size_t len)
   symbol_name sname = { 0 };
   khiter_t k;
 
-  if (len > UINT16_MAX) {
-    mrb_raise(mrb, E_ARGUMENT_ERROR, "symbol length too long");
-  }
+  sym_validate_len(mrb, len);
   sname.len = (uint16_t)len;
   sname.name = name;