Chromium Code Reviews| Index: base/safe_numerics.h |
| diff --git a/base/safe_numerics.h b/base/safe_numerics.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..3d47cf24ed5eb9553ae4edae285699ee36a0e923 |
| --- /dev/null |
| +++ b/base/safe_numerics.h |
| @@ -0,0 +1,135 @@ |
| +// 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. |
| + |
| +#ifndef BASE_SAFE_NUMERICS_H_ |
| +#define BASE_SAFE_NUMERICS_H_ |
| + |
| +#include <limits> |
| + |
| +#include "base/logging.h" |
| + |
| +namespace base { |
| +namespace internal { |
| + |
| +template <bool SameSize, bool DestLarger, |
| + bool DestIsSigned, bool SourceIsSigned> |
| +struct IsValidNumericCastImpl; |
| + |
| +#define BASE_NUMERIC_CAST_CASE_SPECIALIZATION(A, B, C, D, Code) \ |
| +template <> struct IsValidNumericCastImpl<A, B, C, D> { \ |
| + template <class Source, class DestBounds> static inline bool Test( \ |
| + Source source, DestBounds min, DestBounds max) { \ |
| + return Code; \ |
| + } \ |
| +} |
| + |
| +#define BASE_NUMERIC_CAST_CASE_SAME_SIZE(DestSigned, SourceSigned, Code) \ |
| + BASE_NUMERIC_CAST_CASE_SPECIALIZATION( \ |
| + true, true, DestSigned, SourceSigned, Code); \ |
| + BASE_NUMERIC_CAST_CASE_SPECIALIZATION( \ |
| + true, false, DestSigned, SourceSigned, Code) |
| + |
| +#define BASE_NUMERIC_CAST_CASE_SOURCE_LARGER(DestSigned, SourceSigned, Code) \ |
| + BASE_NUMERIC_CAST_CASE_SPECIALIZATION( \ |
| + false, false, DestSigned, SourceSigned, Code); \ |
| + |
| +#define BASE_NUMERIC_CAST_CASE_DEST_LARGER(DestSigned, SourceSigned, Code) \ |
| + BASE_NUMERIC_CAST_CASE_SPECIALIZATION( \ |
| + false, true, DestSigned, SourceSigned, Code); \ |
| + |
| +// The three top level cases are: |
| +// - Same size |
| +// - Source larger |
| +// - Dest larger |
| +// And for each of those three cases, we handle the 4 different possibilities |
| +// of signed and unsigned. This gives 12 cases to handle, which we enumerate |
| +// below. |
| +// |
| +// The last argument in each of the macros is the actual comparison code. It |
| +// has three arguments available, source (the value), and min/max which are |
| +// the ranges of the destination. |
| + |
| + |
| +// These are the cases where both types have the same size. |
| + |
| +// Both signed. |
| +BASE_NUMERIC_CAST_CASE_SAME_SIZE(true, true, true); |
| +// Both unsigned. |
| +BASE_NUMERIC_CAST_CASE_SAME_SIZE(false, false, true); |
| +// Dest unsigned, Source signed. |
| +BASE_NUMERIC_CAST_CASE_SAME_SIZE(false, true, source >= 0); |
| +// Dest signed, Source unsigned. |
| +// This cast is OK because Dest's max must be less than Source's. |
| +BASE_NUMERIC_CAST_CASE_SAME_SIZE(true, false, |
| + source <= static_cast<Source>(max)); |
| + |
| + |
| +// These are the cases where Source is larger. |
| + |
| +// Both unsigned. |
| +BASE_NUMERIC_CAST_CASE_SOURCE_LARGER(false, false, source <= max); |
| +// Both signed. |
| +BASE_NUMERIC_CAST_CASE_SOURCE_LARGER(true, true, |
| + source >= min && source <= max); |
| +// Dest is unsigned, Source is signed. |
| +BASE_NUMERIC_CAST_CASE_SOURCE_LARGER(false, true, |
| + source >= 0 && source <= max); |
| +// Dest is signed, Source is unsigned. |
| +// This cast is OK because Dest's max must be less than Source's. |
| +BASE_NUMERIC_CAST_CASE_SOURCE_LARGER(true, false, |
| + source <= static_cast<Source>(max)); |
| + |
| + |
| +// These are the cases where Dest is larger. |
| + |
| +// Both unsigned. |
| +BASE_NUMERIC_CAST_CASE_DEST_LARGER(false, false, true); |
| +// Both signed. |
| +BASE_NUMERIC_CAST_CASE_DEST_LARGER(true, true, true); |
| +// Dest is unsigned, Source is signed. |
| +BASE_NUMERIC_CAST_CASE_DEST_LARGER(false, true, source >= 0); |
| +// Dest is signed, Source is unsigned. |
| +BASE_NUMERIC_CAST_CASE_DEST_LARGER(true, false, true); |
| + |
| +#undef BASE_NUMERIC_CAST_CASE_SPECIALIZATION |
| +#undef BASE_NUMERIC_CAST_CASE_SAME_SIZE |
| +#undef BASE_NUMERIC_CAST_CASE_SOURCE_LARGER |
| +#undef BASE_NUMERIC_CAST_CASE_DEST_LARGER |
| + |
| + |
| +// The main test for whether the conversion will under or overflow. |
| +template <class Dest, class Source> |
| +inline bool IsValidNumericCast(Source source) { |
| + typedef std::numeric_limits<Source> SourceLimits; |
| + typedef std::numeric_limits<Dest> DestLimits; |
| + COMPILE_ASSERT(SourceLimits::is_specialized, argument_must_be_numeric); |
| + COMPILE_ASSERT(SourceLimits::is_integer, argument_must_be_integral); |
| + COMPILE_ASSERT(DestLimits::is_specialized, result_must_be_numeric); |
| + COMPILE_ASSERT(DestLimits::is_integer, result_must_be_integral); |
| + |
| + return IsValidNumericCastImpl< |
| + sizeof(Dest) == sizeof(Source), |
| + (sizeof(Dest) > sizeof(Source)), |
| + DestLimits::is_signed, |
| + SourceLimits::is_signed>::Test( |
| + source, |
| + DestLimits::min(), |
| + DestLimits::max()); |
| +} |
| + |
| +} // namespace internal |
| + |
| +// numeric_cast<> is analogous to static_cast<> for numeric types, except that |
|
wtc
2013/01/25 15:40:59
The cast is checked_numeric_cast not numeric_cast,
|
| +// it CHECKs that the specified numeric conversion will not overflow or |
| +// underflow. Floating point arguments are not currently allowed (this is |
| +// COMPILE_ASSERTd), though this could be supported if necessary. |
| +template <class Dest, class Source> |
| +inline Dest checked_numeric_cast(Source source) { |
| + CHECK(internal::IsValidNumericCast<Dest>(source)); |
| + return static_cast<Dest>(source); |
| +} |
| + |
| +} // namespace base |
| + |
| +#endif // BASE_SAFE_NUMERICS_H_ |