Skip to content

[WIP] 6 - Propagate discriminator logic through remaining get_fn_ptr calls sites#159084

Draft
jchlanda wants to merge 7 commits into
rust-lang:mainfrom
jchlanda:jakub/pac_ty_disc_PR_6
Draft

[WIP] 6 - Propagate discriminator logic through remaining get_fn_ptr calls sites#159084
jchlanda wants to merge 7 commits into
rust-lang:mainfrom
jchlanda:jakub/pac_ty_disc_PR_6

Conversation

@jchlanda

@jchlanda jchlanda commented Jul 10, 2026

Copy link
Copy Markdown
Contributor

Fill in function pointer type discriminators logic across remaining get_fn_addr call sites and explicitly avoid applying it where discrimination is not meaningful.

Some uses of get_fn_addr are intentionally left unsigned, including the EH personality function, entry wrappers, and compiler-generated Rust ABI shims.


This is part 6 of a sequence of 8 PRs, that together aim to bring function pointer type discrimination support:

  1. Encoder and hash
  2. FnAbi, llvm.ptrauth.resign and Session API change
  3. FPTR_TYPE_DISCR in ABI Version
  4. Static allocs
  5. Transmutes
  6. Propagate discriminator logic through remaining get_fn_ptr calls sites
  7. Minicore updates to support fn ptr type discriminator tests
  8. Fn ptr type discrimination tests

Useful links:

@rustbot rustbot added A-LLVM Area: Code generation parts specific to LLVM. Both correctness bugs and optimization-related issues. S-waiting-on-author Status: This is awaiting some action (such as code changes or more information) from the author. T-compiler Relevant to the compiler team, which will review and decide on the PR/issue. labels Jul 10, 2026
@rust-bors

This comment has been minimized.

@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch 2 times, most recently from 6f06f96 to 17ad6d3 Compare July 14, 2026 07:29
@rust-log-analyzer

This comment has been minimized.

@rust-bors

This comment has been minimized.

@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch from 17ad6d3 to 9a55906 Compare July 14, 2026 08:54
@rust-log-analyzer

This comment has been minimized.

@rust-bors

This comment has been minimized.

@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch from 9a55906 to b331394 Compare July 17, 2026 07:05
@rust-log-analyzer

This comment has been minimized.

@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch from b331394 to d42d5cf Compare July 17, 2026 08:24
This patch implements Rust's equivalent of Clang's function pointer type
discriminator computation used in pointer authentication. Compatibility
with Clang is a primary goal. The discriminator produced for a given
external "C" function type must match the value computed by Clang so
that function pointers can be exchanged safely between Rust and C code
while preserving pointer authentication semantics.

The implementation mirrors Clang's behavior in
`ASTContext::encodeTypeForFunctionPointerAuth`, ensuring that identical
C-compatible function types produce identical discriminators. See:
<https://clang.llvm.org/doxygen/ASTContext_8cpp.html#abb1375e068e807917527842d05cadea3>.
@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch from d42d5cf to aa26645 Compare July 17, 2026 09:49
// the Function object (via LLVM's `setPersonalityFn`) and consumed only by
// exception handling metadata generation (landing pads / unwind tables).
// LLVM never loads or invokes the personality via a function pointer value;
// it is not part of the program's call graph or data flow.

@kovdan01 kovdan01 Jul 20, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you please refer me to a test covering this? BTW, LLVM does sign personality value with a special signing schema late in backend when it's emitted - see llvm/llvm-project#119361.

I suppose that no rust-side handling special is needed here, I'm just wondering if this works as expected end-to-end. Like, when compiling to asm/obj (not just IR), do we expect to end up with a signed personality pointer (with discriminator 0x7EAD) in .data.DW.ref.__gxx_personality_v0 section or smth like this?

View changes since the review

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is slightly different. This code makes sure that we don't generate pathological IR which contains function signatures with eh personality wrapped in ptrauth themselves. Something looking like:

define void @func_that_can_panic(ptr align 8 %arg) unnamed_addr #1 personality ptrauth (ptr @rust_eh_personality, i32 0) {
...

Looking at the upstream LLVM pr, it seems that from the IR perspective the only necessary thing is to provide "ptrauth-sign-personality" module flag, and the rest is handled by the backend at the object file emission time. This should be already happening for us.

I've run a simple rust program that panics and then inspected the binary.

llvm-readelf -sW ./eh_pers_rust | grep DW.ref.rust_eh_personality
  2159: 000000000009f2f0     8 OBJECT  LOCAL  HIDDEN     26 DW.ref.rust_eh_personality

It lives in the data section:

llvm-readelf -SW ./eh_pers_rust | grep '\[26\]'
  [26] .data             PROGBITS        000000000009e880 06e880 000ad8 00  WA  0   0 16

And I can see the desired discriminator (0x7ead) at the correct address (0x9f2f0):

llvm-readelf --hex-dump=.data ./eh_pers_rust | grep -A2 -B2 9f2f0
0x0009f2d0 00000000 00000000 03000000 00000000 ................
0x0009f2e0 00000000 00000000 00000000 00000000 ................
0x0009f2f0 00000000 ad7e0080 00000000 00000000 .....~..........
0x0009f300 00000000 00000000 00000000 00000000 ................
0x0009f310 00000000 00000000 00000000 00000000 ................

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And we check for the correct rust_eh_personality spelling in: eec0477#diff-de25e464708ec5b026aee0d418419886c27fa316fe12808e6519d8f47fc4fea0R11

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! So if backend handles all this for us as expected - all is OK :)

args,
ty::ClosureKind::FnOnce,
);
assert!(

@kovdan01 kovdan01 Jul 20, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this assertion intended to be included in this PR or in the previous PR 5 (which contains changes to the code right after the assertion)?

View changes since the review

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good spot, moved over to the previous one.

Comment thread compiler/rustc_codegen_ssa/src/base.rs Outdated
// used to obtain function pointers, both the user's `main` and `LangItem::Start` use the Rust
// ABI (currently pointer authentication is only supported for C/System ABI). The same applies
// to the logic in `create_entry_fn` further below.
let main_llfn = cx.get_fn_addr(instance, None);

@kovdan01 kovdan01 Jul 20, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nit: maybe add some comment like /*pointer_authentication=*/None (or whatever the argument name is) here and in similar places (applies to all PRs from your stack) so a reader is not surprised by the above comment thinking like "why they are even talking about pointer signing"?

View changes since the review

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done (and in 2 other places that were using None).

}

let main_llfn = cx.get_fn_addr(instance, cx.sess().pointer_authentication_functions());
// No function pointer signing / type discriminator is needed here. Although `get_fn_addr` is

@kovdan01 kovdan01 Jul 20, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nit: probably "type discrimination" is excess info in this and similar places - if we have no function pointer signing, it's implying that we do not have any discrimination as well (because discrimination only applies to signed pointers). So this extra info might just accidentally mislead non-pauth-aware reader while adding not that much new value on top of just pointer signing being mentioned.

Please let me know if I'm missing smth and this should be retained

View changes since the review

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would actually keep this one. Someone unfamiliar with pointer authentication might skip over it without giving the comment much thought. However, someone trying to understand the pointer authentication logic by tracing all uses of get_fn_addr might wonder why this particular case is considered safe to leave unsigned.

jchlanda added 4 commits July 20, 2026 11:55
This patch introduces the following:

* Extends `FnAbi` (`callconv`) with a `ptrauth_type_discriminator`
  field. This field is only used when emitting pointer authentication
  call bundles. It is stored in `FnAbi` because the call site is not
  guaranteed to have access to an `Instance`, so the discriminator
  cannot always be computed on demand.
* Adds support for `llvm.ptrauth.resign`. This intrinsic will be used
  when support for semantic transmute is added.
* Performs a minor API redesign as groundwork for allowing call sites to
  modify schemas in place.
Also remove error messages/tests that used to guarded it.
The codegen now walks the layout of static initializer types to find extern "C"
function pointer fields, computes their type discriminators, and applies those
discriminators when emitting authenticated function pointer relocations.

Also make sure that type discrimination is never applied to init/fini
entries.
@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch 2 times, most recently from 602e8f8 to ddc9997 Compare July 21, 2026 10:23
@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch 3 times, most recently from 94562e4 to b7bbf74 Compare July 21, 2026 12:04
jchlanda added 2 commits July 21, 2026 13:18
Implement pointer authentication domain handling for function pointer
transmutes. When function pointer type discrimination is enabled,
transmuting between function pointer types with different authentication
domains now re-signs the pointer using the appropriate discriminator.
…addr` call sites

Fill in function pointer type discriminators logic across remaining
`get_fn_addr` call sites and explicitly avoid applying it where
discrimination is not meaningful.

Some uses of `get_fn_addr` are intentionally left unsigned, including
the EH personality function, entry wrappers, and compiler-generated Rust
ABI shims.
@jchlanda
jchlanda force-pushed the jakub/pac_ty_disc_PR_6 branch from b7bbf74 to dcb6bf4 Compare July 21, 2026 13:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

A-LLVM Area: Code generation parts specific to LLVM. Both correctness bugs and optimization-related issues. S-waiting-on-author Status: This is awaiting some action (such as code changes or more information) from the author. T-compiler Relevant to the compiler team, which will review and decide on the PR/issue.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants