libTriton version 1.0 build 1592
Loading...
Searching...
No Matches
callbacks.hpp
Go to the documentation of this file.
1
2/*
3** Copyright (C) - Triton
4**
5** This program is under the terms of the Apache License 2.0.
6*/
7
8#ifndef TRITON_CALLBACKS_H
9#define TRITON_CALLBACKS_H
10
11#include <atomic>
12#include <list>
13
14#include <triton/ast.hpp>
15#include <triton/callbacksEnums.hpp>
16#include <triton/comparableFunctor.hpp>
17#include <triton/dllexport.hpp>
18#include <triton/memoryAccess.hpp>
19#include <triton/register.hpp>
20#include <triton/tritonTypes.hpp>
21
22
23
25namespace triton {
31 class Context;
32
34 namespace callbacks {
46 using getConcreteMemoryValueCallback = ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&)>;
47
53 using getConcreteRegisterValueCallback = ComparableFunctor<void(triton::Context&, const triton::arch::Register&)>;
54
60 using setConcreteMemoryValueCallback = ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&, const triton::uint512& value)>;
61
67 using setConcreteRegisterValueCallback = ComparableFunctor<void(triton::Context&, const triton::arch::Register&, const triton::uint512& value)>;
68
75 using symbolicSimplificationCallback = ComparableFunctor<triton::ast::SharedAbstractNode(triton::Context&, const triton::ast::SharedAbstractNode&)>;
76
78
79 class Callbacks {
80 private:
82 triton::Context& ctx;
83
85 std::atomic<bool> mget;
86
88 std::atomic<bool> mload;
89
91 std::atomic<bool> mput;
92
94 std::atomic<bool> mstore;
95
97 std::atomic<bool> defined;
98
99 protected:
101 std::list<triton::callbacks::getConcreteMemoryValueCallback> getConcreteMemoryValueCallbacks;
102
104 std::list<triton::callbacks::getConcreteRegisterValueCallback> getConcreteRegisterValueCallbacks;
105
107 std::list<triton::callbacks::setConcreteMemoryValueCallback> setConcreteMemoryValueCallbacks;
108
110 std::list<triton::callbacks::setConcreteRegisterValueCallback> setConcreteRegisterValueCallbacks;
111
113 std::list<triton::callbacks::symbolicSimplificationCallback> symbolicSimplificationCallbacks;
114
116 triton::usize countCallbacks(void) const;
117
119 template <typename T> void removeSingleCallback(std::list<T>& container, T cb);
120
121 public:
123 TRITON_EXPORT Callbacks(triton::Context& ctx);
124
126 TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&)> cb);
127
129 TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&)> cb);
130
132 TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&, const triton::uint512& value)> cb);
133
135 TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&, const triton::uint512& value)> cb);
136
138 TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor<triton::ast::SharedAbstractNode(triton::Context&, const triton::ast::SharedAbstractNode&)> cb);
139
141 TRITON_EXPORT void clearCallbacks(void);
142
144 TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&)> cb);
145
147 TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&)> cb);
148
150 TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&, const triton::uint512& value)> cb);
151
153 TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&, const triton::uint512& value)> cb);
154
156 TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<triton::ast::SharedAbstractNode(triton::Context&, const triton::ast::SharedAbstractNode&)> cb);
157
159 TRITON_EXPORT triton::ast::SharedAbstractNode processCallbacks(triton::callbacks::callback_e kind, triton::ast::SharedAbstractNode node);
160
162 TRITON_EXPORT void processCallbacks(triton::callbacks::callback_e kind, const triton::arch::MemoryAccess& mem);
163
165 TRITON_EXPORT void processCallbacks(triton::callbacks::callback_e kind, const triton::arch::MemoryAccess& mem, const triton::uint512& value);
166
168 TRITON_EXPORT void processCallbacks(triton::callbacks::callback_e kind, const triton::arch::Register& reg);
169
171 TRITON_EXPORT void processCallbacks(triton::callbacks::callback_e kind, const triton::arch::Register& reg, const triton::uint512& value);
172
174 TRITON_EXPORT bool isDefined(triton::callbacks::callback_e kind) const;
175
177 TRITON_EXPORT bool isDefined(void) const;
178 };
179
181 };
183};
184
185#endif /* TRITON_CALLBACKS_H */
math::wide_integer::uintwide_t
Definition uintwide_t.h:1274
triton::Context
This is the main Triton Context class.
Definition context.hpp:45
triton::arch::MemoryAccess
This class is used to represent a memory access.
Definition memoryAccess.hpp:40
triton::arch::Register
This class is used when an instruction has a register operand.
Definition register.hpp:44
triton::callbacks::Callbacks
The callbacks class.
Definition callbacks.hpp:79
triton::callbacks::Callbacks::removeCallback
TRITON_EXPORT void removeCallback(triton::callbacks::callback_e kind, ComparableFunctor< void(triton::Context &, const triton::arch::MemoryAccess &)> cb)
Deletes a GET_CONCRETE_MEMORY_VALUE callback.
Definition callbacks.cpp:113
triton::callbacks::Callbacks::symbolicSimplificationCallbacks
std::list< triton::callbacks::symbolicSimplificationCallback > symbolicSimplificationCallbacks
[c++] Callbacks for all symbolic simplifications.
Definition callbacks.hpp:113
triton::callbacks::Callbacks::clearCallbacks
TRITON_EXPORT void clearCallbacks(void)
Clears recorded callbacks.
Definition callbacks.cpp:91
triton::callbacks::Callbacks::addCallback
TRITON_EXPORT void addCallback(triton::callbacks::callback_e kind, ComparableFunctor< void(triton::Context &, const triton::arch::MemoryAccess &)> cb)
Adds a GET_CONCRETE_MEMORY_VALUE callback.
Definition callbacks.cpp:26
triton::callbacks::Callbacks::setConcreteMemoryValueCallbacks
std::list< triton::callbacks::setConcreteMemoryValueCallback > setConcreteMemoryValueCallbacks
[c++] Callbacks for all concrete memory needs (STORE).
Definition callbacks.hpp:107
triton::callbacks::Callbacks::countCallbacks
triton::usize countCallbacks(void) const
Returns the number of callbacks recorded.
Definition callbacks.cpp:306
triton::callbacks::Callbacks::getConcreteMemoryValueCallbacks
std::list< triton::callbacks::getConcreteMemoryValueCallback > getConcreteMemoryValueCallbacks
[c++] Callbacks for all concrete memory needs (LOAD).
Definition callbacks.hpp:101
triton::callbacks::Callbacks::setConcreteRegisterValueCallbacks
std::list< triton::callbacks::setConcreteRegisterValueCallback > setConcreteRegisterValueCallbacks
[c++] Callbacks for all concrete register needs (PUT).
Definition callbacks.hpp:110
triton::callbacks::Callbacks::Callbacks
TRITON_EXPORT Callbacks(triton::Context &ctx)
Constructor.
Definition callbacks.cpp:17
triton::callbacks::Callbacks::isDefined
TRITON_EXPORT bool isDefined(void) const
Returns true if at least one callback is defined.
Definition callbacks.cpp:333
triton::callbacks::Callbacks::removeSingleCallback
void removeSingleCallback(std::list< T > &container, T cb)
Trys to find and remove the callback, raises an exception if not able.
Definition callbacks.cpp:102
triton::callbacks::Callbacks::getConcreteRegisterValueCallbacks
std::list< triton::callbacks::getConcreteRegisterValueCallback > getConcreteRegisterValueCallbacks
[c++] Callbacks for all concrete register needs (GET).
Definition callbacks.hpp:104
triton::callbacks::Callbacks::processCallbacks
TRITON_EXPORT triton::ast::SharedAbstractNode processCallbacks(triton::callbacks::callback_e kind, triton::ast::SharedAbstractNode node)
Processes callbacks according to the kind and the C++ polymorphism.
Definition callbacks.cpp:193
triton::ast::SharedAbstractNode
std::shared_ptr< triton::ast::AbstractNode > SharedAbstractNode
Shared Abstract Node.
Definition ast.hpp:59
triton::callbacks::callback_e
callback_e
Definition callbacksEnums.hpp:29
triton::usize
std::size_t usize
unsigned MAX_INT 32 or 64 bits according to the CPU.
Definition tritonTypes.hpp:106
triton
The Triton namespace.
Definition architecture.cpp:27
Generated by doxygen 1.10.0