PiperOrigin-RevId: 469509635pull/13171/head
parent
12166431ed
commit
c67021f84a
5 changed files with 261 additions and 35 deletions
@ -0,0 +1,68 @@ |
||||
/*
|
||||
* Copyright (c) 2009-2021, Google LLC |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* * Neither the name of Google LLC nor the |
||||
* names of its contributors may be used to endorse or promote products |
||||
* derived from this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
* ARE DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY DIRECT, |
||||
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
|
||||
#include "upb/internal/atoi.h" |
||||
|
||||
// Must be last.
|
||||
#include "upb/port_def.inc" |
||||
|
||||
const char* upb_BufToUint64(const char* ptr, const char* end, uint64_t* val) { |
||||
uint64_t u64 = 0; |
||||
while (ptr < end) { |
||||
unsigned ch = *ptr - '0'; |
||||
if (ch >= 10) break; |
||||
if (u64 > UINT64_MAX / 10 || u64 * 10 > UINT64_MAX - ch) { |
||||
return NULL; // integer overflow
|
||||
} |
||||
u64 *= 10; |
||||
u64 += ch; |
||||
ptr++; |
||||
} |
||||
|
||||
*val = u64; |
||||
return ptr; |
||||
} |
||||
|
||||
const char* upb_BufToInt64(const char* ptr, const char* end, int64_t* val, |
||||
bool* is_neg) { |
||||
bool neg = false; |
||||
uint64_t u64; |
||||
|
||||
if (ptr != end && *ptr == '-') { |
||||
ptr++; |
||||
neg = true; |
||||
} |
||||
|
||||
ptr = upb_BufToUint64(ptr, end, &u64); |
||||
if (!ptr || u64 > (uint64_t)INT64_MAX + neg) { |
||||
return NULL; // integer overflow
|
||||
} |
||||
|
||||
*val = neg ? -u64 : u64; |
||||
if (is_neg) *is_neg = neg; |
||||
return ptr; |
||||
} |
@ -0,0 +1,53 @@ |
||||
/*
|
||||
* Copyright (c) 2009-2021, Google LLC |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* * Neither the name of Google LLC nor the |
||||
* names of its contributors may be used to endorse or promote products |
||||
* derived from this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
* ARE DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY DIRECT, |
||||
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
|
||||
#ifndef UPB_INTERNAL_ATOI_H_ |
||||
#define UPB_INTERNAL_ATOI_H_ |
||||
|
||||
// Must be last.
|
||||
#include "upb/port_def.inc" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
// We use these hand-written routines instead of strto[u]l() because the "long
|
||||
// long" variants aren't in c89. Also our version allows setting a ptr limit.
|
||||
// Return the new position of the pointer after parsing the int, or NULL on
|
||||
// integer overflow.
|
||||
|
||||
const char* upb_BufToUint64(const char* ptr, const char* end, uint64_t* val); |
||||
const char* upb_BufToInt64(const char* ptr, const char* end, int64_t* val, |
||||
bool* is_neg); |
||||
|
||||
#ifdef __cplusplus |
||||
} /* extern "C" */ |
||||
#endif |
||||
|
||||
#include "upb/port_undef.inc" |
||||
|
||||
#endif /* UPB_INTERNAL_ATOI_H_ */ |
@ -0,0 +1,109 @@ |
||||
/*
|
||||
* Copyright (c) 2009-2022, Google LLC |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* * Neither the name of Google LLC nor the |
||||
* names of its contributors may be used to endorse or promote products |
||||
* derived from this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
* ARE DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY DIRECT, |
||||
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
|
||||
#include "upb/internal/atoi.h" |
||||
|
||||
#include "gtest/gtest.h" |
||||
#include "absl/strings/str_cat.h" |
||||
|
||||
TEST(AtoiTest, Uint64) { |
||||
uint64_t val; |
||||
|
||||
const char* s = "1234z"; |
||||
EXPECT_EQ(s + 1, upb_BufToUint64(s, s + 1, &val)); |
||||
EXPECT_EQ(val, 1); |
||||
EXPECT_EQ(s + 4, upb_BufToUint64(s, s + 4, &val)); |
||||
EXPECT_EQ(val, 1234); |
||||
EXPECT_EQ(s + 4, upb_BufToUint64(s, s + 5, &val)); |
||||
EXPECT_EQ(val, 1234); |
||||
|
||||
const char* t = "42.6"; |
||||
EXPECT_EQ(t + 1, upb_BufToUint64(t, t + 1, &val)); |
||||
EXPECT_EQ(val, 4); |
||||
EXPECT_EQ(t + 2, upb_BufToUint64(t, t + 2, &val)); |
||||
EXPECT_EQ(val, 42); |
||||
EXPECT_EQ(t + 2, upb_BufToUint64(t, t + 3, &val)); |
||||
EXPECT_EQ(val, 42); |
||||
|
||||
// Integer overflow
|
||||
const char* u = "1000000000000000000000000000000"; |
||||
EXPECT_EQ(NULL, upb_BufToUint64(u, u + strlen(u), &val)); |
||||
|
||||
// Not an integer
|
||||
const char* v = "foobar"; |
||||
EXPECT_EQ(v, upb_BufToUint64(v, v + strlen(v), &val)); |
||||
|
||||
const uint64_t values[] = { |
||||
std::numeric_limits<uint64_t>::max(), |
||||
std::numeric_limits<uint64_t>::min(), |
||||
}; |
||||
for (size_t i = 0; i < ABSL_ARRAYSIZE(values); i++) { |
||||
std::string v = absl::StrCat(values[i]); |
||||
const char* ptr = v.c_str(); |
||||
const char* end = ptr + strlen(ptr); |
||||
EXPECT_EQ(end, upb_BufToUint64(ptr, end, &val)); |
||||
EXPECT_EQ(val, values[i]); |
||||
} |
||||
} |
||||
|
||||
TEST(AtoiTest, Int64) { |
||||
int64_t val; |
||||
bool neg; |
||||
|
||||
const char* s = "1234z"; |
||||
EXPECT_EQ(s + 1, upb_BufToInt64(s, s + 1, &val, &neg)); |
||||
EXPECT_EQ(val, 1); |
||||
EXPECT_EQ(neg, false); |
||||
EXPECT_EQ(s + 4, upb_BufToInt64(s, s + 4, &val, NULL)); |
||||
EXPECT_EQ(val, 1234); |
||||
EXPECT_EQ(s + 4, upb_BufToInt64(s, s + 5, &val, NULL)); |
||||
EXPECT_EQ(val, 1234); |
||||
|
||||
const char* t = "-42.6"; |
||||
EXPECT_EQ(t + 2, upb_BufToInt64(t, t + 2, &val, &neg)); |
||||
EXPECT_EQ(val, -4); |
||||
EXPECT_EQ(neg, true); |
||||
EXPECT_EQ(t + 3, upb_BufToInt64(t, t + 3, &val, NULL)); |
||||
EXPECT_EQ(val, -42); |
||||
EXPECT_EQ(neg, true); |
||||
EXPECT_EQ(t + 3, upb_BufToInt64(t, t + 5, &val, NULL)); |
||||
EXPECT_EQ(val, -42); |
||||
|
||||
const int64_t values[] = { |
||||
std::numeric_limits<int32_t>::max(), |
||||
std::numeric_limits<int32_t>::min(), |
||||
std::numeric_limits<int64_t>::max(), |
||||
std::numeric_limits<int64_t>::min(), |
||||
}; |
||||
for (size_t i = 0; i < ABSL_ARRAYSIZE(values); i++) { |
||||
std::string v = absl::StrCat(values[i]); |
||||
const char* ptr = v.c_str(); |
||||
const char* end = ptr + strlen(ptr); |
||||
EXPECT_EQ(end, upb_BufToInt64(ptr, end, &val, NULL)); |
||||
EXPECT_EQ(val, values[i]); |
||||
} |
||||
} |
Loading…
Reference in new issue