1 files changed, 29 insertions, 0 deletions
diff --git a/tests/core_tests/integer_overflow.h b/tests/core_tests/integer_overflow.h
new file mode 100644
index 000000000..0e00ada32
--- /dev/null
+++ b/tests/core_tests/integer_overflow.h
@@ -0,0 +1,29 @@
+// Copyright (c) 2012-2013 The Cryptonote developers
+// Distributed under the MIT/X11 software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#pragma once 
+#include "chaingen.h"
+
+struct gen_uint_overflow_base : public test_chain_unit_base
+{
+  gen_uint_overflow_base();
+
+  bool check_tx_verification_context(const cryptonote::tx_verification_context& tvc, bool tx_added, size_t event_idx, const cryptonote::transaction& tx);
+  bool check_block_verification_context(const cryptonote::block_verification_context& bvc, size_t event_idx, const cryptonote::block& block);
+
+  bool mark_last_valid_block(cryptonote::core& c, size_t ev_index, const std::vector<test_event_entry>& events);
+
+private:
+  size_t m_last_valid_block_event_idx;
+};
+
+struct gen_uint_overflow_1 : public gen_uint_overflow_base
+{
+  bool generate(std::vector<test_event_entry>& events) const;
+};
+
+struct gen_uint_overflow_2 : public gen_uint_overflow_base
+{
+  bool generate(std::vector<test_event_entry>& events) const;
+};