浏览代码

refuse things like 'inf' or 'Nan' as numerals

Roberto Ierusalimschy 14 年之前
父节点
当前提交
03a078493e
共有 1 个文件被更改,包括 6 次插入4 次删除
  1. 6 4
      lobject.c

+ 6 - 4
lobject.c

@@ -1,5 +1,5 @@
 /*
 /*
-** $Id: lobject.c,v 2.51 2011/06/23 16:01:06 roberto Exp roberto $
+** $Id: lobject.c,v 2.53 2011/07/27 12:09:13 roberto Exp roberto $
 ** Some generic functions over Lua objects
 ** Some generic functions over Lua objects
 ** See Copyright Notice in lua.h
 ** See Copyright Notice in lua.h
 */
 */
@@ -103,8 +103,8 @@ static int isneg (const char **s) {
 
 
 
 
 static lua_Number readhexa (const char **s, lua_Number r, int *count) {
 static lua_Number readhexa (const char **s, lua_Number r, int *count) {
-  while (lisxdigit(cast_uchar(**s))) {  /* read integer part */
-    r = (r * 16.0) + cast_num(luaO_hexavalue(cast_uchar(*(*s)++)));
+  for (; lisxdigit(cast_uchar(**s)); (*s)++) {  /* read integer part */
+    r = (r * 16.0) + cast_num(luaO_hexavalue(cast_uchar(**s)));
     (*count)++;
     (*count)++;
   }
   }
   return r;
   return r;
@@ -157,7 +157,9 @@ static lua_Number lua_strx2number (const char *s, char **endptr) {
 
 
 int luaO_str2d (const char *s, size_t len, lua_Number *result) {
 int luaO_str2d (const char *s, size_t len, lua_Number *result) {
   char *endptr;
   char *endptr;
-  if (strpbrk(s, "xX"))  /* hexa? */
+  if (strpbrk(s, "nN"))  /* reject 'inf' and 'nan' */
+    return 0;
+  else if (strpbrk(s, "xX"))  /* hexa? */
     *result = lua_strx2number(s, &endptr);
     *result = lua_strx2number(s, &endptr);
   else
   else
     *result = lua_str2number(s, &endptr);
     *result = lua_str2number(s, &endptr);