1 /* +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
2 Copyright (c) 2008-2017, Petr Kobalicek
3
4 This software is provided 'as-is', without any express or implied
5 warranty. In no event will the authors be held liable for any damages
6 arising from the use of this software.
7
8 Permission is granted to anyone to use this software for any purpose,
9 including commercial applications, and to alter it and redistribute it
10 freely, subject to the following restrictions:
11
12 1. The origin of this software must not be misrepresented; you must not
13 claim that you wrote the original software. If you use this software
14 in a product, an acknowledgment in the product documentation would be
15 appreciated but is not required.
16 2. Altered source versions must be plainly marked as such, and must not be
17 misrepresented as being the original software.
18 3. This notice may not be removed or altered from any source distribution.
19 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ */
20 #ifndef __PLUMED_asmjit_codeholder_h
21 #define __PLUMED_asmjit_codeholder_h
22 #ifdef __PLUMED_HAS_ASMJIT
23 #pragma GCC diagnostic push
24 #pragma GCC diagnostic ignored "-Wpedantic"
25 // [AsmJit]
26 // Complete x86/x64 JIT and Remote Assembler for C++.
27 //
28 // [License]
29 // Zlib - See LICENSE.md file in the package.
30
31 // [Guard]
32 #ifndef _ASMJIT_BASE_CODEHOLDER_H
33 #define _ASMJIT_BASE_CODEHOLDER_H
34
35 // [Dependencies]
36 #include "./arch.h"
37 #include "./func.h"
38 #include "./logging.h"
39 #include "./operand.h"
40 #include "./simdtypes.h"
41 #include "./utils.h"
42 #include "./zone.h"
43
44 // [Api-Begin]
45 #include "./asmjit_apibegin.h"
46
47 namespace PLMD {
48 namespace asmjit {
49
50 //! \addtogroup asmjit_base
51 //! \{
52
53 // ============================================================================
54 // [Forward Declarations]
55 // ============================================================================
56
57 class Assembler;
58 class CodeEmitter;
59 class CodeHolder;
60
61 // ============================================================================
62 // [asmjit::AlignMode]
63 // ============================================================================
64
65 //! Align mode.
ASMJIT_ENUM(AlignMode)66 ASMJIT_ENUM(AlignMode) {
67 kAlignCode = 0, //!< Align executable code.
68 kAlignData = 1, //!< Align non-executable code.
69 kAlignZero = 2, //!< Align by a sequence of zeros.
70 kAlignCount //!< Count of alignment modes.
71 };
72
73 // ============================================================================
74 // [asmjit::ErrorHandler]
75 // ============================================================================
76
77 //! Error handler can be used to override the default behavior of error handling
78 //! available to all classes that inherit \ref CodeEmitter. See \ref handleError().
79 class ASMJIT_VIRTAPI ErrorHandler {
80 public:
81 // --------------------------------------------------------------------------
82 // [Construction / Destruction]
83 // --------------------------------------------------------------------------
84
85 //! Create a new `ErrorHandler` instance.
86 ASMJIT_API ErrorHandler() noexcept;
87 //! Destroy the `ErrorHandler` instance.
88 ASMJIT_API virtual ~ErrorHandler() noexcept;
89
90 // --------------------------------------------------------------------------
91 // [Handle Error]
92 // --------------------------------------------------------------------------
93
94 //! Error handler (abstract).
95 //!
96 //! Error handler is called after an error happened and before it's propagated
97 //! to the caller. There are multiple ways how the error handler can be used:
98 //!
99 //! 1. Returning `true` or `false` from `handleError()`. If `true` is returned
100 //! it means that the error was reported and AsmJit can continue execution.
101 //! The reported error still be propagated to the caller, but won't put the
102 //! CodeEmitter into an error state (it won't set last-error). However,
103 //! returning `false` means that the error cannot be handled - in such case
104 //! it stores the error, which can be then retrieved by using `getLastError()`.
105 //! Returning `false` is the default behavior when no error handler is present.
106 //! To put the assembler into a non-error state again a `resetLastError()` must
107 //! be called.
108 //!
109 //! 2. Throwing an exception. AsmJit doesn't use exceptions and is completely
110 //! exception-safe, but you can throw exception from your error handler if
111 //! this way is the preferred way of handling errors in your project. Throwing
112 //! an exception acts virtually as returning `true` as AsmJit won't be able
113 //! to store the error because the exception changes execution path.
114 //!
115 //! 3. Using plain old C's `setjmp()` and `longjmp()`. Asmjit always puts
116 //! `CodeEmitter` to a consistent state before calling the `handleError()`
117 //! so `longjmp()` can be used without any issues to cancel the code
118 //! generation if an error occurred. There is no difference between
119 //! exceptions and longjmp() from AsmJit's perspective.
120 virtual bool handleError(Error err, const char* message, CodeEmitter* origin) = 0;
121 };
122
123 // ============================================================================
124 // [asmjit::CodeInfo]
125 // ============================================================================
126
127 //! Basic information about a code (or target). It describes its architecture,
128 //! code generation mode (or optimization level), and base address.
129 class CodeInfo {
130 public:
131 // --------------------------------------------------------------------------
132 // [Construction / Destruction]
133 // --------------------------------------------------------------------------
134
CodeInfo()135 ASMJIT_INLINE CodeInfo() noexcept
136 : _archInfo(),
137 _stackAlignment(0),
138 _cdeclCallConv(CallConv::kIdNone),
139 _stdCallConv(CallConv::kIdNone),
140 _fastCallConv(CallConv::kIdNone),
141 _baseAddress(Globals::kNoBaseAddress) {}
CodeInfo(const CodeInfo & other)142 ASMJIT_INLINE CodeInfo(const CodeInfo& other) noexcept { init(other); }
143
144 explicit ASMJIT_INLINE CodeInfo(uint32_t archType, uint32_t archMode = 0, uint64_t baseAddress = Globals::kNoBaseAddress) noexcept
_archInfo(archType,archMode)145 : _archInfo(archType, archMode),
146 _packedMiscInfo(0),
147 _baseAddress(baseAddress) {}
148
149 // --------------------------------------------------------------------------
150 // [Init / Reset]
151 // --------------------------------------------------------------------------
152
isInitialized()153 ASMJIT_INLINE bool isInitialized() const noexcept {
154 return _archInfo._type != ArchInfo::kTypeNone;
155 }
156
init(const CodeInfo & other)157 ASMJIT_INLINE void init(const CodeInfo& other) noexcept {
158 _archInfo = other._archInfo;
159 _packedMiscInfo = other._packedMiscInfo;
160 _baseAddress = other._baseAddress;
161 }
162
163 ASMJIT_INLINE void init(uint32_t archType, uint32_t archMode = 0, uint64_t baseAddress = Globals::kNoBaseAddress) noexcept {
164 _archInfo.init(archType, archMode);
165 _packedMiscInfo = 0;
166 _baseAddress = baseAddress;
167 }
168
reset()169 ASMJIT_INLINE void reset() noexcept {
170 _archInfo.reset();
171 _stackAlignment = 0;
172 _cdeclCallConv = CallConv::kIdNone;
173 _stdCallConv = CallConv::kIdNone;
174 _fastCallConv = CallConv::kIdNone;
175 _baseAddress = Globals::kNoBaseAddress;
176 }
177
178 // --------------------------------------------------------------------------
179 // [Architecture Information]
180 // --------------------------------------------------------------------------
181
182 //! Get architecture information, see \ref ArchInfo.
getArchInfo()183 ASMJIT_INLINE const ArchInfo& getArchInfo() const noexcept { return _archInfo; }
184
185 //! Get architecture type, see \ref ArchInfo::Type.
getArchType()186 ASMJIT_INLINE uint32_t getArchType() const noexcept { return _archInfo.getType(); }
187 //! Get architecture sub-type, see \ref ArchInfo::SubType.
getArchSubType()188 ASMJIT_INLINE uint32_t getArchSubType() const noexcept { return _archInfo.getSubType(); }
189 //! Get a size of a GP register of the architecture the code is using.
getGpSize()190 ASMJIT_INLINE uint32_t getGpSize() const noexcept { return _archInfo.getGpSize(); }
191 //! Get number of GP registers available of the architecture the code is using.
getGpCount()192 ASMJIT_INLINE uint32_t getGpCount() const noexcept { return _archInfo.getGpCount(); }
193
194 // --------------------------------------------------------------------------
195 // [High-Level Information]
196 // --------------------------------------------------------------------------
197
198 //! Get a natural stack alignment that must be honored (or 0 if not known).
getStackAlignment()199 ASMJIT_INLINE uint32_t getStackAlignment() const noexcept { return _stackAlignment; }
200 //! Set a natural stack alignment that must be honored.
setStackAlignment(uint8_t sa)201 ASMJIT_INLINE void setStackAlignment(uint8_t sa) noexcept { _stackAlignment = static_cast<uint8_t>(sa); }
202
getCdeclCallConv()203 ASMJIT_INLINE uint32_t getCdeclCallConv() const noexcept { return _cdeclCallConv; }
setCdeclCallConv(uint32_t cc)204 ASMJIT_INLINE void setCdeclCallConv(uint32_t cc) noexcept { _cdeclCallConv = static_cast<uint8_t>(cc); }
205
getStdCallConv()206 ASMJIT_INLINE uint32_t getStdCallConv() const noexcept { return _stdCallConv; }
setStdCallConv(uint32_t cc)207 ASMJIT_INLINE void setStdCallConv(uint32_t cc) noexcept { _stdCallConv = static_cast<uint8_t>(cc); }
208
getFastCallConv()209 ASMJIT_INLINE uint32_t getFastCallConv() const noexcept { return _fastCallConv; }
setFastCallConv(uint32_t cc)210 ASMJIT_INLINE void setFastCallConv(uint32_t cc) noexcept { _fastCallConv = static_cast<uint8_t>(cc); }
211
212 // --------------------------------------------------------------------------
213 // [Addressing Information]
214 // --------------------------------------------------------------------------
215
hasBaseAddress()216 ASMJIT_INLINE bool hasBaseAddress() const noexcept { return _baseAddress != Globals::kNoBaseAddress; }
getBaseAddress()217 ASMJIT_INLINE uint64_t getBaseAddress() const noexcept { return _baseAddress; }
setBaseAddress(uint64_t p)218 ASMJIT_INLINE void setBaseAddress(uint64_t p) noexcept { _baseAddress = p; }
resetBaseAddress()219 ASMJIT_INLINE void resetBaseAddress() noexcept { _baseAddress = Globals::kNoBaseAddress; }
220
221 // --------------------------------------------------------------------------
222 // [Operator Overload]
223 // --------------------------------------------------------------------------
224
225 ASMJIT_INLINE CodeInfo& operator=(const CodeInfo& other) noexcept { init(other); return *this; }
226 ASMJIT_INLINE bool operator==(const CodeInfo& other) const noexcept { return ::memcmp(this, &other, sizeof(*this)) == 0; }
227 ASMJIT_INLINE bool operator!=(const CodeInfo& other) const noexcept { return ::memcmp(this, &other, sizeof(*this)) != 0; }
228
229 // --------------------------------------------------------------------------
230 // [Members]
231 // --------------------------------------------------------------------------
232
233 ArchInfo _archInfo; //!< Architecture information.
234
235 union {
236 struct {
237 uint8_t _stackAlignment; //!< Natural stack alignment (ARCH+OS).
238 uint8_t _cdeclCallConv; //!< Default CDECL calling convention.
239 uint8_t _stdCallConv; //!< Default STDCALL calling convention.
240 uint8_t _fastCallConv; //!< Default FASTCALL calling convention.
241 };
242 uint32_t _packedMiscInfo; //!< \internal
243 };
244
245 uint64_t _baseAddress; //!< Base address.
246 };
247
248 // ============================================================================
249 // [asmjit::CodeBuffer]
250 // ============================================================================
251
252 //! Code or data buffer.
253 struct CodeBuffer {
254 // --------------------------------------------------------------------------
255 // [Accessors]
256 // --------------------------------------------------------------------------
257
hasDataCodeBuffer258 ASMJIT_INLINE bool hasData() const noexcept { return _data != nullptr; }
getDataCodeBuffer259 ASMJIT_INLINE uint8_t* getData() noexcept { return _data; }
getDataCodeBuffer260 ASMJIT_INLINE const uint8_t* getData() const noexcept { return _data; }
261
getLengthCodeBuffer262 ASMJIT_INLINE size_t getLength() const noexcept { return _length; }
getCapacityCodeBuffer263 ASMJIT_INLINE size_t getCapacity() const noexcept { return _capacity; }
264
isExternalCodeBuffer265 ASMJIT_INLINE bool isExternal() const noexcept { return _isExternal; }
isFixedSizeCodeBuffer266 ASMJIT_INLINE bool isFixedSize() const noexcept { return _isFixedSize; }
267
268 // --------------------------------------------------------------------------
269 // [Members]
270 // --------------------------------------------------------------------------
271
272 uint8_t* _data; //!< The content of the buffer (data).
273 size_t _length; //!< Number of bytes of `data` used.
274 size_t _capacity; //!< Buffer capacity (in bytes).
275 bool _isExternal; //!< True if this is external buffer.
276 bool _isFixedSize; //!< True if this buffer cannot grow.
277 };
278
279 // ============================================================================
280 // [asmjit::SectionEntry]
281 // ============================================================================
282
283 //! Section entry.
284 class SectionEntry {
285 public:
ASMJIT_ENUM(Id)286 ASMJIT_ENUM(Id) {
287 kInvalidId = 0xFFFFFFFFU //!< Invalid section id.
288 };
289
290 //! Section flags.
ASMJIT_ENUM(Flags)291 ASMJIT_ENUM(Flags) {
292 kFlagExec = 0x00000001U, //!< Executable (.text sections).
293 kFlagConst = 0x00000002U, //!< Read-only (.text and .data sections).
294 kFlagZero = 0x00000004U, //!< Zero initialized by the loader (BSS).
295 kFlagInfo = 0x00000008U, //!< Info / comment flag.
296 kFlagImplicit = 0x80000000U //!< Section created implicitly (can be deleted by the Runtime).
297 };
298
299 // --------------------------------------------------------------------------
300 // [Accessors]
301 // --------------------------------------------------------------------------
302
getId()303 ASMJIT_INLINE uint32_t getId() const noexcept { return _id; }
getName()304 ASMJIT_INLINE const char* getName() const noexcept { return _name; }
305
306 ASMJIT_INLINE void _setDefaultName(
307 char c0 = 0, char c1 = 0, char c2 = 0, char c3 = 0,
308 char c4 = 0, char c5 = 0, char c6 = 0, char c7 = 0) noexcept {
309 _nameAsU32[0] = Utils::pack32_4x8(c0, c1, c2, c3);
310 _nameAsU32[1] = Utils::pack32_4x8(c4, c5, c6, c7);
311 }
312
getFlags()313 ASMJIT_INLINE uint32_t getFlags() const noexcept { return _flags; }
hasFlag(uint32_t flag)314 ASMJIT_INLINE bool hasFlag(uint32_t flag) const noexcept { return (_flags & flag) != 0; }
addFlags(uint32_t flags)315 ASMJIT_INLINE void addFlags(uint32_t flags) noexcept { _flags |= flags; }
clearFlags(uint32_t flags)316 ASMJIT_INLINE void clearFlags(uint32_t flags) noexcept { _flags &= ~flags; }
317
getAlignment()318 ASMJIT_INLINE uint32_t getAlignment() const noexcept { return _alignment; }
setAlignment(uint32_t alignment)319 ASMJIT_INLINE void setAlignment(uint32_t alignment) noexcept { _alignment = alignment; }
320
getPhysicalSize()321 ASMJIT_INLINE size_t getPhysicalSize() const noexcept { return _buffer.getLength(); }
322
getVirtualSize()323 ASMJIT_INLINE size_t getVirtualSize() const noexcept { return _virtualSize; }
setVirtualSize(uint32_t size)324 ASMJIT_INLINE void setVirtualSize(uint32_t size) noexcept { _virtualSize = size; }
325
getBuffer()326 ASMJIT_INLINE CodeBuffer& getBuffer() noexcept { return _buffer; }
getBuffer()327 ASMJIT_INLINE const CodeBuffer& getBuffer() const noexcept { return _buffer; }
328
329 // --------------------------------------------------------------------------
330 // [Members]
331 // --------------------------------------------------------------------------
332
333 uint32_t _id; //!< Section id.
334 uint32_t _flags; //!< Section flags.
335 uint32_t _alignment; //!< Section alignment requirements (0 if no requirements).
336 uint32_t _virtualSize; //!< Virtual size of the section (zero initialized mostly).
337 union {
338 char _name[36]; //!< Section name (max 35 characters, PE allows max 8).
339 uint32_t _nameAsU32[36 / 4]; //!< Section name as `uint32_t[]` (only optimization).
340 };
341 CodeBuffer _buffer; //!< Code or data buffer.
342 };
343
344 // ============================================================================
345 // [asmjit::LabelLink]
346 // ============================================================================
347
348 //! Data structure used to link labels.
349 struct LabelLink {
350 LabelLink* prev; //!< Previous link (single-linked list).
351 uint32_t sectionId; //!< Section id.
352 uint32_t relocId; //!< Relocation id or RelocEntry::kInvalidId.
353 size_t offset; //!< Label offset relative to the start of the section.
354 intptr_t rel; //!< Inlined rel8/rel32.
355 };
356
357 // ============================================================================
358 // [asmjit::LabelEntry]
359 // ============================================================================
360
361 //! Label entry.
362 //!
363 //! Contains the following properties:
364 //! * Label id - This is the only thing that is set to the `Label` operand.
365 //! * Label name - Optional, used mostly to create executables and libraries.
366 //! * Label type - Type of the label, default `Label::kTypeAnonymous`.
367 //! * Label parent id - Derived from many assemblers that allow to define a
368 //! local label that falls under a global label. This allows to define
369 //! many labels of the same name that have different parent (global) label.
370 //! * Offset - offset of the label bound by `Assembler`.
371 //! * Links - single-linked list that contains locations of code that has
372 //! to be patched when the label gets bound. Every use of unbound label
373 //! adds one link to `_links` list.
374 //! * HVal - Hash value of label's name and optionally parentId.
375 //! * HashNext - Hash-table implementation detail.
376 class LabelEntry : public ZoneHashNode {
377 public:
378 // NOTE: Label id is stored in `_customData`, which is provided by ZoneHashNode
379 // to fill a padding that a C++ compiler targeting 64-bit CPU will add to align
380 // the structure to 64-bits.
381
382 //! Get label id.
getId()383 ASMJIT_INLINE uint32_t getId() const noexcept { return _customData; }
384 //! Set label id (internal, used only by \ref CodeHolder).
_setId(uint32_t id)385 ASMJIT_INLINE void _setId(uint32_t id) noexcept { _customData = id; }
386
387 //! Get label type, see \ref Label::Type.
getType()388 ASMJIT_INLINE uint32_t getType() const noexcept { return _type; }
389 //! Get label flags, returns 0 at the moment.
getFlags()390 ASMJIT_INLINE uint32_t getFlags() const noexcept { return _flags; }
391
hasParent()392 ASMJIT_INLINE bool hasParent() const noexcept { return _parentId != 0; }
393 //! Get label's parent id.
getParentId()394 ASMJIT_INLINE uint32_t getParentId() const noexcept { return _parentId; }
395
396 //! Get label's section id where it's bound to (or `SectionEntry::kInvalidId` if it's not bound yet).
getSectionId()397 ASMJIT_INLINE uint32_t getSectionId() const noexcept { return _sectionId; }
398
399 //! Get if the label has name.
hasName()400 ASMJIT_INLINE bool hasName() const noexcept { return !_name.isEmpty(); }
401
402 //! Get the label's name.
403 //!
404 //! NOTE: Local labels will return their local name without their parent
405 //! part, for example ".L1".
getName()406 ASMJIT_INLINE const char* getName() const noexcept { return _name.getData(); }
407
408 //! Get length of label's name.
409 //!
410 //! NOTE: Label name is always null terminated, so you can use `strlen()` to
411 //! get it, however, it's also cached in `LabelEntry`, so if you want to know
412 //! the length the easiest way is to use `LabelEntry::getNameLength()`.
getNameLength()413 ASMJIT_INLINE size_t getNameLength() const noexcept { return _name.getLength(); }
414
415 //! Get if the label is bound.
isBound()416 ASMJIT_INLINE bool isBound() const noexcept { return _sectionId != SectionEntry::kInvalidId; }
417 //! Get the label offset (only useful if the label is bound).
getOffset()418 ASMJIT_INLINE intptr_t getOffset() const noexcept { return _offset; }
419
420 //! Get the hash-value of label's name and its parent label (if any).
421 //!
422 //! Label hash is calculated as `HASH(Name) ^ ParentId`. The hash function
423 //! is implemented in `Utils::hashString()` and `Utils::hashRound()`.
getHVal()424 ASMJIT_INLINE uint32_t getHVal() const noexcept { return _hVal; }
425
426 // ------------------------------------------------------------------------
427 // [Members]
428 // ------------------------------------------------------------------------
429
430 // Let's round the size of `LabelEntry` to 64 bytes (as ZoneHeap has 32
431 // bytes granularity anyway). This gives `_name` the remaining space, which
432 // is roughly 16 bytes on 64-bit and 28 bytes on 32-bit architectures.
433 enum { kNameBytes = 64 - (sizeof(ZoneHashNode) + 16 + sizeof(intptr_t) + sizeof(LabelLink*)) };
434
435 uint8_t _type; //!< Label type, see Label::Type.
436 uint8_t _flags; //!< Must be zero.
437 uint16_t _reserved16; //!< Reserved.
438 uint32_t _parentId; //!< Label parent id or zero.
439 uint32_t _sectionId; //!< Section id or `SectionEntry::kInvalidId`.
440 uint32_t _reserved32; //!< Reserved.
441 intptr_t _offset; //!< Label offset.
442 LabelLink* _links; //!< Label links.
443 SmallString<kNameBytes> _name; //!< Label name.
444 };
445
446 // ============================================================================
447 // [asmjit::RelocEntry]
448 // ============================================================================
449
450 //! Relocation entry.
451 struct RelocEntry {
ASMJIT_ENUMRelocEntry452 ASMJIT_ENUM(Id) {
453 kInvalidId = 0xFFFFFFFFU //!< Invalid relocation id.
454 };
455
456 //! Relocation type.
ASMJIT_ENUMRelocEntry457 ASMJIT_ENUM(Type) {
458 kTypeNone = 0, //!< Deleted entry (no relocation).
459 kTypeAbsToAbs = 1, //!< Relocate absolute to absolute.
460 kTypeRelToAbs = 2, //!< Relocate relative to absolute.
461 kTypeAbsToRel = 3, //!< Relocate absolute to relative.
462 kTypeTrampoline = 4 //!< Relocate absolute to relative or use trampoline.
463 };
464
465 // ------------------------------------------------------------------------
466 // [Accessors]
467 // ------------------------------------------------------------------------
468
getIdRelocEntry469 ASMJIT_INLINE uint32_t getId() const noexcept { return _id; }
470
getTypeRelocEntry471 ASMJIT_INLINE uint32_t getType() const noexcept { return _type; }
getSizeRelocEntry472 ASMJIT_INLINE uint32_t getSize() const noexcept { return _size; }
473
getSourceSectionIdRelocEntry474 ASMJIT_INLINE uint32_t getSourceSectionId() const noexcept { return _sourceSectionId; }
getTargetSectionIdRelocEntry475 ASMJIT_INLINE uint32_t getTargetSectionId() const noexcept { return _targetSectionId; }
476
getSourceOffsetRelocEntry477 ASMJIT_INLINE uint64_t getSourceOffset() const noexcept { return _sourceOffset; }
getDataRelocEntry478 ASMJIT_INLINE uint64_t getData() const noexcept { return _data; }
479
480 // ------------------------------------------------------------------------
481 // [Members]
482 // ------------------------------------------------------------------------
483
484 uint32_t _id; //!< Relocation id.
485 uint8_t _type; //!< Type of the relocation.
486 uint8_t _size; //!< Size of the relocation (1, 2, 4 or 8 bytes).
487 uint8_t _reserved[2]; //!< Reserved.
488 uint32_t _sourceSectionId; //!< Source section id.
489 uint32_t _targetSectionId; //!< Destination section id.
490 uint64_t _sourceOffset; //!< Source offset (relative to start of the section).
491 uint64_t _data; //!< Relocation data (target offset, target address, etc).
492 };
493
494 // ============================================================================
495 // [asmjit::CodeHolder]
496 // ============================================================================
497
498 //! Contains basic information about the target architecture plus its settings,
499 //! and holds code & data (including sections, labels, and relocation information).
500 //! CodeHolder can store both binary and intermediate representation of assembly,
501 //! which can be generated by \ref Assembler and/or \ref CodeBuilder.
502 //!
503 //! NOTE: CodeHolder has ability to attach an \ref ErrorHandler, however, this
504 //! error handler is not triggered by CodeHolder itself, it's only used by the
505 //! attached code generators.
506 class CodeHolder {
507 public:
508 ASMJIT_NONCOPYABLE(CodeHolder)
509
510 // --------------------------------------------------------------------------
511 // [Construction / Destruction]
512 // --------------------------------------------------------------------------
513
514 //! Create an uninitialized CodeHolder (you must init() it before it can be used).
515 ASMJIT_API CodeHolder() noexcept;
516 //! Destroy the CodeHolder.
517 ASMJIT_API ~CodeHolder() noexcept;
518
519 // --------------------------------------------------------------------------
520 // [Init / Reset]
521 // --------------------------------------------------------------------------
522
isInitialized()523 ASMJIT_INLINE bool isInitialized() const noexcept { return _codeInfo.isInitialized(); }
524
525 //! Initialize to CodeHolder to hold code described by `codeInfo`.
526 ASMJIT_API Error init(const CodeInfo& info) noexcept;
527 //! Detach all code-generators attached and reset the \ref CodeHolder.
528 ASMJIT_API void reset(bool releaseMemory = false) noexcept;
529
530 // --------------------------------------------------------------------------
531 // [Attach / Detach]
532 // --------------------------------------------------------------------------
533
534 //! Attach a \ref CodeEmitter to this \ref CodeHolder.
535 ASMJIT_API Error attach(CodeEmitter* emitter) noexcept;
536 //! Detach a \ref CodeEmitter from this \ref CodeHolder.
537 ASMJIT_API Error detach(CodeEmitter* emitter) noexcept;
538
539 // --------------------------------------------------------------------------
540 // [Sync]
541 // --------------------------------------------------------------------------
542
543 //! Synchronize all states of all `CodeEmitter`s associated with the CodeHolder.
544 //! This is required as some code generators don't sync every time they do
545 //! something - for example \ref Assembler generally syncs when it needs to
546 //! reallocate the \ref CodeBuffer, but not each time it encodes instruction
547 //! or directive.
548 ASMJIT_API void sync() noexcept;
549
550 // --------------------------------------------------------------------------
551 // [Code-Information]
552 // --------------------------------------------------------------------------
553
554 //! Get code/target information, see \ref CodeInfo.
getCodeInfo()555 ASMJIT_INLINE const CodeInfo& getCodeInfo() const noexcept { return _codeInfo; }
556 //! Get architecture information, see \ref ArchInfo.
getArchInfo()557 ASMJIT_INLINE const ArchInfo& getArchInfo() const noexcept { return _codeInfo.getArchInfo(); }
558
559 //! Get the target's architecture type.
getArchType()560 ASMJIT_INLINE uint32_t getArchType() const noexcept { return getArchInfo().getType(); }
561 //! Get the target's architecture sub-type.
getArchSubType()562 ASMJIT_INLINE uint32_t getArchSubType() const noexcept { return getArchInfo().getSubType(); }
563
564 //! Get if a static base-address is set.
hasBaseAddress()565 ASMJIT_INLINE bool hasBaseAddress() const noexcept { return _codeInfo.hasBaseAddress(); }
566 //! Get a static base-address (uint64_t).
getBaseAddress()567 ASMJIT_INLINE uint64_t getBaseAddress() const noexcept { return _codeInfo.getBaseAddress(); }
568
569 // --------------------------------------------------------------------------
570 // [Global Information]
571 // --------------------------------------------------------------------------
572
573 //! Get global hints, internally propagated to all `CodeEmitter`s attached.
getGlobalHints()574 ASMJIT_INLINE uint32_t getGlobalHints() const noexcept { return _globalHints; }
575 //! Get global options, internally propagated to all `CodeEmitter`s attached.
getGlobalOptions()576 ASMJIT_INLINE uint32_t getGlobalOptions() const noexcept { return _globalOptions; }
577
578 // --------------------------------------------------------------------------
579 // [Result Information]
580 // --------------------------------------------------------------------------
581
582 //! Get the size code & data of all sections.
583 ASMJIT_API size_t getCodeSize() const noexcept;
584
585 //! Get size of all possible trampolines.
586 //!
587 //! Trampolines are needed to successfully generate relative jumps to absolute
588 //! addresses. This value is only non-zero if jmp of call instructions were
589 //! used with immediate operand (this means jumping or calling an absolute
590 //! address directly).
getTrampolinesSize()591 ASMJIT_INLINE size_t getTrampolinesSize() const noexcept { return _trampolinesSize; }
592
593 // --------------------------------------------------------------------------
594 // [Logging & Error Handling]
595 // --------------------------------------------------------------------------
596
597 #if !defined(ASMJIT_DISABLE_LOGGING)
598 //! Get if a logger attached.
hasLogger()599 ASMJIT_INLINE bool hasLogger() const noexcept { return _logger != nullptr; }
600 //! Get the attached logger.
getLogger()601 ASMJIT_INLINE Logger* getLogger() const noexcept { return _logger; }
602 //! Attach a `logger` to CodeHolder and propagate it to all attached `CodeEmitter`s.
603 ASMJIT_API void setLogger(Logger* logger) noexcept;
604 //! Reset the logger (does nothing if not attached).
resetLogger()605 ASMJIT_INLINE void resetLogger() noexcept { setLogger(nullptr); }
606 #endif // !ASMJIT_DISABLE_LOGGING
607
608 //! Get if error-handler is attached.
hasErrorHandler()609 ASMJIT_INLINE bool hasErrorHandler() const noexcept { return _errorHandler != nullptr; }
610 //! Get the error-handler.
getErrorHandler()611 ASMJIT_INLINE ErrorHandler* getErrorHandler() const noexcept { return _errorHandler; }
612 //! Set the error handler, will affect all attached `CodeEmitter`s.
613 ASMJIT_API Error setErrorHandler(ErrorHandler* handler) noexcept;
614 //! Reset the error handler (does nothing if not attached).
resetErrorHandler()615 ASMJIT_INLINE void resetErrorHandler() noexcept { setErrorHandler(nullptr); }
616
617 // --------------------------------------------------------------------------
618 // [Sections]
619 // --------------------------------------------------------------------------
620
621 //! Get array of `SectionEntry*` records.
getSections()622 ASMJIT_INLINE const ZoneVector<SectionEntry*>& getSections() const noexcept { return _sections; }
623
624 //! Get a section entry of the given index.
getSectionEntry(size_t index)625 ASMJIT_INLINE SectionEntry* getSectionEntry(size_t index) const noexcept { return _sections[index]; }
626
627 ASMJIT_API Error growBuffer(CodeBuffer* cb, size_t n) noexcept;
628 ASMJIT_API Error reserveBuffer(CodeBuffer* cb, size_t n) noexcept;
629
630 // --------------------------------------------------------------------------
631 // [Labels & Symbols]
632 // --------------------------------------------------------------------------
633
634 //! Create a new anonymous label and return its id in `idOut`.
635 //!
636 //! Returns `Error`, does not report error to \ref ErrorHandler.
637 ASMJIT_API Error newLabelId(uint32_t& idOut) noexcept;
638
639 //! Create a new named label label-type `type`.
640 //!
641 //! Returns `Error`, does not report error to \ref ErrorHandler.
642 ASMJIT_API Error newNamedLabelId(uint32_t& idOut, const char* name, size_t nameLength, uint32_t type, uint32_t parentId) noexcept;
643
644 //! Get a label id by name.
645 ASMJIT_API uint32_t getLabelIdByName(const char* name, size_t nameLength = Globals::kInvalidIndex, uint32_t parentId = 0) noexcept;
646
647 //! Create a new label-link used to store information about yet unbound labels.
648 //!
649 //! Returns `null` if the allocation failed.
650 ASMJIT_API LabelLink* newLabelLink(LabelEntry* le, uint32_t sectionId, size_t offset, intptr_t rel) noexcept;
651
652 //! Get array of `LabelEntry*` records.
getLabelEntries()653 ASMJIT_INLINE const ZoneVector<LabelEntry*>& getLabelEntries() const noexcept { return _labels; }
654
655 //! Get number of labels created.
getLabelsCount()656 ASMJIT_INLINE size_t getLabelsCount() const noexcept { return _labels.getLength(); }
657
658 //! Get number of label references, which are unresolved at the moment.
getUnresolvedLabelsCount()659 ASMJIT_INLINE size_t getUnresolvedLabelsCount() const noexcept { return _unresolvedLabelsCount; }
660
661 //! Get if the `label` is valid (i.e. created by `newLabelId()`).
isLabelValid(const Label & label)662 ASMJIT_INLINE bool isLabelValid(const Label& label) const noexcept {
663 return isLabelValid(label.getId());
664 }
665 //! Get if the label having `id` is valid (i.e. created by `newLabelId()`).
isLabelValid(uint32_t labelId)666 ASMJIT_INLINE bool isLabelValid(uint32_t labelId) const noexcept {
667 size_t index = Operand::unpackId(labelId);
668 return index < _labels.getLength();
669 }
670
671 //! Get if the `label` is already bound.
672 //!
673 //! Returns `false` if the `label` is not valid.
isLabelBound(const Label & label)674 ASMJIT_INLINE bool isLabelBound(const Label& label) const noexcept {
675 return isLabelBound(label.getId());
676 }
677 //! \overload
isLabelBound(uint32_t id)678 ASMJIT_INLINE bool isLabelBound(uint32_t id) const noexcept {
679 size_t index = Operand::unpackId(id);
680 return index < _labels.getLength() && _labels[index]->isBound();
681 }
682
683 //! Get a `label` offset or -1 if the label is not yet bound.
getLabelOffset(const Label & label)684 ASMJIT_INLINE intptr_t getLabelOffset(const Label& label) const noexcept {
685 return getLabelOffset(label.getId());
686 }
687 //! \overload
getLabelOffset(uint32_t id)688 ASMJIT_INLINE intptr_t getLabelOffset(uint32_t id) const noexcept {
689 ASMJIT_ASSERT(isLabelValid(id));
690 return _labels[Operand::unpackId(id)]->getOffset();
691 }
692
693 //! Get information about the given `label`.
getLabelEntry(const Label & label)694 ASMJIT_INLINE LabelEntry* getLabelEntry(const Label& label) const noexcept {
695 return getLabelEntry(label.getId());
696 }
697 //! Get information about a label having the given `id`.
getLabelEntry(uint32_t id)698 ASMJIT_INLINE LabelEntry* getLabelEntry(uint32_t id) const noexcept {
699 size_t index = static_cast<size_t>(Operand::unpackId(id));
700 return index < _labels.getLength() ? _labels[index] : static_cast<LabelEntry*>(nullptr);
701 }
702
703 // --------------------------------------------------------------------------
704 // [Relocations]
705 // --------------------------------------------------------------------------
706
707 //! Create a new relocation entry of type `type` and size `size`.
708 //!
709 //! Additional fields can be set after the relocation entry was created.
710 ASMJIT_API Error newRelocEntry(RelocEntry** dst, uint32_t type, uint32_t size) noexcept;
711
712 //! Get if the code contains relocations.
hasRelocations()713 ASMJIT_INLINE bool hasRelocations() const noexcept { return !_relocations.isEmpty(); }
714 //! Get array of `RelocEntry*` records.
getRelocEntries()715 ASMJIT_INLINE const ZoneVector<RelocEntry*>& getRelocEntries() const noexcept { return _relocations; }
716
getRelocEntry(uint32_t id)717 ASMJIT_INLINE RelocEntry* getRelocEntry(uint32_t id) const noexcept { return _relocations[id]; }
718
719 //! Relocate the code to `baseAddress` and copy it to `dst`.
720 //!
721 //! \param dst Contains the location where the relocated code should be
722 //! copied. The pointer can be address returned by virtual memory allocator
723 //! or any other address that has sufficient space.
724 //!
725 //! \param baseAddress Base address used for relocation. `JitRuntime` always
726 //! sets the `baseAddress` to be the same as `dst`.
727 //!
728 //! \return The number bytes actually used. If the code emitter reserved
729 //! space for possible trampolines, but didn't use it, the number of bytes
730 //! used can actually be less than the expected worst case. Virtual memory
731 //! allocator can shrink the memory it allocated initially.
732 //!
733 //! A given buffer will be overwritten, to get the number of bytes required,
734 //! use `getCodeSize()`.
735 ASMJIT_API size_t relocate(void* dst, uint64_t baseAddress = Globals::kNoBaseAddress) const noexcept;
736
737 // --------------------------------------------------------------------------
738 // [Members]
739 // --------------------------------------------------------------------------
740
741 CodeInfo _codeInfo; //!< Basic information about the code (architecture and other info).
742
743 uint32_t _globalHints; //!< Global hints, propagated to all `CodeEmitter`s.
744 uint32_t _globalOptions; //!< Global options, propagated to all `CodeEmitter`s.
745
746 CodeEmitter* _emitters; //!< Linked-list of all attached `CodeEmitter`s.
747 Assembler* _cgAsm; //!< Attached \ref Assembler (only one at a time).
748
749 Logger* _logger; //!< Attached \ref Logger, used by all consumers.
750 ErrorHandler* _errorHandler; //!< Attached \ref ErrorHandler.
751
752 uint32_t _unresolvedLabelsCount; //!< Count of label references which were not resolved.
753 uint32_t _trampolinesSize; //!< Size of all possible trampolines.
754
755 Zone _baseZone; //!< Base zone (used to allocate core structures).
756 Zone _dataZone; //!< Data zone (used to allocate extra data like label names).
757 ZoneHeap _baseHeap; //!< Zone allocator, used to manage internal containers.
758
759 ZoneVector<SectionEntry*> _sections; //!< Section entries.
760 ZoneVector<LabelEntry*> _labels; //!< Label entries (each label is stored here).
761 ZoneVector<RelocEntry*> _relocations; //!< Relocation entries.
762 ZoneHash<LabelEntry> _namedLabels; //!< Label name -> LabelEntry (only named labels).
763 };
764
765 //! \}
766
767 } // asmjit namespace
768 } // namespace PLMD
769
770 // [Api-End]
771 #include "./asmjit_apiend.h"
772
773 // [Guard]
774 #endif // _ASMJIT_BASE_CODEHOLDER_H
775 #pragma GCC diagnostic pop
776 #endif // __PLUMED_HAS_ASMJIT
777 #endif
778