| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <gtest/gtest.h> |
| |
| #include <sstream> |
| #include <vector> |
| |
| #include "base/safe_numerics.h" |
| |
| namespace base { |
| namespace internal { |
| |
| // This is far (far, far) too slow to run normally, but if you're refactoring |
| // it might be useful. |
| // #define RUN_EXHAUSTIVE_TEST |
| |
| #ifdef RUN_EXHAUSTIVE_TEST |
| |
| template <class From, class To> void ExhaustiveCheckFromTo() { |
| fprintf(stderr, "."); |
| From i = std::numeric_limits<From>::min(); |
| for (;;) { |
| std::ostringstream str_from, str_to; |
| str_from << i; |
| To to = static_cast<To>(i); |
| str_to << to; |
| bool strings_equal = str_from.str() == str_to.str(); |
| EXPECT_EQ(IsValidNumericCast<To>(i), strings_equal); |
| fprintf(stderr, "\r%s vs %s\x1B[K", |
| str_from.str().c_str(), str_to.str().c_str()); |
| ++i; |
| // If we wrap, then we've tested everything. |
| if (i == std::numeric_limits<From>::min()) |
| break; |
| } |
| } |
| |
| template <class From> void ExhaustiveCheckFrom() { |
| ExhaustiveCheckFromTo<From, short>(); |
| ExhaustiveCheckFromTo<From, unsigned short>(); |
| ExhaustiveCheckFromTo<From, int>(); |
| ExhaustiveCheckFromTo<From, unsigned int>(); |
| ExhaustiveCheckFromTo<From, long long>(); |
| ExhaustiveCheckFromTo<From, unsigned long long>(); |
| ExhaustiveCheckFromTo<From, size_t>(); |
| fprintf(stderr, "\n"); |
| } |
| |
| #endif |
| |
| |
| TEST(SafeNumerics, NumericCast) { |
| int small_positive = 1; |
| int small_negative = -1; |
| int large_positive = INT_MAX; |
| int large_negative = INT_MIN; |
| size_t size_t_small = 1; |
| size_t size_t_large = UINT_MAX; |
| |
| // Narrow signed destination. |
| EXPECT_TRUE(IsValidNumericCast<signed char>(small_positive)); |
| EXPECT_TRUE(IsValidNumericCast<signed char>(small_negative)); |
| EXPECT_FALSE(IsValidNumericCast<signed char>(large_positive)); |
| EXPECT_FALSE(IsValidNumericCast<signed char>(large_negative)); |
| EXPECT_TRUE(IsValidNumericCast<signed short>(small_positive)); |
| EXPECT_TRUE(IsValidNumericCast<signed short>(small_negative)); |
| |
| // Narrow unsigned destination. |
| EXPECT_TRUE(IsValidNumericCast<unsigned char>(small_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned char>(small_negative)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned char>(large_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned char>(large_negative)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned short>(small_negative)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned short>(large_negative)); |
| |
| // Same width signed destination. |
| EXPECT_TRUE(IsValidNumericCast<signed int>(small_positive)); |
| EXPECT_TRUE(IsValidNumericCast<signed int>(small_negative)); |
| EXPECT_TRUE(IsValidNumericCast<signed int>(large_positive)); |
| EXPECT_TRUE(IsValidNumericCast<signed int>(large_negative)); |
| |
| // Same width unsigned destination. |
| EXPECT_TRUE(IsValidNumericCast<unsigned int>(small_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned int>(small_negative)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned int>(large_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned int>(large_negative)); |
| |
| // Wider signed destination. |
| EXPECT_TRUE(IsValidNumericCast<long long>(small_positive)); |
| EXPECT_TRUE(IsValidNumericCast<long long>(large_negative)); |
| EXPECT_TRUE(IsValidNumericCast<long long>(small_positive)); |
| EXPECT_TRUE(IsValidNumericCast<long long>(large_negative)); |
| |
| // Wider unsigned destination. |
| EXPECT_TRUE(IsValidNumericCast<unsigned long long>(small_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned long long>(small_negative)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned long long>(large_positive)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned long long>(large_negative)); |
| |
| // Negative to size_t. |
| EXPECT_FALSE(IsValidNumericCast<size_t>(small_negative)); |
| EXPECT_FALSE(IsValidNumericCast<size_t>(large_negative)); |
| |
| // From unsigned. |
| // Small. |
| EXPECT_TRUE(IsValidNumericCast<signed char>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned char>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<short>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned short>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<int>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned int>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<long long>(size_t_small)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned long long>(size_t_small)); |
| |
| // Large. |
| EXPECT_FALSE(IsValidNumericCast<signed char>(size_t_large)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned char>(size_t_large)); |
| EXPECT_FALSE(IsValidNumericCast<short>(size_t_large)); |
| EXPECT_FALSE(IsValidNumericCast<unsigned short>(size_t_large)); |
| EXPECT_FALSE(IsValidNumericCast<int>(size_t_large)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned int>(size_t_large)); |
| EXPECT_TRUE(IsValidNumericCast<long long>(size_t_large)); |
| EXPECT_TRUE(IsValidNumericCast<unsigned long long>(size_t_large)); |
| |
| // Various edge cases. |
| EXPECT_TRUE(IsValidNumericCast<int>(static_cast<short>(SHRT_MIN))); |
| EXPECT_FALSE( |
| IsValidNumericCast<unsigned short>(static_cast<short>(SHRT_MIN))); |
| EXPECT_FALSE(IsValidNumericCast<unsigned short>(SHRT_MIN)); |
| |
| // Confirm that checked_numeric_cast<> actually compiles. |
| std::vector<int> v; |
| unsigned int checked_size = |
| base::checked_numeric_cast<unsigned int>(v.size()); |
| EXPECT_EQ(0u, checked_size); |
| |
| #ifdef RUN_EXHAUSTIVE_TEST |
| ExhaustiveCheckFrom<short>(); |
| ExhaustiveCheckFrom<unsigned short>(); |
| ExhaustiveCheckFrom<int>(); |
| ExhaustiveCheckFrom<unsigned int>(); |
| ExhaustiveCheckFrom<long long>(); |
| ExhaustiveCheckFrom<unsigned long long>(); |
| ExhaustiveCheckFrom<size_t>(); |
| #endif |
| } |
| |
| } // namespace internal |
| } // namespace base |