blob: 6fa7ea7ae493981e412a5b0712476e0ed6e4ea1a [file] [log] [blame]
David Tolnayf3a9afa2020-10-08 23:22:12 -07001use crate::derive::DeriveAttribute;
David Tolnaya52602b2020-03-06 10:24:34 -08002use crate::syntax::atom::Atom::{self, *};
David Tolnay05ef6ff2020-08-29 11:27:05 -07003use crate::syntax::file::Module;
David Tolnay08419302020-04-19 20:38:20 -07004use crate::syntax::namespace::Namespace;
David Tolnay0dd85ff2020-05-03 23:43:33 -07005use crate::syntax::report::Errors;
David Tolnay891061b2020-04-19 22:42:33 -07006use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07007use crate::syntax::{
David Tolnay0531f432020-10-03 23:50:28 -07008 self, check, mangle, Api, Enum, ExternFn, ExternType, Impl, Signature, Struct, Type, TypeAlias,
9 Types,
David Tolnay3caa50a2020-04-19 21:25:34 -070010};
David Tolnay0531f432020-10-03 23:50:28 -070011use proc_macro2::{Ident, Span, TokenStream};
David Tolnayf9ffb932020-04-20 02:22:57 -070012use quote::{format_ident, quote, quote_spanned, ToTokens};
David Tolnay05ef6ff2020-08-29 11:27:05 -070013use std::mem;
14use syn::{parse_quote, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040015
David Tolnay3c64a4e2020-08-29 14:07:38 -070016pub fn bridge(mut ffi: Module) -> Result<TokenStream> {
David Tolnay0dd85ff2020-05-03 23:43:33 -070017 let ref mut errors = Errors::new();
David Tolnay05ef6ff2020-08-29 11:27:05 -070018 let content = mem::take(&mut ffi.content);
David Tolnay805dca32020-08-29 19:09:55 -070019 let trusted = ffi.unsafety.is_some();
20 let ref apis = syntax::parse_items(errors, content, trusted);
David Tolnay52759782020-05-03 23:59:40 -070021 let ref types = Types::collect(errors, apis);
22 errors.propagate()?;
David Tolnay3c64a4e2020-08-29 14:07:38 -070023 let namespace = &ffi.namespace;
David Tolnay0dd85ff2020-05-03 23:43:33 -070024 check::typecheck(errors, namespace, apis, types);
25 errors.propagate()?;
David Tolnay7db73692019-10-20 14:51:12 -040026
David Tolnay3c64a4e2020-08-29 14:07:38 -070027 Ok(expand(ffi, apis, types))
David Tolnaycbf3f032020-05-03 23:51:14 -070028}
29
David Tolnay3c64a4e2020-08-29 14:07:38 -070030fn expand(ffi: Module, apis: &[Api], types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -040031 let mut expanded = TokenStream::new();
32 let mut hidden = TokenStream::new();
David Tolnay3c64a4e2020-08-29 14:07:38 -070033 let namespace = &ffi.namespace;
David Tolnay7db73692019-10-20 14:51:12 -040034
David Tolnay2ec14632020-05-04 00:47:10 -070035 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040036 if let Api::RustType(ety) = api {
37 expanded.extend(expand_rust_type(ety));
David Tolnayc8361022020-08-25 21:57:53 -070038 hidden.extend(expand_rust_type_assert_sized(ety));
David Tolnay7db73692019-10-20 14:51:12 -040039 }
40 }
41
David Tolnay2ec14632020-05-04 00:47:10 -070042 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040043 match api {
David Tolnay7e69f892020-10-03 22:20:22 -070044 Api::Include(_) | Api::RustType(_) | Api::Impl(_) => {}
David Tolnay8684cc52020-10-07 17:09:48 -070045 Api::Struct(strct) => expanded.extend(expand_struct(namespace, strct)),
46 Api::Enum(enm) => expanded.extend(expand_enum(namespace, enm)),
Joel Galenson905eb2e2020-05-04 14:58:14 -070047 Api::CxxType(ety) => {
David Tolnaya593d6e2020-08-29 19:48:08 -070048 let ident = &ety.ident;
49 if !types.structs.contains_key(ident) && !types.enums.contains_key(ident) {
David Tolnay5f9e8ca2020-05-07 17:21:05 -070050 expanded.extend(expand_cxx_type(namespace, ety));
Joel Galenson905eb2e2020-05-04 14:58:14 -070051 }
52 }
David Tolnay7db73692019-10-20 14:51:12 -040053 Api::CxxFunction(efn) => {
54 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
55 }
56 Api::RustFunction(efn) => {
57 hidden.extend(expand_rust_function_shim(namespace, efn, types))
58 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -070059 Api::TypeAlias(alias) => {
60 expanded.extend(expand_type_alias(alias));
David Tolnay63a0e4e2020-10-03 18:52:32 -070061 hidden.extend(expand_type_alias_verify(namespace, alias, types));
David Tolnay5f9e8ca2020-05-07 17:21:05 -070062 }
David Tolnay7db73692019-10-20 14:51:12 -040063 }
64 }
65
66 for ty in types {
David Tolnay0531f432020-10-03 23:50:28 -070067 let explicit_impl = types.explicit_impls.get(ty);
David Tolnay7db73692019-10-20 14:51:12 -040068 if let Type::RustBox(ty) = ty {
69 if let Type::Ident(ident) = &ty.inner {
70 if Atom::from(ident).is_none() {
71 hidden.extend(expand_rust_box(namespace, ident));
72 }
73 }
Myron Ahneba35cf2020-02-05 19:41:51 +070074 } else if let Type::RustVec(ty) = ty {
75 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070076 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070077 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070078 }
Myron Ahneba35cf2020-02-05 19:41:51 +070079 }
David Tolnay7db73692019-10-20 14:51:12 -040080 } else if let Type::UniquePtr(ptr) = ty {
81 if let Type::Ident(ident) = &ptr.inner {
David Tolnay7e69f892020-10-03 22:20:22 -070082 if Atom::from(ident).is_none()
David Tolnay0531f432020-10-03 23:50:28 -070083 && (explicit_impl.is_some() || !types.aliases.contains_key(ident))
David Tolnay7e69f892020-10-03 22:20:22 -070084 {
David Tolnay0531f432020-10-03 23:50:28 -070085 expanded.extend(expand_unique_ptr(namespace, ident, types, explicit_impl));
Myron Ahneba35cf2020-02-05 19:41:51 +070086 }
87 }
David Tolnay4377a9e2020-04-24 15:20:26 -070088 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070089 if let Type::Ident(ident) = &ptr.inner {
David Tolnay7e69f892020-10-03 22:20:22 -070090 if Atom::from(ident).is_none()
David Tolnay0531f432020-10-03 23:50:28 -070091 && (explicit_impl.is_some() || !types.aliases.contains_key(ident))
David Tolnay7e69f892020-10-03 22:20:22 -070092 {
David Tolnay17631102020-04-24 19:01:30 -070093 // Generate impl for CxxVector<T> if T is a struct or opaque
94 // C++ type. Impl for primitives is already provided by cxx
95 // crate.
David Tolnay0531f432020-10-03 23:50:28 -070096 expanded.extend(expand_cxx_vector(namespace, ident, explicit_impl));
David Tolnay7db73692019-10-20 14:51:12 -040097 }
98 }
99 }
100 }
101
102 // Work around https://github.com/rust-lang/rust/issues/67851.
103 if !hidden.is_empty() {
104 expanded.extend(quote! {
105 #[doc(hidden)]
106 const _: () = {
107 #hidden
108 };
109 });
110 }
111
112 let attrs = ffi
113 .attrs
114 .into_iter()
115 .filter(|attr| attr.path.is_ident("doc"));
116 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700117 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400118
David Tolnaycbf3f032020-05-03 23:51:14 -0700119 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400120 #(#attrs)*
121 #[deny(improper_ctypes)]
122 #[allow(non_snake_case)]
123 #vis mod #ident {
124 #expanded
125 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700126 }
David Tolnay7db73692019-10-20 14:51:12 -0400127}
128
David Tolnay8684cc52020-10-07 17:09:48 -0700129fn expand_struct(namespace: &Namespace, strct: &Struct) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400130 let ident = &strct.ident;
131 let doc = &strct.doc;
David Tolnayf3a9afa2020-10-08 23:22:12 -0700132 let derives = DeriveAttribute(&strct.derives);
David Tolnay8684cc52020-10-07 17:09:48 -0700133 let type_id = type_id(namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400134 let fields = strct.fields.iter().map(|field| {
135 // This span on the pub makes "private type in public interface" errors
136 // appear in the right place.
137 let vis = Token![pub](field.ident.span());
138 quote!(#vis #field)
139 });
David Tolnay8684cc52020-10-07 17:09:48 -0700140
David Tolnay7db73692019-10-20 14:51:12 -0400141 quote! {
142 #doc
David Tolnayf3a9afa2020-10-08 23:22:12 -0700143 #derives
David Tolnay7db73692019-10-20 14:51:12 -0400144 #[repr(C)]
145 pub struct #ident {
146 #(#fields,)*
147 }
David Tolnay8684cc52020-10-07 17:09:48 -0700148
149 unsafe impl ::cxx::ExternType for #ident {
150 type Id = #type_id;
151 type Kind = ::cxx::kind::Trivial;
152 }
David Tolnay7db73692019-10-20 14:51:12 -0400153 }
154}
155
David Tolnay8684cc52020-10-07 17:09:48 -0700156fn expand_enum(namespace: &Namespace, enm: &Enum) -> TokenStream {
Joel Galensonc03402a2020-04-23 17:31:09 -0700157 let ident = &enm.ident;
158 let doc = &enm.doc;
David Tolnayc605e6f2020-05-10 23:37:12 -0700159 let repr = enm.repr;
David Tolnay8684cc52020-10-07 17:09:48 -0700160 let type_id = type_id(namespace, ident);
Joel Galenson88547732020-05-05 08:23:42 -0700161 let variants = enm.variants.iter().map(|variant| {
162 let variant_ident = &variant.ident;
163 let discriminant = &variant.discriminant;
164 Some(quote! {
165 pub const #variant_ident: Self = #ident { repr: #discriminant };
166 })
167 });
David Tolnay8684cc52020-10-07 17:09:48 -0700168
Joel Galensonc03402a2020-04-23 17:31:09 -0700169 quote! {
170 #doc
171 #[derive(Copy, Clone, PartialEq, Eq)]
172 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700173 pub struct #ident {
David Tolnayc605e6f2020-05-10 23:37:12 -0700174 pub repr: #repr,
Joel Galensondb1ec312020-05-01 13:57:32 -0700175 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700176
177 #[allow(non_upper_case_globals)]
178 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700179 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700180 }
David Tolnay8684cc52020-10-07 17:09:48 -0700181
182 unsafe impl ::cxx::ExternType for #ident {
183 type Id = #type_id;
184 type Kind = ::cxx::kind::Trivial;
185 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700186 }
187}
188
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700189fn expand_cxx_type(namespace: &Namespace, ety: &ExternType) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400190 let ident = &ety.ident;
191 let doc = &ety.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700192 let type_id = type_id(namespace, ident);
193
David Tolnay7db73692019-10-20 14:51:12 -0400194 quote! {
195 #doc
196 #[repr(C)]
197 pub struct #ident {
198 _private: ::cxx::private::Opaque,
199 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700200
201 unsafe impl ::cxx::ExternType for #ident {
202 type Id = #type_id;
David Tolnay38f5ad62020-10-03 18:06:44 -0700203 type Kind = ::cxx::kind::Opaque;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700204 }
David Tolnay7db73692019-10-20 14:51:12 -0400205 }
206}
207
208fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
David Tolnay18ba92c2020-04-22 16:17:30 -0700209 let receiver = efn.receiver.iter().map(|receiver| {
210 let receiver_type = receiver.ty();
211 quote!(_: #receiver_type)
212 });
David Tolnay39d575f2020-03-03 00:10:56 -0800213 let args = efn.args.iter().map(|arg| {
214 let ident = &arg.ident;
215 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800216 if arg.ty == RustString {
217 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700218 } else if let Type::RustVec(_) = arg.ty {
219 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700220 } else if let Type::Fn(_) = arg.ty {
221 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800222 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800223 quote!(#ident: *mut #ty)
224 } else {
225 quote!(#ident: #ty)
226 }
227 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700228 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700229 let ret = if efn.throws {
230 quote!(-> ::cxx::private::Result)
231 } else {
232 expand_extern_return_type(&efn.ret, types)
233 };
David Tolnay7db73692019-10-20 14:51:12 -0400234 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700235 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400236 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
237 outparam = Some(quote!(__return: *mut #ret));
238 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700239 let link_name = mangle::extern_fn(namespace, efn);
David Tolnaya4641c72020-09-08 14:05:53 -0700240 let local_name = format_ident!("__{}", efn.ident.rust);
David Tolnay7db73692019-10-20 14:51:12 -0400241 quote! {
242 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700243 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400244 }
245}
246
247fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400248 let doc = &efn.doc;
249 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700250 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700251 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700252 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700253 let var = receiver.var;
254 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700255 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700256 let args = efn.args.iter().map(|arg| quote!(#arg));
257 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700258 let ret = if efn.throws {
259 let ok = match &efn.ret {
260 Some(ret) => quote!(#ret),
261 None => quote!(()),
262 };
263 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
264 } else {
265 expand_return_type(&efn.ret)
266 };
David Tolnay1e548172020-03-16 13:37:09 -0700267 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700268 let receiver_var = efn
269 .receiver
270 .iter()
271 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700272 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400273 let var = &arg.ident;
274 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800275 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800276 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400277 }
278 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
279 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700280 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400281 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700282 Type::Ident(ident) if ident == RustString => match ty.mutability {
283 None => quote!(::cxx::private::RustString::from_ref(#var)),
284 Some(_) => quote!(::cxx::private::RustString::from_mut(#var)),
285 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700286 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
287 None => quote!(::cxx::private::RustVec::from_ref_vec_string(#var)),
288 Some(_) => quote!(::cxx::private::RustVec::from_mut_vec_string(#var)),
289 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700290 Type::RustVec(_) => match ty.mutability {
291 None => quote!(::cxx::private::RustVec::from_ref(#var)),
292 Some(_) => quote!(::cxx::private::RustVec::from_mut(#var)),
293 },
David Tolnay7db73692019-10-20 14:51:12 -0400294 _ => quote!(#var),
295 },
296 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700297 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400298 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
299 _ => quote!(#var),
300 }
301 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700302 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700303 let trampolines = efn
304 .args
305 .iter()
306 .filter_map(|arg| {
307 if let Type::Fn(f) = &arg.ty {
308 let var = &arg.ident;
309 Some(expand_function_pointer_trampoline(
310 namespace, efn, var, f, types,
311 ))
312 } else {
313 None
314 }
315 })
316 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400317 let mut setup = efn
318 .args
319 .iter()
320 .filter(|arg| types.needs_indirect_abi(&arg.ty))
321 .map(|arg| {
322 let var = &arg.ident;
323 // These are arguments for which C++ has taken ownership of the data
324 // behind the mut reference it received.
325 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700326 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400327 }
328 })
329 .collect::<TokenStream>();
David Tolnaya4641c72020-09-08 14:05:53 -0700330 let local_name = format_ident!("__{}", efn.ident.rust);
David Tolnay7db73692019-10-20 14:51:12 -0400331 let call = if indirect_return {
332 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
333 setup.extend(quote! {
334 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400335 });
David Tolnayebef4a22020-03-17 15:33:47 -0700336 if efn.throws {
337 setup.extend(quote! {
338 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
339 });
340 quote!(::std::result::Result::Ok(__return.assume_init()))
341 } else {
342 setup.extend(quote! {
343 #local_name(#(#vars,)* __return.as_mut_ptr());
344 });
345 quote!(__return.assume_init())
346 }
347 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400348 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700349 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400350 }
351 } else {
352 quote! {
353 #local_name(#(#vars),*)
354 }
355 };
Myron Ahn84849302020-03-25 22:00:58 +0700356 let expr = if efn.throws {
357 efn.ret.as_ref().and_then(|ret| match ret {
358 Type::Ident(ident) if ident == RustString => {
359 Some(quote!(#call.map(|r| r.into_string())))
360 }
361 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700362 Type::RustVec(vec) => {
363 if vec.inner == RustString {
364 Some(quote!(#call.map(|r| r.into_vec_string())))
365 } else {
366 Some(quote!(#call.map(|r| r.into_vec())))
367 }
368 }
Myron Ahn84849302020-03-25 22:00:58 +0700369 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
370 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700371 Type::Ident(ident) if ident == RustString => match ty.mutability {
372 None => Some(quote!(#call.map(|r| r.as_string()))),
373 Some(_) => Some(quote!(#call.map(|r| r.as_mut_string()))),
374 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700375 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
376 None => Some(quote!(#call.map(|r| r.as_vec_string()))),
377 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec_string()))),
378 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700379 Type::RustVec(_) => match ty.mutability {
380 None => Some(quote!(#call.map(|r| r.as_vec()))),
381 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec()))),
382 },
Myron Ahn84849302020-03-25 22:00:58 +0700383 _ => None,
384 },
385 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700386 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700387 _ => None,
388 })
389 } else {
390 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800391 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400392 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700393 Type::RustVec(vec) => {
394 if vec.inner == RustString {
395 Some(quote!(#call.into_vec_string()))
396 } else {
397 Some(quote!(#call.into_vec()))
398 }
399 }
David Tolnay7db73692019-10-20 14:51:12 -0400400 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
401 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700402 Type::Ident(ident) if ident == RustString => match ty.mutability {
403 None => Some(quote!(#call.as_string())),
404 Some(_) => Some(quote!(#call.as_mut_string())),
405 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700406 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
407 None => Some(quote!(#call.as_vec_string())),
408 Some(_) => Some(quote!(#call.as_mut_vec_string())),
409 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700410 Type::RustVec(_) => match ty.mutability {
411 None => Some(quote!(#call.as_vec())),
412 Some(_) => Some(quote!(#call.as_mut_vec())),
413 },
David Tolnay7db73692019-10-20 14:51:12 -0400414 _ => None,
415 },
416 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700417 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400418 _ => None,
419 })
Myron Ahn84849302020-03-25 22:00:58 +0700420 }
421 .unwrap_or(call);
David Tolnaye67bcf52020-09-06 23:50:44 -0700422 let mut dispatch = quote!(#setup #expr);
423 let unsafety = &efn.sig.unsafety;
424 if unsafety.is_none() {
425 dispatch = quote!(unsafe { #dispatch });
426 }
David Tolnaya4641c72020-09-08 14:05:53 -0700427 let ident = &efn.ident.rust;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700428 let function_shim = quote! {
429 #doc
David Tolnaye67bcf52020-09-06 23:50:44 -0700430 pub #unsafety fn #ident(#(#all_args,)*) #ret {
David Tolnayc66cdbb2020-04-20 01:41:15 -0700431 extern "C" {
432 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400433 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700434 #trampolines
David Tolnaye67bcf52020-09-06 23:50:44 -0700435 #dispatch
David Tolnayc66cdbb2020-04-20 01:41:15 -0700436 }
437 };
438 match &efn.receiver {
439 None => function_shim,
440 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700441 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700442 quote!(impl #receiver_type { #function_shim })
443 }
David Tolnay7db73692019-10-20 14:51:12 -0400444 }
445}
446
David Tolnay75dca2e2020-03-25 20:17:52 -0700447fn expand_function_pointer_trampoline(
448 namespace: &Namespace,
449 efn: &ExternFn,
450 var: &Ident,
451 sig: &Signature,
452 types: &Types,
453) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700454 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
455 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700456 let local_name = parse_quote!(__);
David Tolnaya4641c72020-09-08 14:05:53 -0700457 let catch_unwind_label = format!("::{}::{}", efn.ident.rust, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700458 let shim = expand_rust_function_shim_impl(
459 sig,
460 types,
461 &r_trampoline,
462 local_name,
463 catch_unwind_label,
464 None,
465 );
466
467 quote! {
468 let #var = ::cxx::private::FatFunction {
469 trampoline: {
470 extern "C" {
471 #[link_name = #c_trampoline]
472 fn trampoline();
473 }
474 #shim
475 trampoline as usize as *const ()
476 },
477 ptr: #var as usize as *const (),
478 };
479 }
480}
481
David Tolnay7db73692019-10-20 14:51:12 -0400482fn expand_rust_type(ety: &ExternType) -> TokenStream {
483 let ident = &ety.ident;
484 quote! {
485 use super::#ident;
486 }
487}
488
David Tolnayc8361022020-08-25 21:57:53 -0700489fn expand_rust_type_assert_sized(ety: &ExternType) -> TokenStream {
490 // Rustc will render as follows if not sized:
491 //
492 // type TheirType;
493 // -----^^^^^^^^^-
494 // | |
495 // | doesn't have a size known at compile-time
496 // required by this bound in `ffi::_::__AssertSized`
497
498 let ident = &ety.ident;
499 let begin_span = Token![::](ety.type_token.span);
500 let sized = quote_spanned! {ety.semi_token.span=>
501 #begin_span std::marker::Sized
502 };
503 quote_spanned! {ident.span()=>
504 let _ = {
505 fn __AssertSized<T: ?#sized + #sized>() {}
506 __AssertSized::<#ident>
507 };
508 }
509}
510
David Tolnay7db73692019-10-20 14:51:12 -0400511fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
David Tolnay3caa50a2020-04-19 21:25:34 -0700512 let link_name = mangle::extern_fn(namespace, efn);
David Tolnaya4641c72020-09-08 14:05:53 -0700513 let local_name = format_ident!("__{}", efn.ident.rust);
514 let catch_unwind_label = format!("::{}", efn.ident.rust);
515 let invoke = Some(&efn.ident.rust);
David Tolnay75dca2e2020-03-25 20:17:52 -0700516 expand_rust_function_shim_impl(
517 efn,
518 types,
519 &link_name,
520 local_name,
521 catch_unwind_label,
522 invoke,
523 )
524}
525
526fn expand_rust_function_shim_impl(
527 sig: &Signature,
528 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700529 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700530 local_name: Ident,
531 catch_unwind_label: String,
532 invoke: Option<&Ident>,
533) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700534 let receiver_var = sig
535 .receiver
536 .as_ref()
537 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700538 let receiver = sig.receiver.as_ref().map(|receiver| {
539 let receiver_type = receiver.ty();
540 quote!(#receiver_var: #receiver_type)
541 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700542 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800543 let ident = &arg.ident;
544 let ty = expand_extern_type(&arg.ty);
545 if types.needs_indirect_abi(&arg.ty) {
546 quote!(#ident: *mut #ty)
547 } else {
548 quote!(#ident: #ty)
549 }
550 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700551 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700552
David Tolnay3a45f2d2020-04-20 01:51:12 -0700553 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400554 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800555 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800556 Type::Ident(i) if i == RustString => {
557 quote!(::std::mem::take((*#ident).as_mut_string()))
558 }
David Tolnay40226ab2020-03-03 00:05:35 -0800559 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay33f56ad2020-08-27 17:06:35 -0700560 Type::RustVec(vec) => {
561 if vec.inner == RustString {
562 quote!(::std::mem::take((*#ident).as_mut_vec_string()))
563 } else {
564 quote!(::std::mem::take((*#ident).as_mut_vec()))
565 }
566 }
David Tolnay40226ab2020-03-03 00:05:35 -0800567 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800568 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700569 Type::Ident(i) if i == RustString => match ty.mutability {
570 None => quote!(#ident.as_string()),
571 Some(_) => quote!(#ident.as_mut_string()),
572 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700573 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
574 None => quote!(#ident.as_vec_string()),
575 Some(_) => quote!(#ident.as_mut_vec_string()),
576 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700577 Type::RustVec(_) => match ty.mutability {
578 None => quote!(#ident.as_vec()),
579 Some(_) => quote!(#ident.as_mut_vec()),
580 },
David Tolnay40226ab2020-03-03 00:05:35 -0800581 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800582 },
David Tolnay40226ab2020-03-03 00:05:35 -0800583 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700584 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800585 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
586 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400587 }
588 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700589 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700590
591 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700592 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700593 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700594 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700595 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700596 quote!(#receiver_type::#ident)
597 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700598 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700599 None => quote!(__extern),
600 };
601 call.extend(quote! { (#(#vars),*) });
602
David Tolnay22602b42020-09-21 18:04:05 -0400603 let conversion = sig.ret.as_ref().and_then(|ret| match ret {
604 Type::Ident(ident) if ident == RustString => Some(quote!(::cxx::private::RustString::from)),
605 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw)),
606 Type::RustVec(vec) => {
607 if vec.inner == RustString {
608 Some(quote!(::cxx::private::RustVec::from_vec_string))
609 } else {
610 Some(quote!(::cxx::private::RustVec::from))
David Tolnay7db73692019-10-20 14:51:12 -0400611 }
David Tolnay22602b42020-09-21 18:04:05 -0400612 }
613 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw)),
614 Type::Ref(ty) => match &ty.inner {
615 Type::Ident(ident) if ident == RustString => match ty.mutability {
616 None => Some(quote!(::cxx::private::RustString::from_ref)),
617 Some(_) => Some(quote!(::cxx::private::RustString::from_mut)),
David Tolnay7db73692019-10-20 14:51:12 -0400618 },
David Tolnay22602b42020-09-21 18:04:05 -0400619 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
620 None => Some(quote!(::cxx::private::RustVec::from_ref_vec_string)),
621 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut_vec_string)),
622 },
623 Type::RustVec(_) => match ty.mutability {
624 None => Some(quote!(::cxx::private::RustVec::from_ref)),
625 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut)),
626 },
David Tolnay7db73692019-10-20 14:51:12 -0400627 _ => None,
David Tolnay22602b42020-09-21 18:04:05 -0400628 },
629 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from)),
630 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from)),
631 _ => None,
632 });
633
634 let mut expr = match conversion {
635 None => call,
636 Some(conversion) if !sig.throws => quote!(#conversion(#call)),
637 Some(conversion) => quote!(::std::result::Result::map(#call, #conversion)),
638 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700639
640 let mut outparam = None;
641 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700642 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700643 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
644 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700645 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700646 if sig.throws {
647 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700648 Some(_) => quote!(__return),
649 None => quote!(&mut ()),
650 };
651 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700652 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400653 expr = quote!(::std::ptr::write(__return, #expr));
654 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700655
David Tolnay1e548172020-03-16 13:37:09 -0700656 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700657
658 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700659 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700660 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700661 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700662 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700663
664 let pointer = match invoke {
665 None => Some(quote!(__extern: #sig)),
666 Some(_) => None,
667 };
668
David Tolnay7db73692019-10-20 14:51:12 -0400669 quote! {
670 #[doc(hidden)]
671 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700672 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400673 let __fn = concat!(module_path!(), #catch_unwind_label);
674 #expr
675 }
676 }
677}
678
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700679fn expand_type_alias(alias: &TypeAlias) -> TokenStream {
Bryan Henry890083d2020-09-13 10:34:31 -0700680 let doc = &alias.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700681 let ident = &alias.ident;
682 let ty = &alias.ty;
683 quote! {
Bryan Henry890083d2020-09-13 10:34:31 -0700684 #doc
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700685 pub type #ident = #ty;
686 }
687}
688
David Tolnay63a0e4e2020-10-03 18:52:32 -0700689fn expand_type_alias_verify(
690 namespace: &Namespace,
691 alias: &TypeAlias,
692 types: &Types,
693) -> TokenStream {
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700694 let ident = &alias.ident;
695 let type_id = type_id(namespace, ident);
David Tolnay83fe0f02020-05-07 20:00:15 -0700696 let begin_span = alias.type_token.span;
697 let end_span = alias.semi_token.span;
698 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_type::<);
699 let end = quote_spanned!(end_span=> >);
700
David Tolnay63a0e4e2020-10-03 18:52:32 -0700701 let mut verify = quote! {
David Tolnay83fe0f02020-05-07 20:00:15 -0700702 const _: fn() = #begin #ident, #type_id #end;
David Tolnay63a0e4e2020-10-03 18:52:32 -0700703 };
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700704
David Tolnay3208fd72020-10-03 20:19:40 -0700705 if types.required_trivial.contains_key(&alias.ident) {
David Tolnay63a0e4e2020-10-03 18:52:32 -0700706 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_kind::<);
707 verify.extend(quote! {
708 const _: fn() = #begin #ident, ::cxx::kind::Trivial #end;
709 });
Adrian Taylorc7043292020-09-25 12:48:36 -0700710 }
David Tolnay63a0e4e2020-10-03 18:52:32 -0700711
712 verify
Adrian Taylorc7043292020-09-25 12:48:36 -0700713}
714
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700715fn type_id(namespace: &Namespace, ident: &Ident) -> TokenStream {
716 let mut path = String::new();
717 for name in namespace {
718 path += &name.to_string();
719 path += "::";
720 }
721 path += &ident.to_string();
722
723 quote! {
724 ::cxx::type_id!(#path)
725 }
726}
727
David Tolnay7db73692019-10-20 14:51:12 -0400728fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay8f16ae72020-10-08 18:21:13 -0700729 let link_prefix = format!("cxxbridge05$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400730 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400731 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400732
733 let local_prefix = format_ident!("{}__box_", ident);
734 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400735 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400736
737 let span = ident.span();
738 quote_spanned! {span=>
739 #[doc(hidden)]
740 #[export_name = #link_uninit]
741 unsafe extern "C" fn #local_uninit(
742 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
743 ) {
744 ::std::ptr::write(
745 this,
746 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
747 );
748 }
749 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400750 #[export_name = #link_drop]
751 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
752 ::std::ptr::drop_in_place(this);
753 }
David Tolnay7db73692019-10-20 14:51:12 -0400754 }
755}
756
David Tolnay83a7ec92020-04-25 12:37:42 -0700757fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay8f16ae72020-10-08 18:21:13 -0700758 let link_prefix = format!("cxxbridge05$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700759 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700760 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700761 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700762 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700763 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700764
David Tolnay83a7ec92020-04-25 12:37:42 -0700765 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700766 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700767 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700768 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700769 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700770 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700771
David Tolnay83a7ec92020-04-25 12:37:42 -0700772 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700773 quote_spanned! {span=>
774 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700775 #[export_name = #link_new]
776 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
777 ::std::ptr::write(this, ::cxx::private::RustVec::new());
778 }
779 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700780 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700781 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700782 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700783 }
David Tolnay85db5a02020-04-25 13:17:27 -0700784 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700785 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700786 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700787 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700788 }
David Tolnay219c0792020-04-24 20:31:37 -0700789 #[doc(hidden)]
790 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700791 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700792 (*this).as_ptr()
793 }
David Tolnay503d0192020-04-24 22:18:56 -0700794 #[doc(hidden)]
795 #[export_name = #link_stride]
796 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700797 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700798 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700799 }
800}
801
David Tolnay0531f432020-10-03 23:50:28 -0700802fn expand_unique_ptr(
803 namespace: &Namespace,
804 ident: &Ident,
805 types: &Types,
806 explicit_impl: Option<&Impl>,
807) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700808 let name = ident.to_string();
David Tolnay8f16ae72020-10-08 18:21:13 -0700809 let prefix = format!("cxxbridge05$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400810 let link_null = format!("{}null", prefix);
811 let link_new = format!("{}new", prefix);
812 let link_raw = format!("{}raw", prefix);
813 let link_get = format!("{}get", prefix);
814 let link_release = format!("{}release", prefix);
815 let link_drop = format!("{}drop", prefix);
816
David Tolnay3b40b6f2020-04-24 17:58:24 -0700817 let new_method = if types.structs.contains_key(ident) {
818 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700819 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
820 extern "C" {
821 #[link_name = #link_new]
822 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
823 }
824 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
825 unsafe { __new(&mut repr, &mut value) }
826 repr
827 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700828 })
829 } else {
830 None
David Tolnay53838912020-04-09 20:56:44 -0700831 };
832
David Tolnay0531f432020-10-03 23:50:28 -0700833 let begin_span =
834 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
835 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
836 let unsafe_token = format_ident!("unsafe", span = begin_span);
837
838 quote_spanned! {end_span=>
839 #unsafe_token impl ::cxx::private::UniquePtrTarget for #ident {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700840 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400841 fn __null() -> *mut ::std::ffi::c_void {
842 extern "C" {
843 #[link_name = #link_null]
844 fn __null(this: *mut *mut ::std::ffi::c_void);
845 }
846 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
847 unsafe { __null(&mut repr) }
848 repr
849 }
David Tolnay53838912020-04-09 20:56:44 -0700850 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400851 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
852 extern "C" {
853 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700854 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400855 }
856 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
857 __raw(&mut repr, raw);
858 repr
859 }
860 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
861 extern "C" {
862 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700863 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400864 }
865 __get(&repr)
866 }
867 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
868 extern "C" {
869 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700870 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400871 }
872 __release(&mut repr)
873 }
874 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
875 extern "C" {
876 #[link_name = #link_drop]
877 fn __drop(this: *mut *mut ::std::ffi::c_void);
878 }
879 __drop(&mut repr);
880 }
881 }
882 }
883}
884
David Tolnay0531f432020-10-03 23:50:28 -0700885fn expand_cxx_vector(
886 namespace: &Namespace,
887 elem: &Ident,
888 explicit_impl: Option<&Impl>,
889) -> TokenStream {
890 let _ = explicit_impl;
David Tolnay3b40b6f2020-04-24 17:58:24 -0700891 let name = elem.to_string();
David Tolnay8f16ae72020-10-08 18:21:13 -0700892 let prefix = format!("cxxbridge05$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700893 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700894 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay8f16ae72020-10-08 18:21:13 -0700895 let unique_ptr_prefix = format!("cxxbridge05$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700896 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
897 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
898 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
899 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
900 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700901
David Tolnay0531f432020-10-03 23:50:28 -0700902 let begin_span =
903 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
904 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
905 let unsafe_token = format_ident!("unsafe", span = begin_span);
906
907 quote_spanned! {end_span=>
908 #unsafe_token impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700909 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700910 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700911 extern "C" {
912 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700913 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700914 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700915 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700916 }
David Tolnay93637ca2020-09-24 15:58:20 -0400917 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> *const Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700918 extern "C" {
919 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700920 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700921 }
David Tolnay93637ca2020-09-24 15:58:20 -0400922 __get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700923 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700924 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
925 extern "C" {
926 #[link_name = #link_unique_ptr_null]
927 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
928 }
929 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
930 unsafe { __unique_ptr_null(&mut repr) }
931 repr
932 }
933 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
934 extern "C" {
935 #[link_name = #link_unique_ptr_raw]
936 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
937 }
938 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
939 __unique_ptr_raw(&mut repr, raw);
940 repr
941 }
942 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
943 extern "C" {
944 #[link_name = #link_unique_ptr_get]
945 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
946 }
947 __unique_ptr_get(&repr)
948 }
949 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
950 extern "C" {
951 #[link_name = #link_unique_ptr_release]
952 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
953 }
954 __unique_ptr_release(&mut repr)
955 }
956 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
957 extern "C" {
958 #[link_name = #link_unique_ptr_drop]
959 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
960 }
961 __unique_ptr_drop(&mut repr);
962 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700963 }
964 }
965}
966
David Tolnay7db73692019-10-20 14:51:12 -0400967fn expand_return_type(ret: &Option<Type>) -> TokenStream {
968 match ret {
969 Some(ret) => quote!(-> #ret),
970 None => TokenStream::new(),
971 }
972}
973
David Tolnay75dca2e2020-03-25 20:17:52 -0700974fn indirect_return(sig: &Signature, types: &Types) -> bool {
975 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700976 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700977 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400978}
979
980fn expand_extern_type(ty: &Type) -> TokenStream {
981 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800982 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400983 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700984 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400985 quote!(*mut #inner)
986 }
David Tolnaye6d50212020-04-25 15:38:23 -0700987 Type::RustVec(ty) => {
988 let elem = expand_extern_type(&ty.inner);
989 quote!(::cxx::private::RustVec<#elem>)
990 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700991 Type::Ref(ty) => {
992 let mutability = ty.mutability;
993 match &ty.inner {
994 Type::Ident(ident) if ident == RustString => {
995 quote!(&#mutability ::cxx::private::RustString)
996 }
997 Type::RustVec(ty) => {
998 let inner = expand_extern_type(&ty.inner);
999 quote!(&#mutability ::cxx::private::RustVec<#inner>)
1000 }
1001 _ => quote!(#ty),
Myron Ahneba35cf2020-02-05 19:41:51 +07001002 }
David Tolnayf1c7f322020-08-27 00:46:01 -07001003 }
David Tolnay7db73692019-10-20 14:51:12 -04001004 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001005 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -04001006 _ => quote!(#ty),
1007 }
1008}
1009
1010fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
1011 let ret = match ret {
1012 Some(ret) if !types.needs_indirect_abi(ret) => ret,
1013 _ => return TokenStream::new(),
1014 };
1015 let ty = expand_extern_type(ret);
1016 quote!(-> #ty)
1017}