1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
// Copyright (c) 2015 Egor Tensin <Egor.Tensin@gmail.com>
// This file is part of the "AES tools" project.
// For details, see https://github.com/egor-tensin/aes-tools.
// Distributed under the MIT License.
#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 aes
{
class Error : public std::runtime_error
{
public:
Error(const AES_ErrorDetails& err_details)
: std::runtime_error{format_error_message(err_details)}
{
fill_call_stack(err_details);
}
void for_each_addr(
const std::function<void (const void*, const std::string&)>& callback) const
{
aux::CallStackFormatter formatter;
std::for_each(
call_stack,
call_stack + call_stack_len,
[&formatter, &callback] (const void* addr)
{
callback(addr, formatter.format_address(addr));
});
}
private:
static std::string format_error_message(const AES_ErrorDetails& err_details)
{
std::vector<char> buf;
buf.resize(aes_format_error(&err_details, NULL, 0));
aes_format_error(&err_details, buf.data(), buf.size());
return {buf.begin(), buf.end()};
}
void fill_call_stack(const AES_ErrorDetails& err_details)
{
call_stack_len = err_details.call_stack_len;
if (call_stack_len > AES_MAX_CALL_STACK_LENGTH)
call_stack_len = AES_MAX_CALL_STACK_LENGTH;
std::memcpy(call_stack, err_details.call_stack, call_stack_len * sizeof(const void*));
}
std::size_t call_stack_len = 0;
const void* call_stack[AES_MAX_CALL_STACK_LENGTH] = {nullptr};
};
inline std::ostream& operator<<(std::ostream& os, const Error& error)
{
os << "AES error: " << error.what() << '\n';
os << "Call stack:\n";
error.for_each_addr([&os] (const void* addr, const std::string& name)
{
os << '\t' << addr << ' ' << name << '\n';
});
return os;
}
class ErrorDetailsThrowsInDestructor
{
public:
ErrorDetailsThrowsInDestructor()
{
aes_success(get());
}
~ErrorDetailsThrowsInDestructor() BOOST_NOEXCEPT_IF(false)
{
if (aes_is_error(aes_get_error_code(get())))
throw Error(impl);
}
AES_ErrorDetails* get() { return &impl; }
operator AES_ErrorDetails*() { return get(); }
private:
AES_ErrorDetails impl;
};
}
|
