blob: 23b31fa6c4b63ab807845f93de3f26c927ef882e [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 Tolnay0dd85ff2020-05-03 23:43:33 -07004use crate::syntax::report::Errors;
David Tolnay891061b2020-04-19 22:42:33 -07005use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07006use crate::syntax::{
David Tolnay8faec772020-11-02 00:18:19 -08007 self, check, mangle, Api, Enum, ExternFn, ExternType, Impl, Pair, ResolvableName, Signature,
Adrian Taylorc8713432020-10-21 18:20:55 -07008 Struct, Type, TypeAlias, Types,
David Tolnay3caa50a2020-04-19 21:25:34 -07009};
David Tolnay0531f432020-10-03 23:50:28 -070010use proc_macro2::{Ident, Span, TokenStream};
David Tolnayf9ffb932020-04-20 02:22:57 -070011use quote::{format_ident, quote, quote_spanned, ToTokens};
David Tolnay05ef6ff2020-08-29 11:27:05 -070012use std::mem;
13use syn::{parse_quote, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040014
David Tolnay3c64a4e2020-08-29 14:07:38 -070015pub fn bridge(mut ffi: Module) -> Result<TokenStream> {
David Tolnay0dd85ff2020-05-03 23:43:33 -070016 let ref mut errors = Errors::new();
David Tolnay05ef6ff2020-08-29 11:27:05 -070017 let content = mem::take(&mut ffi.content);
David Tolnay805dca32020-08-29 19:09:55 -070018 let trusted = ffi.unsafety.is_some();
Adrian Taylorc8713432020-10-21 18:20:55 -070019 let namespace = &ffi.namespace;
20 let ref apis = syntax::parse_items(errors, content, trusted, namespace);
David Tolnay52759782020-05-03 23:59:40 -070021 let ref types = Types::collect(errors, apis);
22 errors.propagate()?;
Adrian Taylorc8713432020-10-21 18:20:55 -070023 check::typecheck(errors, apis, types);
David Tolnay0dd85ff2020-05-03 23:43:33 -070024 errors.propagate()?;
David Tolnay7db73692019-10-20 14:51:12 -040025
David Tolnay3c64a4e2020-08-29 14:07:38 -070026 Ok(expand(ffi, apis, types))
David Tolnaycbf3f032020-05-03 23:51:14 -070027}
28
David Tolnay3c64a4e2020-08-29 14:07:38 -070029fn expand(ffi: Module, apis: &[Api], types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -040030 let mut expanded = TokenStream::new();
31 let mut hidden = TokenStream::new();
David Tolnay7db73692019-10-20 14:51:12 -040032
David Tolnay2ec14632020-05-04 00:47:10 -070033 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040034 if let Api::RustType(ety) = api {
35 expanded.extend(expand_rust_type(ety));
David Tolnayc8361022020-08-25 21:57:53 -070036 hidden.extend(expand_rust_type_assert_sized(ety));
David Tolnay7db73692019-10-20 14:51:12 -040037 }
38 }
39
David Tolnay2ec14632020-05-04 00:47:10 -070040 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040041 match api {
David Tolnay7e69f892020-10-03 22:20:22 -070042 Api::Include(_) | Api::RustType(_) | Api::Impl(_) => {}
Adrian Taylorc8713432020-10-21 18:20:55 -070043 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
44 Api::Enum(enm) => expanded.extend(expand_enum(enm)),
Joel Galenson905eb2e2020-05-04 14:58:14 -070045 Api::CxxType(ety) => {
David Tolnay17a934c2020-11-02 00:40:04 -080046 let ident = &ety.name.rust;
47 if !types.structs.contains_key(ident) && !types.enums.contains_key(ident) {
Adrian Taylorc8713432020-10-21 18:20:55 -070048 expanded.extend(expand_cxx_type(ety));
Joel Galenson905eb2e2020-05-04 14:58:14 -070049 }
50 }
David Tolnay7db73692019-10-20 14:51:12 -040051 Api::CxxFunction(efn) => {
Adrian Taylorc8713432020-10-21 18:20:55 -070052 expanded.extend(expand_cxx_function_shim(efn, types));
David Tolnay7db73692019-10-20 14:51:12 -040053 }
Adrian Taylorc8713432020-10-21 18:20:55 -070054 Api::RustFunction(efn) => hidden.extend(expand_rust_function_shim(efn, types)),
David Tolnay5f9e8ca2020-05-07 17:21:05 -070055 Api::TypeAlias(alias) => {
56 expanded.extend(expand_type_alias(alias));
Adrian Taylorc8713432020-10-21 18:20:55 -070057 hidden.extend(expand_type_alias_verify(alias, types));
David Tolnay5f9e8ca2020-05-07 17:21:05 -070058 }
David Tolnay7db73692019-10-20 14:51:12 -040059 }
60 }
61
62 for ty in types {
David Tolnay0531f432020-10-03 23:50:28 -070063 let explicit_impl = types.explicit_impls.get(ty);
David Tolnay7db73692019-10-20 14:51:12 -040064 if let Type::RustBox(ty) = ty {
65 if let Type::Ident(ident) = &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -070066 if Atom::from(&ident.rust).is_none() {
67 hidden.extend(expand_rust_box(ident, types));
David Tolnay7db73692019-10-20 14:51:12 -040068 }
69 }
Myron Ahneba35cf2020-02-05 19:41:51 +070070 } else if let Type::RustVec(ty) = ty {
71 if let Type::Ident(ident) = &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -070072 if Atom::from(&ident.rust).is_none() {
73 hidden.extend(expand_rust_vec(ident, types));
David Tolnaya006bca2020-04-25 11:28:13 -070074 }
Myron Ahneba35cf2020-02-05 19:41:51 +070075 }
David Tolnay7db73692019-10-20 14:51:12 -040076 } else if let Type::UniquePtr(ptr) = ty {
77 if let Type::Ident(ident) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -070078 if Atom::from(&ident.rust).is_none()
79 && (explicit_impl.is_some() || !types.aliases.contains_key(&ident.rust))
David Tolnay7e69f892020-10-03 22:20:22 -070080 {
Adrian Taylorc8713432020-10-21 18:20:55 -070081 expanded.extend(expand_unique_ptr(ident, types, explicit_impl));
Myron Ahneba35cf2020-02-05 19:41:51 +070082 }
83 }
David Tolnay4377a9e2020-04-24 15:20:26 -070084 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070085 if let Type::Ident(ident) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -070086 if Atom::from(&ident.rust).is_none()
87 && (explicit_impl.is_some() || !types.aliases.contains_key(&ident.rust))
David Tolnay7e69f892020-10-03 22:20:22 -070088 {
David Tolnay17631102020-04-24 19:01:30 -070089 // Generate impl for CxxVector<T> if T is a struct or opaque
90 // C++ type. Impl for primitives is already provided by cxx
91 // crate.
Adrian Taylorc8713432020-10-21 18:20:55 -070092 expanded.extend(expand_cxx_vector(ident, explicit_impl, types));
David Tolnay7db73692019-10-20 14:51:12 -040093 }
94 }
95 }
96 }
97
98 // Work around https://github.com/rust-lang/rust/issues/67851.
99 if !hidden.is_empty() {
100 expanded.extend(quote! {
101 #[doc(hidden)]
102 const _: () = {
103 #hidden
104 };
105 });
106 }
107
108 let attrs = ffi
109 .attrs
110 .into_iter()
111 .filter(|attr| attr.path.is_ident("doc"));
112 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700113 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400114
David Tolnaycbf3f032020-05-03 23:51:14 -0700115 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400116 #(#attrs)*
117 #[deny(improper_ctypes)]
118 #[allow(non_snake_case)]
119 #vis mod #ident {
120 #expanded
121 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700122 }
David Tolnay7db73692019-10-20 14:51:12 -0400123}
124
Adrian Taylorc8713432020-10-21 18:20:55 -0700125fn expand_struct(strct: &Struct) -> TokenStream {
David Tolnay17a934c2020-11-02 00:40:04 -0800126 let ident = &strct.name.rust;
David Tolnay7db73692019-10-20 14:51:12 -0400127 let doc = &strct.doc;
David Tolnayf3a9afa2020-10-08 23:22:12 -0700128 let derives = DeriveAttribute(&strct.derives);
David Tolnay17a934c2020-11-02 00:40:04 -0800129 let type_id = type_id(&strct.name);
David Tolnay7db73692019-10-20 14:51:12 -0400130 let fields = strct.fields.iter().map(|field| {
131 // This span on the pub makes "private type in public interface" errors
132 // appear in the right place.
133 let vis = Token![pub](field.ident.span());
134 quote!(#vis #field)
135 });
David Tolnay8684cc52020-10-07 17:09:48 -0700136
David Tolnay7db73692019-10-20 14:51:12 -0400137 quote! {
138 #doc
David Tolnayf3a9afa2020-10-08 23:22:12 -0700139 #derives
David Tolnay7db73692019-10-20 14:51:12 -0400140 #[repr(C)]
141 pub struct #ident {
142 #(#fields,)*
143 }
David Tolnay8684cc52020-10-07 17:09:48 -0700144
145 unsafe impl ::cxx::ExternType for #ident {
146 type Id = #type_id;
147 type Kind = ::cxx::kind::Trivial;
148 }
David Tolnay7db73692019-10-20 14:51:12 -0400149 }
150}
151
Adrian Taylorc8713432020-10-21 18:20:55 -0700152fn expand_enum(enm: &Enum) -> TokenStream {
David Tolnay17a934c2020-11-02 00:40:04 -0800153 let ident = &enm.name.rust;
Joel Galensonc03402a2020-04-23 17:31:09 -0700154 let doc = &enm.doc;
David Tolnayc605e6f2020-05-10 23:37:12 -0700155 let repr = enm.repr;
David Tolnay17a934c2020-11-02 00:40:04 -0800156 let type_id = type_id(&enm.name);
Joel Galenson88547732020-05-05 08:23:42 -0700157 let variants = enm.variants.iter().map(|variant| {
158 let variant_ident = &variant.ident;
159 let discriminant = &variant.discriminant;
160 Some(quote! {
161 pub const #variant_ident: Self = #ident { repr: #discriminant };
162 })
163 });
David Tolnay8684cc52020-10-07 17:09:48 -0700164
Joel Galensonc03402a2020-04-23 17:31:09 -0700165 quote! {
166 #doc
167 #[derive(Copy, Clone, PartialEq, Eq)]
168 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700169 pub struct #ident {
David Tolnayc605e6f2020-05-10 23:37:12 -0700170 pub repr: #repr,
Joel Galensondb1ec312020-05-01 13:57:32 -0700171 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700172
173 #[allow(non_upper_case_globals)]
174 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700175 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700176 }
David Tolnay8684cc52020-10-07 17:09:48 -0700177
178 unsafe impl ::cxx::ExternType for #ident {
179 type Id = #type_id;
180 type Kind = ::cxx::kind::Trivial;
181 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700182 }
183}
184
Adrian Taylorc8713432020-10-21 18:20:55 -0700185fn expand_cxx_type(ety: &ExternType) -> TokenStream {
David Tolnay17a934c2020-11-02 00:40:04 -0800186 let ident = &ety.name.rust;
David Tolnay7db73692019-10-20 14:51:12 -0400187 let doc = &ety.doc;
David Tolnay17a934c2020-11-02 00:40:04 -0800188 let type_id = type_id(&ety.name);
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700189
David Tolnay7db73692019-10-20 14:51:12 -0400190 quote! {
191 #doc
192 #[repr(C)]
193 pub struct #ident {
194 _private: ::cxx::private::Opaque,
195 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700196
197 unsafe impl ::cxx::ExternType for #ident {
198 type Id = #type_id;
David Tolnay38f5ad62020-10-03 18:06:44 -0700199 type Kind = ::cxx::kind::Opaque;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700200 }
David Tolnay7db73692019-10-20 14:51:12 -0400201 }
202}
203
Adrian Taylorc8713432020-10-21 18:20:55 -0700204fn expand_cxx_function_decl(efn: &ExternFn, types: &Types) -> TokenStream {
David Tolnay18ba92c2020-04-22 16:17:30 -0700205 let receiver = efn.receiver.iter().map(|receiver| {
206 let receiver_type = receiver.ty();
207 quote!(_: #receiver_type)
208 });
David Tolnay39d575f2020-03-03 00:10:56 -0800209 let args = efn.args.iter().map(|arg| {
210 let ident = &arg.ident;
211 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800212 if arg.ty == RustString {
213 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700214 } else if let Type::RustVec(_) = arg.ty {
215 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700216 } else if let Type::Fn(_) = arg.ty {
217 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800218 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800219 quote!(#ident: *mut #ty)
220 } else {
221 quote!(#ident: #ty)
222 }
223 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700224 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700225 let ret = if efn.throws {
226 quote!(-> ::cxx::private::Result)
227 } else {
228 expand_extern_return_type(&efn.ret, types)
229 };
David Tolnay7db73692019-10-20 14:51:12 -0400230 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700231 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400232 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
233 outparam = Some(quote!(__return: *mut #ret));
234 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700235 let link_name = mangle::extern_fn(efn, types);
David Tolnay17a934c2020-11-02 00:40:04 -0800236 let local_name = format_ident!("__{}", efn.name.rust);
David Tolnay7db73692019-10-20 14:51:12 -0400237 quote! {
238 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700239 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400240 }
241}
242
Adrian Taylorc8713432020-10-21 18:20:55 -0700243fn expand_cxx_function_shim(efn: &ExternFn, types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400244 let doc = &efn.doc;
Adrian Taylorc8713432020-10-21 18:20:55 -0700245 let decl = expand_cxx_function_decl(efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700246 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700247 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700248 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700249 let var = receiver.var;
250 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700251 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700252 let args = efn.args.iter().map(|arg| quote!(#arg));
253 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700254 let ret = if efn.throws {
255 let ok = match &efn.ret {
256 Some(ret) => quote!(#ret),
257 None => quote!(()),
258 };
259 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
260 } else {
261 expand_return_type(&efn.ret)
262 };
David Tolnay1e548172020-03-16 13:37:09 -0700263 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700264 let receiver_var = efn
265 .receiver
266 .iter()
267 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700268 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400269 let var = &arg.ident;
270 match &arg.ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700271 Type::Ident(ident) if ident.rust == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800272 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400273 }
274 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
275 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700276 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400277 Type::Ref(ty) => match &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -0700278 Type::Ident(ident) if ident.rust == RustString => match ty.mutability {
David Tolnayf1c7f322020-08-27 00:46:01 -0700279 None => quote!(::cxx::private::RustString::from_ref(#var)),
280 Some(_) => quote!(::cxx::private::RustString::from_mut(#var)),
281 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700282 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
283 None => quote!(::cxx::private::RustVec::from_ref_vec_string(#var)),
284 Some(_) => quote!(::cxx::private::RustVec::from_mut_vec_string(#var)),
285 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700286 Type::RustVec(_) => match ty.mutability {
287 None => quote!(::cxx::private::RustVec::from_ref(#var)),
288 Some(_) => quote!(::cxx::private::RustVec::from_mut(#var)),
289 },
David Tolnay7db73692019-10-20 14:51:12 -0400290 _ => quote!(#var),
291 },
292 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700293 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400294 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
295 _ => quote!(#var),
296 }
297 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700298 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700299 let trampolines = efn
300 .args
301 .iter()
302 .filter_map(|arg| {
303 if let Type::Fn(f) = &arg.ty {
304 let var = &arg.ident;
Adrian Taylorc8713432020-10-21 18:20:55 -0700305 Some(expand_function_pointer_trampoline(efn, var, f, types))
David Tolnay75dca2e2020-03-25 20:17:52 -0700306 } else {
307 None
308 }
309 })
310 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400311 let mut setup = efn
312 .args
313 .iter()
314 .filter(|arg| types.needs_indirect_abi(&arg.ty))
315 .map(|arg| {
316 let var = &arg.ident;
317 // These are arguments for which C++ has taken ownership of the data
318 // behind the mut reference it received.
319 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700320 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400321 }
322 })
323 .collect::<TokenStream>();
David Tolnay17a934c2020-11-02 00:40:04 -0800324 let local_name = format_ident!("__{}", efn.name.rust);
David Tolnay7db73692019-10-20 14:51:12 -0400325 let call = if indirect_return {
326 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
327 setup.extend(quote! {
328 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400329 });
David Tolnayebef4a22020-03-17 15:33:47 -0700330 if efn.throws {
331 setup.extend(quote! {
332 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
333 });
334 quote!(::std::result::Result::Ok(__return.assume_init()))
335 } else {
336 setup.extend(quote! {
337 #local_name(#(#vars,)* __return.as_mut_ptr());
338 });
339 quote!(__return.assume_init())
340 }
341 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400342 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700343 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400344 }
345 } else {
346 quote! {
347 #local_name(#(#vars),*)
348 }
349 };
Myron Ahn84849302020-03-25 22:00:58 +0700350 let expr = if efn.throws {
David Tolnaya1f02e52020-11-02 17:10:54 -0800351 efn.ret
352 .as_ref()
353 .and_then(|ret| match ret {
354 Type::Ident(ident) if ident.rust == RustString => {
355 Some(quote!(#call.map(|r| r.into_string())))
David Tolnay33f56ad2020-08-27 17:06:35 -0700356 }
David Tolnaya1f02e52020-11-02 17:10:54 -0800357 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
358 Type::RustVec(vec) => {
359 if vec.inner == RustString {
360 Some(quote!(#call.map(|r| r.into_vec_string())))
361 } else {
362 Some(quote!(#call.map(|r| r.into_vec())))
363 }
364 }
365 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
366 Type::Ref(ty) => match &ty.inner {
367 Type::Ident(ident) if ident.rust == RustString => match ty.mutability {
368 None => Some(quote!(#call.map(|r| r.as_string()))),
369 Some(_) => Some(quote!(#call.map(|r| r.as_mut_string()))),
370 },
371 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
372 None => Some(quote!(#call.map(|r| r.as_vec_string()))),
373 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec_string()))),
374 },
375 Type::RustVec(_) => match ty.mutability {
376 None => Some(quote!(#call.map(|r| r.as_vec()))),
377 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec()))),
378 },
379 _ => None,
David Tolnayf1c7f322020-08-27 00:46:01 -0700380 },
David Tolnaya1f02e52020-11-02 17:10:54 -0800381 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
382 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700383 _ => None,
David Tolnaya1f02e52020-11-02 17:10:54 -0800384 })
385 .unwrap_or(call)
Myron Ahn84849302020-03-25 22:00:58 +0700386 } else {
David Tolnaya1f02e52020-11-02 17:10:54 -0800387 efn.ret
388 .as_ref()
389 .and_then(|ret| match ret {
390 Type::Ident(ident) if ident.rust == RustString => Some(quote!(#call.into_string())),
391 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
392 Type::RustVec(vec) => {
393 if vec.inner == RustString {
394 Some(quote!(#call.into_vec_string()))
395 } else {
396 Some(quote!(#call.into_vec()))
397 }
David Tolnay33f56ad2020-08-27 17:06:35 -0700398 }
David Tolnaya1f02e52020-11-02 17:10:54 -0800399 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
400 Type::Ref(ty) => match &ty.inner {
401 Type::Ident(ident) if ident.rust == RustString => match ty.mutability {
402 None => Some(quote!(#call.as_string())),
403 Some(_) => Some(quote!(#call.as_mut_string())),
404 },
405 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
406 None => Some(quote!(#call.as_vec_string())),
407 Some(_) => Some(quote!(#call.as_mut_vec_string())),
408 },
409 Type::RustVec(_) => match ty.mutability {
410 None => Some(quote!(#call.as_vec())),
411 Some(_) => Some(quote!(#call.as_mut_vec())),
412 },
413 _ => None,
David Tolnayf1c7f322020-08-27 00:46:01 -0700414 },
David Tolnaya1f02e52020-11-02 17:10:54 -0800415 Type::Str(_) => Some(quote!(#call.as_str())),
416 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400417 _ => None,
David Tolnaya1f02e52020-11-02 17:10:54 -0800418 })
419 .unwrap_or(call)
420 };
David Tolnaye67bcf52020-09-06 23:50:44 -0700421 let mut dispatch = quote!(#setup #expr);
422 let unsafety = &efn.sig.unsafety;
423 if unsafety.is_none() {
424 dispatch = quote!(unsafe { #dispatch });
425 }
David Tolnay17a934c2020-11-02 00:40:04 -0800426 let ident = &efn.name.rust;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700427 let function_shim = quote! {
428 #doc
David Tolnaye67bcf52020-09-06 23:50:44 -0700429 pub #unsafety fn #ident(#(#all_args,)*) #ret {
David Tolnayc66cdbb2020-04-20 01:41:15 -0700430 extern "C" {
431 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400432 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700433 #trampolines
David Tolnaye67bcf52020-09-06 23:50:44 -0700434 #dispatch
David Tolnayc66cdbb2020-04-20 01:41:15 -0700435 }
436 };
437 match &efn.receiver {
438 None => function_shim,
439 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700440 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700441 quote!(impl #receiver_type { #function_shim })
442 }
David Tolnay7db73692019-10-20 14:51:12 -0400443 }
444}
445
David Tolnay75dca2e2020-03-25 20:17:52 -0700446fn expand_function_pointer_trampoline(
David Tolnay75dca2e2020-03-25 20:17:52 -0700447 efn: &ExternFn,
448 var: &Ident,
449 sig: &Signature,
450 types: &Types,
451) -> TokenStream {
Adrian Taylorc8713432020-10-21 18:20:55 -0700452 let c_trampoline = mangle::c_trampoline(efn, var, types);
453 let r_trampoline = mangle::r_trampoline(efn, var, types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700454 let local_name = parse_quote!(__);
David Tolnay17a934c2020-11-02 00:40:04 -0800455 let catch_unwind_label = format!("::{}::{}", efn.name.rust, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700456 let shim = expand_rust_function_shim_impl(
457 sig,
458 types,
459 &r_trampoline,
460 local_name,
461 catch_unwind_label,
462 None,
463 );
464
465 quote! {
466 let #var = ::cxx::private::FatFunction {
467 trampoline: {
468 extern "C" {
469 #[link_name = #c_trampoline]
470 fn trampoline();
471 }
472 #shim
473 trampoline as usize as *const ()
474 },
475 ptr: #var as usize as *const (),
476 };
477 }
478}
479
David Tolnay7db73692019-10-20 14:51:12 -0400480fn expand_rust_type(ety: &ExternType) -> TokenStream {
David Tolnay17a934c2020-11-02 00:40:04 -0800481 let ident = &ety.name.rust;
David Tolnay7db73692019-10-20 14:51:12 -0400482 quote! {
483 use super::#ident;
484 }
485}
486
David Tolnayc8361022020-08-25 21:57:53 -0700487fn expand_rust_type_assert_sized(ety: &ExternType) -> TokenStream {
488 // Rustc will render as follows if not sized:
489 //
490 // type TheirType;
491 // -----^^^^^^^^^-
492 // | |
493 // | doesn't have a size known at compile-time
494 // required by this bound in `ffi::_::__AssertSized`
495
David Tolnay17a934c2020-11-02 00:40:04 -0800496 let ident = &ety.name.rust;
David Tolnayc8361022020-08-25 21:57:53 -0700497 let begin_span = Token![::](ety.type_token.span);
498 let sized = quote_spanned! {ety.semi_token.span=>
499 #begin_span std::marker::Sized
500 };
David Tolnay17a934c2020-11-02 00:40:04 -0800501 quote_spanned! {ident.span()=>
David Tolnayc8361022020-08-25 21:57:53 -0700502 let _ = {
503 fn __AssertSized<T: ?#sized + #sized>() {}
504 __AssertSized::<#ident>
505 };
506 }
507}
508
Adrian Taylorc8713432020-10-21 18:20:55 -0700509fn expand_rust_function_shim(efn: &ExternFn, types: &Types) -> TokenStream {
510 let link_name = mangle::extern_fn(efn, types);
David Tolnay17a934c2020-11-02 00:40:04 -0800511 let local_name = format_ident!("__{}", efn.name.rust);
512 let catch_unwind_label = format!("::{}", efn.name.rust);
513 let invoke = Some(&efn.name.rust);
David Tolnay75dca2e2020-03-25 20:17:52 -0700514 expand_rust_function_shim_impl(
515 efn,
516 types,
517 &link_name,
518 local_name,
519 catch_unwind_label,
520 invoke,
521 )
522}
523
524fn expand_rust_function_shim_impl(
525 sig: &Signature,
526 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700527 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700528 local_name: Ident,
529 catch_unwind_label: String,
530 invoke: Option<&Ident>,
531) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700532 let receiver_var = sig
533 .receiver
534 .as_ref()
535 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700536 let receiver = sig.receiver.as_ref().map(|receiver| {
537 let receiver_type = receiver.ty();
538 quote!(#receiver_var: #receiver_type)
539 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700540 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800541 let ident = &arg.ident;
542 let ty = expand_extern_type(&arg.ty);
543 if types.needs_indirect_abi(&arg.ty) {
544 quote!(#ident: *mut #ty)
545 } else {
546 quote!(#ident: #ty)
547 }
548 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700549 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700550
David Tolnay3a45f2d2020-04-20 01:51:12 -0700551 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400552 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800553 match &arg.ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700554 Type::Ident(i) if i.rust == RustString => {
David Tolnaycc3767f2020-03-06 10:41:51 -0800555 quote!(::std::mem::take((*#ident).as_mut_string()))
556 }
David Tolnay40226ab2020-03-03 00:05:35 -0800557 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay33f56ad2020-08-27 17:06:35 -0700558 Type::RustVec(vec) => {
559 if vec.inner == RustString {
560 quote!(::std::mem::take((*#ident).as_mut_vec_string()))
561 } else {
562 quote!(::std::mem::take((*#ident).as_mut_vec()))
563 }
564 }
David Tolnay40226ab2020-03-03 00:05:35 -0800565 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800566 Type::Ref(ty) => match &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -0700567 Type::Ident(i) if i.rust == RustString => match ty.mutability {
David Tolnayf1c7f322020-08-27 00:46:01 -0700568 None => quote!(#ident.as_string()),
569 Some(_) => quote!(#ident.as_mut_string()),
570 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700571 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
572 None => quote!(#ident.as_vec_string()),
573 Some(_) => quote!(#ident.as_mut_vec_string()),
574 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700575 Type::RustVec(_) => match ty.mutability {
576 None => quote!(#ident.as_vec()),
577 Some(_) => quote!(#ident.as_mut_vec()),
578 },
David Tolnay40226ab2020-03-03 00:05:35 -0800579 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800580 },
David Tolnay40226ab2020-03-03 00:05:35 -0800581 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700582 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800583 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
584 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400585 }
586 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700587 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700588
589 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700590 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700591 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700592 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700593 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700594 quote!(#receiver_type::#ident)
595 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700596 },
David Tolnayb8ebeb02020-10-31 23:52:06 -0700597 None => quote!(::std::mem::transmute::<*const (), #sig>(__extern)),
David Tolnay75dca2e2020-03-25 20:17:52 -0700598 };
599 call.extend(quote! { (#(#vars),*) });
600
David Tolnay22602b42020-09-21 18:04:05 -0400601 let conversion = sig.ret.as_ref().and_then(|ret| match ret {
Adrian Taylorc8713432020-10-21 18:20:55 -0700602 Type::Ident(ident) if ident.rust == RustString => {
603 Some(quote!(::cxx::private::RustString::from))
604 }
David Tolnay22602b42020-09-21 18:04:05 -0400605 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 {
Adrian Taylorc8713432020-10-21 18:20:55 -0700615 Type::Ident(ident) if ident.rust == RustString => match ty.mutability {
David Tolnay22602b42020-09-21 18:04:05 -0400616 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 {
David Tolnayb8ebeb02020-10-31 23:52:06 -0700665 None => Some(quote!(__extern: *const ())),
David Tolnay75dca2e2020-03-25 20:17:52 -0700666 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 Tolnay17a934c2020-11-02 00:40:04 -0800681 let ident = &alias.name.rust;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700682 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
Adrian Taylorc8713432020-10-21 18:20:55 -0700689fn expand_type_alias_verify(alias: &TypeAlias, types: &Types) -> TokenStream {
David Tolnay17a934c2020-11-02 00:40:04 -0800690 let ident = &alias.name.rust;
691 let type_id = type_id(&alias.name);
David Tolnay83fe0f02020-05-07 20:00:15 -0700692 let begin_span = alias.type_token.span;
693 let end_span = alias.semi_token.span;
694 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_type::<);
695 let end = quote_spanned!(end_span=> >);
696
David Tolnay63a0e4e2020-10-03 18:52:32 -0700697 let mut verify = quote! {
David Tolnay83fe0f02020-05-07 20:00:15 -0700698 const _: fn() = #begin #ident, #type_id #end;
David Tolnay63a0e4e2020-10-03 18:52:32 -0700699 };
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700700
David Tolnay17a934c2020-11-02 00:40:04 -0800701 if types.required_trivial.contains_key(&alias.name.rust) {
David Tolnay63a0e4e2020-10-03 18:52:32 -0700702 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_kind::<);
703 verify.extend(quote! {
704 const _: fn() = #begin #ident, ::cxx::kind::Trivial #end;
705 });
Adrian Taylorc7043292020-09-25 12:48:36 -0700706 }
David Tolnay63a0e4e2020-10-03 18:52:32 -0700707
708 verify
Adrian Taylorc7043292020-09-25 12:48:36 -0700709}
710
David Tolnay8faec772020-11-02 00:18:19 -0800711fn type_id(name: &Pair) -> TokenStream {
712 let path = name.to_fully_qualified();
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700713 quote! {
714 ::cxx::type_id!(#path)
715 }
716}
717
Adrian Taylorc8713432020-10-21 18:20:55 -0700718fn expand_rust_box(ident: &ResolvableName, types: &Types) -> TokenStream {
719 let link_prefix = format!("cxxbridge05$box${}$", types.resolve(ident).to_symbol());
David Tolnay7db73692019-10-20 14:51:12 -0400720 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400721 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400722
Adrian Taylorc8713432020-10-21 18:20:55 -0700723 let local_prefix = format_ident!("{}__box_", &ident.rust);
David Tolnay7db73692019-10-20 14:51:12 -0400724 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400725 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400726
727 let span = ident.span();
728 quote_spanned! {span=>
729 #[doc(hidden)]
730 #[export_name = #link_uninit]
731 unsafe extern "C" fn #local_uninit(
732 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
733 ) {
734 ::std::ptr::write(
735 this,
736 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
737 );
738 }
739 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400740 #[export_name = #link_drop]
741 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
742 ::std::ptr::drop_in_place(this);
743 }
David Tolnay7db73692019-10-20 14:51:12 -0400744 }
745}
746
Adrian Taylorc8713432020-10-21 18:20:55 -0700747fn expand_rust_vec(elem: &ResolvableName, types: &Types) -> TokenStream {
748 let link_prefix = format!("cxxbridge05$rust_vec${}$", elem.to_symbol(types));
David Tolnayf97c2d52020-04-25 16:37:48 -0700749 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700750 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700751 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700752 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700753 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700754
Adrian Taylorc8713432020-10-21 18:20:55 -0700755 let local_prefix = format_ident!("{}__vec_", elem.rust);
David Tolnayf97c2d52020-04-25 16:37:48 -0700756 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700757 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700758 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700759 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700760 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700761
David Tolnay83a7ec92020-04-25 12:37:42 -0700762 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700763 quote_spanned! {span=>
764 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700765 #[export_name = #link_new]
766 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
767 ::std::ptr::write(this, ::cxx::private::RustVec::new());
768 }
769 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700770 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700771 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700772 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700773 }
David Tolnay85db5a02020-04-25 13:17:27 -0700774 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700775 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700776 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700777 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700778 }
David Tolnay219c0792020-04-24 20:31:37 -0700779 #[doc(hidden)]
780 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700781 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700782 (*this).as_ptr()
783 }
David Tolnay503d0192020-04-24 22:18:56 -0700784 #[doc(hidden)]
785 #[export_name = #link_stride]
786 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700787 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700788 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700789 }
790}
791
David Tolnay0531f432020-10-03 23:50:28 -0700792fn expand_unique_ptr(
Adrian Taylorc8713432020-10-21 18:20:55 -0700793 ident: &ResolvableName,
David Tolnay0531f432020-10-03 23:50:28 -0700794 types: &Types,
795 explicit_impl: Option<&Impl>,
796) -> TokenStream {
Adrian Taylorc8713432020-10-21 18:20:55 -0700797 let name = ident.rust.to_string();
798 let prefix = format!("cxxbridge05$unique_ptr${}$", ident.to_symbol(types));
David Tolnay7db73692019-10-20 14:51:12 -0400799 let link_null = format!("{}null", prefix);
800 let link_new = format!("{}new", prefix);
801 let link_raw = format!("{}raw", prefix);
802 let link_get = format!("{}get", prefix);
803 let link_release = format!("{}release", prefix);
804 let link_drop = format!("{}drop", prefix);
805
Adrian Taylorc8713432020-10-21 18:20:55 -0700806 let new_method =
807 if types.structs.contains_key(&ident.rust) || types.aliases.contains_key(&ident.rust) {
808 Some(quote! {
809 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
810 extern "C" {
811 #[link_name = #link_new]
812 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
813 }
814 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
815 unsafe { __new(&mut repr, &mut value) }
816 repr
David Tolnay53838912020-04-09 20:56:44 -0700817 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700818 })
819 } else {
820 None
821 };
David Tolnay53838912020-04-09 20:56:44 -0700822
David Tolnay0531f432020-10-03 23:50:28 -0700823 let begin_span =
824 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
825 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
826 let unsafe_token = format_ident!("unsafe", span = begin_span);
827
828 quote_spanned! {end_span=>
829 #unsafe_token impl ::cxx::private::UniquePtrTarget for #ident {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700830 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400831 fn __null() -> *mut ::std::ffi::c_void {
832 extern "C" {
833 #[link_name = #link_null]
834 fn __null(this: *mut *mut ::std::ffi::c_void);
835 }
836 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
837 unsafe { __null(&mut repr) }
838 repr
839 }
David Tolnay53838912020-04-09 20:56:44 -0700840 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400841 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
842 extern "C" {
843 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700844 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400845 }
846 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
847 __raw(&mut repr, raw);
848 repr
849 }
850 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
851 extern "C" {
852 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700853 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400854 }
855 __get(&repr)
856 }
857 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
858 extern "C" {
859 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700860 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400861 }
862 __release(&mut repr)
863 }
864 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
865 extern "C" {
866 #[link_name = #link_drop]
867 fn __drop(this: *mut *mut ::std::ffi::c_void);
868 }
869 __drop(&mut repr);
870 }
871 }
872 }
873}
874
David Tolnay0531f432020-10-03 23:50:28 -0700875fn expand_cxx_vector(
Adrian Taylorc8713432020-10-21 18:20:55 -0700876 elem: &ResolvableName,
David Tolnay0531f432020-10-03 23:50:28 -0700877 explicit_impl: Option<&Impl>,
Adrian Taylorc8713432020-10-21 18:20:55 -0700878 types: &Types,
David Tolnay0531f432020-10-03 23:50:28 -0700879) -> TokenStream {
880 let _ = explicit_impl;
Adrian Taylorc8713432020-10-21 18:20:55 -0700881 let name = elem.rust.to_string();
882 let prefix = format!("cxxbridge05$std$vector${}$", elem.to_symbol(types));
David Tolnaya83247c2020-04-24 14:36:10 -0700883 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700884 let link_get_unchecked = format!("{}get_unchecked", prefix);
Adrian Taylorc8713432020-10-21 18:20:55 -0700885 let unique_ptr_prefix = format!(
886 "cxxbridge05$unique_ptr$std$vector${}$",
887 elem.to_symbol(types)
888 );
David Tolnay3b40b6f2020-04-24 17:58:24 -0700889 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
890 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
891 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
892 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
893 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700894
David Tolnay0531f432020-10-03 23:50:28 -0700895 let begin_span =
896 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
897 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
898 let unsafe_token = format_ident!("unsafe", span = begin_span);
899
900 quote_spanned! {end_span=>
901 #unsafe_token impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700902 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700903 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700904 extern "C" {
905 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700906 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700907 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700908 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700909 }
David Tolnay93637ca2020-09-24 15:58:20 -0400910 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> *const Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700911 extern "C" {
912 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700913 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700914 }
David Tolnay93637ca2020-09-24 15:58:20 -0400915 __get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700916 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700917 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
918 extern "C" {
919 #[link_name = #link_unique_ptr_null]
920 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
921 }
922 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
923 unsafe { __unique_ptr_null(&mut repr) }
924 repr
925 }
926 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
927 extern "C" {
928 #[link_name = #link_unique_ptr_raw]
929 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
930 }
931 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
932 __unique_ptr_raw(&mut repr, raw);
933 repr
934 }
935 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
936 extern "C" {
937 #[link_name = #link_unique_ptr_get]
938 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
939 }
940 __unique_ptr_get(&repr)
941 }
942 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
943 extern "C" {
944 #[link_name = #link_unique_ptr_release]
945 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
946 }
947 __unique_ptr_release(&mut repr)
948 }
949 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
950 extern "C" {
951 #[link_name = #link_unique_ptr_drop]
952 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
953 }
954 __unique_ptr_drop(&mut repr);
955 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700956 }
957 }
958}
959
David Tolnay7db73692019-10-20 14:51:12 -0400960fn expand_return_type(ret: &Option<Type>) -> TokenStream {
961 match ret {
962 Some(ret) => quote!(-> #ret),
963 None => TokenStream::new(),
964 }
965}
966
David Tolnay75dca2e2020-03-25 20:17:52 -0700967fn indirect_return(sig: &Signature, types: &Types) -> bool {
968 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700969 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700970 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400971}
972
973fn expand_extern_type(ty: &Type) -> TokenStream {
974 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700975 Type::Ident(ident) if ident.rust == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400976 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700977 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400978 quote!(*mut #inner)
979 }
David Tolnaye6d50212020-04-25 15:38:23 -0700980 Type::RustVec(ty) => {
981 let elem = expand_extern_type(&ty.inner);
982 quote!(::cxx::private::RustVec<#elem>)
983 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700984 Type::Ref(ty) => {
985 let mutability = ty.mutability;
986 match &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -0700987 Type::Ident(ident) if ident.rust == RustString => {
David Tolnayf1c7f322020-08-27 00:46:01 -0700988 quote!(&#mutability ::cxx::private::RustString)
989 }
990 Type::RustVec(ty) => {
991 let inner = expand_extern_type(&ty.inner);
992 quote!(&#mutability ::cxx::private::RustVec<#inner>)
993 }
994 _ => quote!(#ty),
Myron Ahneba35cf2020-02-05 19:41:51 +0700995 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700996 }
David Tolnay7db73692019-10-20 14:51:12 -0400997 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700998 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400999 _ => quote!(#ty),
1000 }
1001}
1002
1003fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
1004 let ret = match ret {
1005 Some(ret) if !types.needs_indirect_abi(ret) => ret,
1006 _ => return TokenStream::new(),
1007 };
1008 let ty = expand_extern_type(ret);
1009 quote!(-> #ty)
1010}