diff options
Diffstat (limited to 'src/utils')
-rw-r--r-- | src/utils/info.txt | 1 | ||||
-rw-r--r-- | src/utils/safeint.h | 39 |
2 files changed, 40 insertions, 0 deletions
diff --git a/src/utils/info.txt b/src/utils/info.txt index 57b6a2740..bbca1985c 100644 --- a/src/utils/info.txt +++ b/src/utils/info.txt @@ -20,6 +20,7 @@ ct_utils.h mlock.h prefetch.h rounding.h +safeint.h stl_util.h xor_buf.h </header:internal> diff --git a/src/utils/safeint.h b/src/utils/safeint.h new file mode 100644 index 000000000..e0bd66232 --- /dev/null +++ b/src/utils/safeint.h @@ -0,0 +1,39 @@ +/* +* Safe(r) Integer Handling +* (C) 2016 Jack Lloyd +* +* Botan is released under the Simplified BSD License (see license.txt) +*/ + +#ifndef BOTAN_UTILS_SAFE_INT_H__ +#define BOTAN_UTILS_SAFE_INT_H__ + +#include <botan/exceptn.h> +#include <string> + +namespace Botan { + +class Integer_Overflow_Detected : public Exception + { + public: + Integer_Overflow_Detected(const std::string& file, int line) : + Exception("Integer overflow detected at " + file + ":" + std::to_string(line)) + {} + }; + +inline size_t checked_add(size_t x, size_t y, const char* file, int line) + { + // TODO: use __builtin_x_overflow on GCC and Clang + size_t z = x + y; + if(z < x) + { + throw Integer_Overflow_Detected(file, line); + } + return z; + } + +#define BOTAN_CHECKED_ADD(x,y) checked_add(x,y,__FILE__,__LINE__) + +} + +#endif |