1 files changed, 98 insertions, 0 deletions
diff --git a/cxx/include/aesxx/error.hpp b/cxx/include/aesxx/error.hpp
new file mode 100644
index 0000000..1f22535
--- /dev/null
+++ b/cxx/include/aesxx/error.hpp
@@ -0,0 +1,98 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ *            See LICENSE.txt for details.
+ */
+
+#pragma once
+
+#include "debug.hpp"
+
+#include <aes/all.h>
+
+#include <boost/config.hpp>
+
+#include <cstdlib>
+#include <cstring>
+
+#include <algorithm>
+#include <functional>
+#include <ostream>
+#include <stdexcept>
+#include <string>
+#include <vector>
+
+namespace aesni
+{
+    class Error : public std::runtime_error
+    {
+    public:
+        Error(const AesNI_ErrorDetails& err_details)
+            : std::runtime_error(format_error_message(err_details))
+        {
+            copy_call_stack(err_details);
+        }
+
+        void for_each_in_call_stack(const std::function<void (void*, const std::string&)>& callback) const
+        {
+            aux::CallStackFormatter formatter;
+            std::for_each(call_stack, call_stack + call_stack_size, [&formatter, &callback] (void* addr)
+            {
+                callback(addr, formatter.format_address(addr));
+            });
+        }
+
+    private:
+        static std::string format_error_message(const AesNI_ErrorDetails& err_details)
+        {
+            std::vector<char> buf;
+            buf.resize(aesni_format_error(&err_details, NULL, 0));
+            aesni_format_error(&err_details, buf.data(), buf.size());
+            return { buf.begin(), buf.end() };
+        }
+
+        void copy_call_stack(const AesNI_ErrorDetails& err_details)
+        {
+            call_stack_size = err_details.call_stack_size;
+            std::memcpy(call_stack, err_details.call_stack, call_stack_size * sizeof(void*));
+        }
+
+        void* call_stack[AESNI_MAX_CALL_STACK_LENGTH];
+        std::size_t call_stack_size;
+    };
+
+    std::ostream& operator<<(std::ostream& os, const Error& e)
+    {
+        os << "AesNI error: " << e.what() << '\n';
+        os << "Call stack:\n";
+        e.for_each_in_call_stack([&os] (void* addr, const std::string& name)
+        {
+            os << '\t' << addr << ' ' << name << '\n';
+        });
+        return os;
+    }
+
+    class ErrorDetailsThrowsInDestructor
+    {
+    public:
+        ErrorDetailsThrowsInDestructor()
+        {
+            aesni_success(get());
+        }
+
+        ~ErrorDetailsThrowsInDestructor() BOOST_NOEXCEPT_IF(false)
+        {
+            if (aesni_is_error(aesni_get_error_code(get())))
+                throw Error(impl);
+        }
+
+        AesNI_ErrorDetails* get() { return &impl; }
+
+        operator AesNI_ErrorDetails*() { return get(); }
+
+    private:
+        AesNI_ErrorDetails impl;
+    };
+}