Browse Source

own implementation of 'tunumber', so that it works correctly with

numbers outside the 'int' range
pull/9/head
Roberto Ierusalimschy 14 years ago
parent
commit
aa6faa6331
  1. 42
      lbaselib.c

42
lbaselib.c

@ -1,5 +1,5 @@
/*
** $Id: lbaselib.c,v 1.255 2010/12/13 16:38:00 roberto Exp roberto $
** $Id: lbaselib.c,v 1.256 2010/12/17 15:14:58 roberto Exp roberto $
** Basic library
** See Copyright Notice in lua.h
*/
@ -47,33 +47,39 @@ static int luaB_print (lua_State *L) {
static int luaB_tonumber (lua_State *L) {
int base = luaL_optint(L, 2, 10);
luaL_argcheck(L, 2 <= base && base <= 36, 2, "base out of range");
if (base == 10) { /* standard conversion */
luaL_checkany(L, 1);
if (lua_isnumber(L, 1)) {
lua_pushnumber(L, lua_tonumber(L, 1));
return 1;
}
} /* else not a number */
}
else {
size_t l1;
const char *s1 = luaL_checklstring(L, 1, &l1);
const char *e1 = s1 + l1;
char *s2;
unsigned long n;
size_t l;
const char *s = luaL_checklstring(L, 1, &l);
const char *e = s + l; /* end point for 's' */
int neg = 0;
luaL_argcheck(L, 2 <= base && base <= 36, 2, "base out of range");
s1 += strspn(s1, SPACECHARS); /* skip initial spaces */
if (*s1 == '-') { s1++; neg = 1; }
n = strtoul(s1, &s2, base);
if (s1 != s2) { /* at least one valid digit? */
s2 += strspn(s2, SPACECHARS); /* skip trailing spaces */
if (s2 == e1) { /* no invalid trailing characters? */
lua_pushnumber(L, (neg) ? -(lua_Number)n : (lua_Number)n);
s += strspn(s, SPACECHARS); /* skip initial spaces */
if (*s == '-') { s++; neg = 1; } /* handle signal */
else if (*s == '+') s++;
if (isalnum((unsigned char)*s)) {
lua_Number n = 0;
do {
int digit = (isdigit((unsigned char)*s)) ? *s - '0'
: toupper((unsigned char)*s) - 'A' + 10;
if (digit >= base) break; /* invalid numeral; force a fail */
n = n * (lua_Number)base + (lua_Number)digit;
s++;
} while (isalnum((unsigned char)*s));
s += strspn(s, SPACECHARS); /* skip trailing spaces */
if (s == e) { /* no invalid trailing characters? */
lua_pushnumber(L, (neg) ? -n : n);
return 1;
}
}
} /* else not a number */
} /* else not a number */
}
lua_pushnil(L); /* else not a number */
lua_pushnil(L); /* not a number */
return 1;
}

Loading…
Cancel
Save