-
Notifications
You must be signed in to change notification settings - Fork 13.8k
Add #[rustc_pass_indirectly_in_non_rustic_abis]
#144529
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Conversation
r? @SparrowLii rustbot has assigned @SparrowLii. Use |
Some changes occurred in compiler/rustc_attr_data_structures Some changes occurred in compiler/rustc_attr_parsing Some changes occurred in compiler/rustc_passes/src/check_attr.rs |
Ty: TyAbiInterface<'a, C> + Copy, | ||
C: HasDataLayout, | ||
{ | ||
if arg.layout.pass_indirectly_in_non_rustic_abis(cx) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If we rely on every single callconv getting this right, we're toast. It's way too easy to forget this somewhere.
Is there some way we can do this centrally for all ABIs?
For instance, we could apply this logic after the target-specific ABI stuff has been done.
Cc @workingjubilee
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The individual calling conventions sometimes still need to be aware of the parameters, to update the number of remaining general-purpose registers, and the current design of the calling convention code makes it hard to abstract this. Separately from this PR, I've been planning to refactor the calling convention handling a bit as even without this change there's a lot of code duplication already (all the compute_abi_info
functions are essentially variants of the same function with calls to classify_arg
and classify_ret
); this refactoring should make it possible to do this in a more centralised way.
For now, this PR previously had a cfg!(debug_assertions)
-guarded check at the end of adjust_for_foreign_abi
in callconv/mod.rs
that asserts that individual calling convention correctly set all the #[rustc_pass_indirectly_in_non_rustic_abis]
arguments to be passed indirectly. I've updated the check so it now always run rather than just running when debug_assetions
are enabled.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The individual calling conventions sometimes still need to be aware of the parameters, to update the number of remaining general-purpose registers,
Urgh, right, I forgot we need to care about low-level nonsense like that here. :/
Regarding refactoring the ABI code, also see #119183. I think @workingjubilee also has some thoughts in that direction. I'm happy to discuss design options and provide feedback.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not the most robust idea, but we there could be some kind of ICE-causing bomb that gets defused when checking an arg's pass_indirectly_in_non_rustic_abis
and ignored if there are no args. This at least makes sure that new targets don't get very far if they miss this important detail.
Or a codegen test that gets run on all targets?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We have some code doing sanity checks on the ABI after it got computed. We could probably add an assertion there.
rust/compiler/rustc_ty_utils/src/abi.rs
Lines 368 to 372 in d71ed8d
fn fn_abi_sanity_check<'tcx>( | |
cx: &LayoutCx<'tcx>, | |
fn_abi: &FnAbi<'tcx, Ty<'tcx>>, | |
spec_abi: ExternAbi, | |
) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We totally can
if arg.layout.pass_indirectly_in_non_rustic_abis(cx) {
assert_matches!(arg.mode, PassMode::Indirect { on_stack: false, .. });
}
How do we make sure that triggers for every ABI though? Can we add an unused definition to core
that uses this attribute and a function using it, so that the code path is exercised? E.g. put this somewhere
#[rustc_pass_indirectly_in_non_rustic_abis]
struct S(usize);
#[expect(unused)]
fn indirect_argument(_: S) {}
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The indirect_argument
function would not be codegened if it doesn't get called, so that wouldn't trigger the ABI check. It should probably just be a ui test. This won't run in CI for tier 3 targets, but the worst that can happen is that you get an ICE, not a silent miscompilation.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How do we make sure that triggers for every ABI though?
It's an unstable attribute, so worst case we get an ICE when building libcore. That seems fine.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The indirect_argument function would not be codegened if it doesn't get called, so that wouldn't trigger the ABI check.
what if we use a const fn
? Maybe using some const _: () = assert!(/* ... */)
, would that work?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Using const
does work, I added that and a fix for transparent wrappers ignoring the attribute at master...folkertdev:rust:pass-indirectly-attr-updates
@beetrees feel free to steal or chery-pick from that. I'd also happily force-push to this branch if you don't have time/interest. (this is on the critical path for c-variadics now that the error messages are in a good state, and I'd hate to waste the reviewer momentum).
c0357c1
to
61196cb
Compare
@jdonszelmann could you have a look at the attribute code here? This is my first time actually seeing the new infrastructure so I can't say if the way it is used here is correct. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The compiler part LGTM apart from these comments, but I can't really review these ABI adjustments.
61196cb
to
ed746a3
Compare
☔ The latest upstream changes (presumably #144740) made this pull request unmergeable. Please resolve the merge conflicts. |
ed746a3
to
d91160a
Compare
Some changes occurred in compiler/rustc_hir/src/attrs |
This comment has been minimized.
This comment has been minimized.
d91160a
to
0401bf3
Compare
r? @joshtriplett :) |
|
0401bf3
to
b8bd968
Compare
This comment has been minimized.
This comment has been minimized.
Besides Jubilee, who is not available, the only other person I can think of to review ABI code is r? @bjorn3 |
☔ The latest upstream changes (presumably #146360) made this pull request unmergeable. Please resolve the merge conflicts. |
b8bd968
to
1953e54
Compare
This PR was rebased onto a different master commit. Here's a range-diff highlighting what actually changed. Rebasing is a normal part of keeping PRs up to date, so no action is needed—this note is just to help reviewers. |
☔ The latest upstream changes (presumably #146494) made this pull request unmergeable. Please resolve the merge conflicts. |
Broadly speaking, this seems like a reasonable way to tag types for use with this special calling convention. I can't speak to the maintainability of this approach, but generally speaking, I'm not sure it's possible to handle |
/// If this method returns `true`, then this type should always have a `PassMode` of | ||
/// `Indirect { on_stack: false, .. }` when being used as the argument type of a function with a | ||
/// non-Rustic ABI (this is true for structs annotated with the | ||
/// `#[rustc_pass_indirectly_in_non_rustic_abis]` attribute). | ||
/// | ||
/// This function handles transparent types automatically. | ||
pub fn pass_indirectly_in_non_rustic_abis<C>(mut self, cx: &C) -> bool |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some reference to array-to-pointer decay might be useful here to motivate why this function exists?
while self.is_transparent() | ||
&& let Some((_, field)) = self.non_1zst_field(cx) | ||
{ | ||
self = field; | ||
} | ||
Ty::is_pass_indirectly_in_non_rustic_abis_flag_set(self) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
so right now the type is only passed indirectly if the base type is passed indirectly. This just confused me for a while, I think this should also work:
#[repr(C)]
struct Vanilla(i32);
#[repr(transparent)]
#[rustc_pass_indirectly_in_non_rustic_abis]
struct Indirectly(Vanilla);
I.e. if any of the transparent wrappers is rustc_pass_indirectly_in_non_rustic_abis
then the whole type should be passed indirectly.
Correct use of the attribute depends not just on the type but also its surrounding conventions -- for instance, it is already UB in C to even look at a va_args after it has been passed to another function, if I understood correctly. That's crucial because a Rust "pass_indirectly" type still passes a copy of the argument value to the callee, whereas in C it passes effectively an implicit reference -- if the callee mutates the argument, a C caller will see the mutations, a Rust caller will not. |
This PR adds an internal
#[rustc_pass_indirectly_in_non_rustic_abis]
attribute that can be applied to structs. Structs marked with this attribute will always be passed usingPassMode::Indirect { on_stack: false, .. }
when being passed by value to functions with non-Rustic calling conventions. This is needed by #141980; see that PR for further details.cc @joshtriplett