152 lines
5.6 KiB
C++
152 lines
5.6 KiB
C++
// 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
|