charon_lib/transform/normalize/
transform_dyn_trait_calls.rs

1//! Transform method calls on `&dyn Trait` to vtable function pointer calls.
2//!
3//! This pass converts direct method calls on trait objects into calls through vtable
4//! function pointers. For example:
5//!
6//! ```rust,ignore
7//! let x: &dyn Trait = &obj;
8//! x.method(args);
9//! ```
10//!
11//! is transformed from:
12//! ```text
13//! @0 := call <dyn Trait as Trait>::method(x, args)
14//! ```
15//! to:
16//! ```text
17//! @0 := (move (*@receiver.ptr_metadata).method_check)(move (@receiver), move (@args)) // Call through function pointer
18//! ```
19
20use super::super::ctx::UllbcPass;
21use crate::{
22    errors::Error,
23    formatter::IntoFormatter,
24    pretty::FmtWithCtx,
25    raise_error, register_error,
26    transform::{TransformCtx, ctx::UllbcStatementTransformCtx},
27    ullbc_ast::*,
28};
29
30impl<'a> UllbcStatementTransformCtx<'a> {
31    /// Transform a call to a trait method on a dyn trait object
32    fn transform_dyn_trait_call(&mut self, call: &mut Call) -> Result<(), Error> {
33        let fmt_ctx = &self.ctx.into_fmt();
34
35        // Detect if this call should be transformed
36        let FnOperand::Regular(fn_ptr) = &call.func else {
37            return Ok(()); // Not a regular function call
38        };
39        let FnPtrKind::Trait(trait_ref, method_name, _) = &fn_ptr.kind else {
40            return Ok(()); // Not a trait method call
41        };
42        let TraitRefKind::Dyn = &trait_ref.kind else {
43            return Ok(()); // Not a dyn trait trait call
44        };
45        let trait_pred = trait_ref.trait_decl_ref.clone().erase();
46
47        // Get the type of the vtable struct.
48        let vtable_decl_ref: TypeDeclRef = {
49            // Get the trait declaration by its ID
50            let Some(trait_decl) = self.ctx.translated.trait_decls.get(trait_pred.id) else {
51                return Ok(()); // Unknown trait
52            };
53            // Get vtable ref from definition for correct ID.
54            let Some(vtable_ty) = &trait_decl.vtable else {
55                raise_error!(
56                    self.ctx,
57                    self.span,
58                    "Found a `dyn Trait` method call for non-dyn-compatible trait `{}`!",
59                    trait_pred.id.with_ctx(fmt_ctx)
60                );
61            };
62            vtable_ty
63                .clone()
64                .substitute_with_self(&trait_pred.generics, &trait_ref.kind)
65        };
66        let vtable_decl_id = *vtable_decl_ref.id.as_adt().unwrap();
67        let Some(vtable_decl) = self.ctx.translated.type_decls.get(vtable_decl_id) else {
68            return Ok(()); // Missing data
69        };
70
71        // Retreive the method field from the vtable struct definition.
72        let method_field_name = format!("method_{}", method_name);
73        let Some((method_field_id, method_field)) =
74            vtable_decl.get_field_by_name(None, &method_field_name)
75        else {
76            let vtable_name = vtable_decl_ref.id.with_ctx(fmt_ctx).to_string();
77            raise_error!(
78                self.ctx,
79                self.span,
80                "Could not determine method index for {} in vtable {}",
81                method_name,
82                vtable_name
83            );
84        };
85        let method_ty = method_field
86            .ty
87            .clone()
88            .substitute(&vtable_decl_ref.generics);
89
90        // Get the receiver (first argument).
91        if call.args.is_empty() {
92            raise_error!(self.ctx, self.span, "Dyn trait call has no arguments!");
93        }
94        let dyn_trait_place = match &call.args[0] {
95            Operand::Copy(place) | Operand::Move(place) => place,
96            Operand::Const(_) => {
97                panic!("Unexpected constant as receiver for dyn trait method call")
98            }
99        };
100
101        // Construct the `(*ptr.ptr_metadata).method_field` place.
102        let vtable_ty = TyKind::Adt(vtable_decl_ref).into_ty();
103        let ptr_to_vtable_ty = Ty::new(TyKind::RawPtr(vtable_ty.clone(), RefKind::Shared));
104        let method_field_place = dyn_trait_place
105            .clone()
106            .project(ProjectionElem::PtrMetadata, ptr_to_vtable_ty)
107            .project(ProjectionElem::Deref, vtable_ty)
108            .project(
109                ProjectionElem::Field(FieldProjKind::Adt(vtable_decl_id, None), method_field_id),
110                method_ty,
111            );
112
113        // Transform the original call to use the function pointer
114        call.func = FnOperand::Move(method_field_place);
115
116        Ok(())
117    }
118}
119
120pub struct Transform;
121
122impl UllbcPass for Transform {
123    fn transform_function(&self, ctx: &mut TransformCtx, decl: &mut FunDecl) {
124        decl.transform_ullbc_terminators(ctx, |ctx, term| {
125            if let TerminatorKind::Call { call, .. } = &mut term.kind {
126                let _ = ctx.transform_dyn_trait_call(call);
127            }
128        });
129    }
130}
charon_lib/transform/normalize/transform_dyn_trait_calls.rs

charon_lib/transform/normalize/
transform_dyn_trait_calls.rs
