From b64881077e272f421759149bff40e8d472621423 Mon Sep 17 00:00:00 2001 From: nobu Date: Sun, 15 Jul 2007 10:05:37 +0000 Subject: * bignum.c (rb_cstr_to_inum): check leading non-digits. [ruby-core:11691] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@12795 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- bignum.c | 31 ++++++++++++++++--------------- 1 file changed, 16 insertions(+), 15 deletions(-) (limited to 'bignum.c') diff --git a/bignum.c b/bignum.c index 51a5a43d85..74659dd40d 100644 --- a/bignum.c +++ b/bignum.c @@ -325,6 +325,13 @@ rb_cstr_to_inum(const char *str, int base, int badcheck) VALUE z; BDIGIT *zds; +#define conv_digit(c) \ + (!ISASCII(c) ? -1 : \ + isdigit(c) ? ((c) - '0') : \ + islower(c) ? ((c) - 'a' + 10) : \ + isupper(c) ? ((c) - 'A' + 10) : \ + -1) + if (!str) { if (badcheck) goto bad; return INT2FIX(0); @@ -412,7 +419,13 @@ rb_cstr_to_inum(const char *str, int base, int badcheck) } if (*str == '0') { /* squeeze preceeding 0s */ while (*++str == '0'); - --str; + if (!*str) --str; + } + c = *str; + c = conv_digit(c); + if (c < 0 || c >= base) { + if (badcheck) goto bad; + return INT2FIX(0); } len *= strlen(str)*sizeof(char); @@ -446,7 +459,7 @@ rb_cstr_to_inum(const char *str, int base, int badcheck) z = bignew(len, sign); zds = BDIGITS(z); for (i=len;i--;) zds[i]=0; - while (c = *str++) { + while ((c = *str++) != 0) { if (c == '_') { if (badcheck) { if (nondigit) goto bad; @@ -454,19 +467,7 @@ rb_cstr_to_inum(const char *str, int base, int badcheck) } continue; } - else if (!ISASCII(c)) { - break; - } - else if (isdigit(c)) { - c -= '0'; - } - else if (islower(c)) { - c -= 'a' - 10; - } - else if (isupper(c)) { - c -= 'A' - 10; - } - else { + else if ((c = conv_digit(c)) < 0) { break; } if (c >= base) break; -- cgit v1.2.3