strtol.c 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. /* $OpenBSD: strtol.c,v 1.9 2013/04/17 17:40:35 tedu Exp $ */
  2. /*-
  3. * Copyright (c) 1990 The Regents of the University of California.
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions
  8. * are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the University nor the names of its contributors
  15. * may be used to endorse or promote products derived from this software
  16. * without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21. * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  22. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28. * SUCH DAMAGE.
  29. */
  30. #include <ctype.h>
  31. #include <errno.h>
  32. #include <limits.h>
  33. #include <stdlib.h>
  34. /*
  35. * Convert a string to a long integer.
  36. *
  37. * Ignores `locale' stuff. Assumes that the upper and lower case
  38. * alphabets and digits are each contiguous.
  39. */
  40. long
  41. strtol(const char *nptr, char **endptr, int base)
  42. {
  43. const char *s;
  44. long acc, cutoff;
  45. int c;
  46. int neg, any, cutlim;
  47. /*
  48. * Ensure that base is between 2 and 36 inclusive, or the special
  49. * value of 0.
  50. */
  51. if (base != 0 && (base < 2 || base > 36)) {
  52. if (endptr != 0)
  53. *endptr = (char *)nptr;
  54. errno = EINVAL;
  55. return 0;
  56. }
  57. /*
  58. * Skip white space and pick up leading +/- sign if any.
  59. * If base is 0, allow 0x for hex and 0 for octal, else
  60. * assume decimal; if base is already 16, allow 0x.
  61. */
  62. s = nptr;
  63. do {
  64. c = (unsigned char) *s++;
  65. } while (isspace(c));
  66. if (c == '-') {
  67. neg = 1;
  68. c = *s++;
  69. } else {
  70. neg = 0;
  71. if (c == '+')
  72. c = *s++;
  73. }
  74. if ((base == 0 || base == 16) &&
  75. c == '0' && (*s == 'x' || *s == 'X')) {
  76. c = s[1];
  77. s += 2;
  78. base = 16;
  79. }
  80. if (base == 0)
  81. base = c == '0' ? 8 : 10;
  82. /*
  83. * Compute the cutoff value between legal numbers and illegal
  84. * numbers. That is the largest legal value, divided by the
  85. * base. An input number that is greater than this value, if
  86. * followed by a legal input character, is too big. One that
  87. * is equal to this value may be valid or not; the limit
  88. * between valid and invalid numbers is then based on the last
  89. * digit. For instance, if the range for longs is
  90. * [-2147483648..2147483647] and the input base is 10,
  91. * cutoff will be set to 214748364 and cutlim to either
  92. * 7 (neg==0) or 8 (neg==1), meaning that if we have accumulated
  93. * a value > 214748364, or equal but the next digit is > 7 (or 8),
  94. * the number is too big, and we will return a range error.
  95. *
  96. * Set any if any `digits' consumed; make it negative to indicate
  97. * overflow.
  98. */
  99. cutoff = neg ? LONG_MIN : LONG_MAX;
  100. cutlim = cutoff % base;
  101. cutoff /= base;
  102. if (neg) {
  103. if (cutlim > 0) {
  104. cutlim -= base;
  105. cutoff += 1;
  106. }
  107. cutlim = -cutlim;
  108. }
  109. for (acc = 0, any = 0;; c = (unsigned char) *s++) {
  110. if (isdigit(c))
  111. c -= '0';
  112. else if (isalpha(c))
  113. c -= isupper(c) ? 'A' - 10 : 'a' - 10;
  114. else
  115. break;
  116. if (c >= base)
  117. break;
  118. if (any < 0)
  119. continue;
  120. if (neg) {
  121. if (acc < cutoff || (acc == cutoff && c > cutlim)) {
  122. any = -1;
  123. acc = LONG_MIN;
  124. errno = ERANGE;
  125. } else {
  126. any = 1;
  127. acc *= base;
  128. acc -= c;
  129. }
  130. } else {
  131. if (acc > cutoff || (acc == cutoff && c > cutlim)) {
  132. any = -1;
  133. acc = LONG_MAX;
  134. errno = ERANGE;
  135. } else {
  136. any = 1;
  137. acc *= base;
  138. acc += c;
  139. }
  140. }
  141. }
  142. if (endptr != 0)
  143. *endptr = (char *) (any ? s - 1 : nptr);
  144. return (acc);
  145. }