strtol.c 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. /*-
  2. * SPDX-License-Identifier: BSD-3-Clause
  3. *
  4. * Copyright (c) 1990, 1993
  5. * The Regents of the University of California. All rights reserved.
  6. *
  7. * Copyright (c) 2011 The FreeBSD Foundation
  8. * All rights reserved.
  9. * Portions of this software were developed by David Chisnall
  10. * under sponsorship from the FreeBSD Foundation.
  11. *
  12. * Redistribution and use in source and binary forms, with or without
  13. * modification, are permitted provided that the following conditions
  14. * are met:
  15. * 1. Redistributions of source code must retain the above copyright
  16. * notice, this list of conditions and the following disclaimer.
  17. * 2. Redistributions in binary form must reproduce the above copyright
  18. * notice, this list of conditions and the following disclaimer in the
  19. * documentation and/or other materials provided with the distribution.
  20. * 3. Neither the name of the University nor the names of its contributors
  21. * may be used to endorse or promote products derived from this software
  22. * without specific prior written permission.
  23. *
  24. * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27. * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34. * SUCH DAMAGE.
  35. */
  36. #include <errno.h>
  37. #include <limits.h>
  38. #include <stddef.h>
  39. #include <stdlib.h>
  40. /*
  41. * Convert a string to a long integer.
  42. *
  43. * Assumes that the upper and lower case
  44. * alphabets and digits are each contiguous.
  45. */
  46. long strtol(const char *nptr, char **endptr, int base)
  47. {
  48. const char *s;
  49. unsigned long acc;
  50. char c;
  51. unsigned long cutoff;
  52. int neg, any, cutlim;
  53. /*
  54. * Skip white space and pick up leading +/- sign if any.
  55. * If base is 0, allow 0x for hex and 0 for octal, else
  56. * assume decimal; if base is already 16, allow 0x.
  57. */
  58. s = nptr;
  59. do {
  60. c = *s++;
  61. } while (isspace((unsigned char)c));
  62. if (c == '-') {
  63. neg = 1;
  64. c = *s++;
  65. } else {
  66. neg = 0;
  67. if (c == '+')
  68. c = *s++;
  69. }
  70. if ((base == 0 || base == 16) &&
  71. c == '0' && (*s == 'x' || *s == 'X') &&
  72. ((s[1] >= '0' && s[1] <= '9') ||
  73. (s[1] >= 'A' && s[1] <= 'F') ||
  74. (s[1] >= 'a' && s[1] <= 'f'))) {
  75. c = s[1];
  76. s += 2;
  77. base = 16;
  78. }
  79. if (base == 0)
  80. base = c == '0' ? 8 : 10;
  81. acc = any = 0;
  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 ? (unsigned long)-(LONG_MIN + LONG_MAX) + LONG_MAX
  100. : LONG_MAX;
  101. cutlim = cutoff % base;
  102. cutoff /= base;
  103. for ( ; ; c = *s++) {
  104. if (c >= '0' && c <= '9')
  105. c -= '0';
  106. else if (c >= 'A' && c <= 'Z')
  107. c -= 'A' - 10;
  108. else if (c >= 'a' && c <= 'z')
  109. c -= 'a' - 10;
  110. else
  111. break;
  112. if (c >= base)
  113. break;
  114. if (any < 0 || acc > cutoff || (acc == cutoff && c > cutlim))
  115. any = -1;
  116. else {
  117. any = 1;
  118. acc *= base;
  119. acc += c;
  120. }
  121. }
  122. if (any < 0) {
  123. acc = neg ? LONG_MIN : LONG_MAX;
  124. } else if (neg)
  125. acc = -acc;
  126. if (endptr != NULL)
  127. *endptr = (char *)(any ? s - 1 : nptr);
  128. return (acc);
  129. }