diff options
| author | Marco Elver <elver@google.com> | 2025-10-28 16:55:29 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-10-28 16:55:29 +0100 |
| commit | 8c8f2df23239914f3276aef02eb89a78373fcaa3 (patch) | |
| tree | 98a0a2eae040b304afc6b84237fcf5ddbe3f70c6 /llvm/lib/Bitcode/Writer/BitcodeWriter.cpp | |
| parent | 624d4f67549c0740e86293dab46b9be9291cdd4d (diff) | |
| download | llvm-8c8f2df23239914f3276aef02eb89a78373fcaa3.zip llvm-8c8f2df23239914f3276aef02eb89a78373fcaa3.tar.gz llvm-8c8f2df23239914f3276aef02eb89a78373fcaa3.tar.bz2 | |
[Clang][CodeGen] Implement code generation for __builtin_infer_alloc_token() (#156842)
Implement code generation for `__builtin_infer_alloc_token()`. The
`AllocToken` pass is now registered to run unconditionally in the
optimization pipeline. This ensures that all instances of the
`llvm.alloc.token.id` intrinsic are lowered to constant token IDs,
regardless of whether `-fsanitize=alloc-token` is enabled. This
guarantees that the builtin always resolves to a token value, providing
a consistent and reliable mechanism for compile-time token querying.
This completes `__builtin_infer_alloc_token(<malloc-args>, ...)` to
allow compile-time querying of the token ID, where the builtin arguments
mirror those normally passed to any allocation function. The argument
expressions are unevaluated operands. For type-based token modes, the
same type inference logic is used as for untyped allocation calls.
For example the ID that is passed to (with `-fsanitize=alloc-token`):
some_malloc(sizeof(Type), ...)
is equivalent to the token ID returned by
__builtin_infer_alloc_token(sizeof(Type), ...)
The builtin provides a mechanism to pass or compare token IDs in code
that needs to be explicitly allocation token-aware (such as inside an
allocator, or through wrapper macros).
A more concrete demonstration of __builtin_infer_alloc_token's use is
enabling type-aware Slab allocations in the Linux kernel:
https://lore.kernel.org/all/20250825154505.1558444-1-elver@google.com/
Notably, any kind of allocation-call rewriting is a poor fit for the
Linux kernel's kmalloc-family functions, which are macros that wrap
(multiple) layers of inline and non-inline wrapper functions. Given the
Linux kernel defines its own allocation APIs, the more explicit builtin
gives the right level of control over where the type inference happens
and the resulting token is passed.
Diffstat (limited to 'llvm/lib/Bitcode/Writer/BitcodeWriter.cpp')
0 files changed, 0 insertions, 0 deletions
