blob: d8810dcb99b4dee60069a3482a4e82d892d768d3 [file] [log] [blame]
David Tolnaya52602b2020-03-06 10:24:34 -08001use crate::syntax::atom::Atom::{self, *};
David Tolnay08419302020-04-19 20:38:20 -07002use crate::syntax::namespace::Namespace;
David Tolnay0dd85ff2020-05-03 23:43:33 -07003use crate::syntax::report::Errors;
David Tolnay891061b2020-04-19 22:42:33 -07004use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07005use crate::syntax::{
Joel Galensonc03402a2020-04-23 17:31:09 -07006 self, check, mangle, Api, Enum, ExternFn, ExternType, Signature, Struct, Type, Types,
David Tolnay3caa50a2020-04-19 21:25:34 -07007};
David Tolnay7db73692019-10-20 14:51:12 -04008use proc_macro2::{Ident, Span, TokenStream};
David Tolnayf9ffb932020-04-20 02:22:57 -07009use quote::{format_ident, quote, quote_spanned, ToTokens};
David Tolnay83a7ec92020-04-25 12:37:42 -070010use syn::{parse_quote, Error, ItemMod, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040011
David Tolnaycbf3f032020-05-03 23:51:14 -070012pub fn bridge(namespace: &Namespace, mut ffi: ItemMod) -> Result<TokenStream> {
David Tolnay0dd85ff2020-05-03 23:43:33 -070013 let ref mut errors = Errors::new();
David Tolnaycbf3f032020-05-03 23:51:14 -070014 let content = ffi.content.take().ok_or(Error::new(
David Tolnay7db73692019-10-20 14:51:12 -040015 Span::call_site(),
16 "#[cxx::bridge] module must have inline contents",
17 ))?;
David Tolnay52759782020-05-03 23:59:40 -070018 let ref apis = syntax::parse_items(errors, content.1);
19 let ref types = Types::collect(errors, apis);
20 errors.propagate()?;
David Tolnay0dd85ff2020-05-03 23:43:33 -070021 check::typecheck(errors, namespace, apis, types);
22 errors.propagate()?;
David Tolnay7db73692019-10-20 14:51:12 -040023
David Tolnaycbf3f032020-05-03 23:51:14 -070024 Ok(expand(namespace, ffi, apis, types))
25}
26
27fn expand(namespace: &Namespace, ffi: ItemMod, apis: &[Api], types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -040028 let mut expanded = TokenStream::new();
29 let mut hidden = TokenStream::new();
David Tolnay7db73692019-10-20 14:51:12 -040030
David Tolnay2ec14632020-05-04 00:47:10 -070031 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040032 if let Api::RustType(ety) = api {
33 expanded.extend(expand_rust_type(ety));
David Tolnay7db73692019-10-20 14:51:12 -040034 let ident = &ety.ident;
David Tolnay4b07ab92020-04-22 19:50:21 -070035 let span = ident.span();
David Tolnay1044d442020-04-22 19:58:30 -070036 hidden.extend(quote_spanned! {span=>
37 let _ = ::std::ptr::read::<#ident>;
38 });
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 {
44 Api::Include(_) | Api::RustType(_) => {}
45 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
Joel Galensonc03402a2020-04-23 17:31:09 -070046 Api::Enum(enm) => expanded.extend(expand_enum(enm)),
Joel Galenson905eb2e2020-05-04 14:58:14 -070047 Api::CxxType(ety) => {
Joel Galenson0f654ff2020-05-04 20:04:21 -070048 if !types.enums.contains_key(&ety.ident) {
Joel Galenson905eb2e2020-05-04 14:58:14 -070049 expanded.extend(expand_cxx_type(ety));
50 }
51 }
David Tolnay7db73692019-10-20 14:51:12 -040052 Api::CxxFunction(efn) => {
53 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
54 }
55 Api::RustFunction(efn) => {
56 hidden.extend(expand_rust_function_shim(namespace, efn, types))
57 }
58 }
59 }
60
61 for ty in types {
62 if let Type::RustBox(ty) = ty {
63 if let Type::Ident(ident) = &ty.inner {
64 if Atom::from(ident).is_none() {
65 hidden.extend(expand_rust_box(namespace, ident));
66 }
67 }
Myron Ahneba35cf2020-02-05 19:41:51 +070068 } else if let Type::RustVec(ty) = ty {
69 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070070 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070071 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070072 }
Myron Ahneba35cf2020-02-05 19:41:51 +070073 }
David Tolnay7db73692019-10-20 14:51:12 -040074 } else if let Type::UniquePtr(ptr) = ty {
75 if let Type::Ident(ident) = &ptr.inner {
76 if Atom::from(ident).is_none() {
David Tolnay3b40b6f2020-04-24 17:58:24 -070077 expanded.extend(expand_unique_ptr(namespace, ident, types));
Myron Ahneba35cf2020-02-05 19:41:51 +070078 }
79 }
David Tolnay4377a9e2020-04-24 15:20:26 -070080 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070081 if let Type::Ident(ident) = &ptr.inner {
82 if Atom::from(ident).is_none() {
David Tolnay17631102020-04-24 19:01:30 -070083 // Generate impl for CxxVector<T> if T is a struct or opaque
84 // C++ type. Impl for primitives is already provided by cxx
85 // crate.
86 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040087 }
88 }
89 }
90 }
91
92 // Work around https://github.com/rust-lang/rust/issues/67851.
93 if !hidden.is_empty() {
94 expanded.extend(quote! {
95 #[doc(hidden)]
96 const _: () = {
97 #hidden
98 };
99 });
100 }
101
102 let attrs = ffi
103 .attrs
104 .into_iter()
105 .filter(|attr| attr.path.is_ident("doc"));
106 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700107 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400108
David Tolnaycbf3f032020-05-03 23:51:14 -0700109 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400110 #(#attrs)*
111 #[deny(improper_ctypes)]
112 #[allow(non_snake_case)]
113 #vis mod #ident {
114 #expanded
115 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700116 }
David Tolnay7db73692019-10-20 14:51:12 -0400117}
118
119fn expand_struct(strct: &Struct) -> TokenStream {
120 let ident = &strct.ident;
121 let doc = &strct.doc;
122 let derives = &strct.derives;
123 let fields = strct.fields.iter().map(|field| {
124 // This span on the pub makes "private type in public interface" errors
125 // appear in the right place.
126 let vis = Token![pub](field.ident.span());
127 quote!(#vis #field)
128 });
129 quote! {
130 #doc
131 #[derive(#(#derives),*)]
132 #[repr(C)]
133 pub struct #ident {
134 #(#fields,)*
135 }
136 }
137}
138
Joel Galensonc03402a2020-04-23 17:31:09 -0700139fn expand_enum(enm: &Enum) -> TokenStream {
140 let ident = &enm.ident;
141 let doc = &enm.doc;
Joel Galenson04fa0962020-05-01 10:00:31 -0700142 let variants = enm
143 .variants
144 .iter()
145 .scan(None, |prev_discriminant, variant| {
146 let variant_ident = &variant.ident;
147 let discriminant = variant
148 .discriminant
149 .unwrap_or_else(|| prev_discriminant.map_or(0, |n| n + 1));
150 *prev_discriminant = Some(discriminant);
151 Some(quote! {
Joel Galensondb1ec312020-05-01 13:57:32 -0700152 pub const #variant_ident: Self = #ident { repr: #discriminant };
Joel Galenson04fa0962020-05-01 10:00:31 -0700153 })
154 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700155 quote! {
156 #doc
157 #[derive(Copy, Clone, PartialEq, Eq)]
158 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700159 pub struct #ident {
160 pub repr: u32,
161 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700162
163 #[allow(non_upper_case_globals)]
164 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700165 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700166 }
167 }
168}
169
David Tolnay7db73692019-10-20 14:51:12 -0400170fn expand_cxx_type(ety: &ExternType) -> TokenStream {
171 let ident = &ety.ident;
172 let doc = &ety.doc;
173 quote! {
174 #doc
175 #[repr(C)]
176 pub struct #ident {
177 _private: ::cxx::private::Opaque,
178 }
179 }
180}
181
182fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
183 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700184 let receiver = efn.receiver.iter().map(|receiver| {
185 let receiver_type = receiver.ty();
186 quote!(_: #receiver_type)
187 });
David Tolnay39d575f2020-03-03 00:10:56 -0800188 let args = efn.args.iter().map(|arg| {
189 let ident = &arg.ident;
190 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800191 if arg.ty == RustString {
192 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700193 } else if let Type::RustVec(_) = arg.ty {
194 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700195 } else if let Type::Fn(_) = arg.ty {
196 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800197 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800198 quote!(#ident: *mut #ty)
199 } else {
200 quote!(#ident: #ty)
201 }
202 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700203 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700204 let ret = if efn.throws {
205 quote!(-> ::cxx::private::Result)
206 } else {
207 expand_extern_return_type(&efn.ret, types)
208 };
David Tolnay7db73692019-10-20 14:51:12 -0400209 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700210 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400211 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
212 outparam = Some(quote!(__return: *mut #ret));
213 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700214 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400215 let local_name = format_ident!("__{}", ident);
216 quote! {
217 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700218 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400219 }
220}
221
222fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
223 let ident = &efn.ident;
224 let doc = &efn.doc;
225 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700226 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700227 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700228 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700229 let var = receiver.var;
230 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700231 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700232 let args = efn.args.iter().map(|arg| quote!(#arg));
233 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700234 let ret = if efn.throws {
235 let ok = match &efn.ret {
236 Some(ret) => quote!(#ret),
237 None => quote!(()),
238 };
239 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
240 } else {
241 expand_return_type(&efn.ret)
242 };
David Tolnay1e548172020-03-16 13:37:09 -0700243 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700244 let receiver_var = efn
245 .receiver
246 .iter()
247 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700248 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400249 let var = &arg.ident;
250 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800251 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800252 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400253 }
254 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
255 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700256 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400257 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800258 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400259 quote!(::cxx::private::RustString::from_ref(#var))
260 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700261 Type::RustVec(_) => quote!(::cxx::private::RustVec::from_ref(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400262 _ => quote!(#var),
263 },
264 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700265 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400266 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
267 _ => quote!(#var),
268 }
269 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700270 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700271 let trampolines = efn
272 .args
273 .iter()
274 .filter_map(|arg| {
275 if let Type::Fn(f) = &arg.ty {
276 let var = &arg.ident;
277 Some(expand_function_pointer_trampoline(
278 namespace, efn, var, f, types,
279 ))
280 } else {
281 None
282 }
283 })
284 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400285 let mut setup = efn
286 .args
287 .iter()
288 .filter(|arg| types.needs_indirect_abi(&arg.ty))
289 .map(|arg| {
290 let var = &arg.ident;
291 // These are arguments for which C++ has taken ownership of the data
292 // behind the mut reference it received.
293 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700294 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400295 }
296 })
297 .collect::<TokenStream>();
298 let local_name = format_ident!("__{}", ident);
299 let call = if indirect_return {
300 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
301 setup.extend(quote! {
302 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400303 });
David Tolnayebef4a22020-03-17 15:33:47 -0700304 if efn.throws {
305 setup.extend(quote! {
306 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
307 });
308 quote!(::std::result::Result::Ok(__return.assume_init()))
309 } else {
310 setup.extend(quote! {
311 #local_name(#(#vars,)* __return.as_mut_ptr());
312 });
313 quote!(__return.assume_init())
314 }
315 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400316 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700317 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400318 }
319 } else {
320 quote! {
321 #local_name(#(#vars),*)
322 }
323 };
Myron Ahn84849302020-03-25 22:00:58 +0700324 let expr = if efn.throws {
325 efn.ret.as_ref().and_then(|ret| match ret {
326 Type::Ident(ident) if ident == RustString => {
327 Some(quote!(#call.map(|r| r.into_string())))
328 }
329 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay8b9d1762020-04-25 16:05:46 -0700330 Type::RustVec(_) => Some(quote!(#call.map(|r| r.into_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700331 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
332 Type::Ref(ty) => match &ty.inner {
333 Type::Ident(ident) if ident == RustString => {
334 Some(quote!(#call.map(|r| r.as_string())))
335 }
David Tolnay77989692020-04-25 15:57:47 -0700336 Type::RustVec(_) => Some(quote!(#call.map(|r| r.as_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700337 _ => None,
338 },
339 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700340 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700341 _ => None,
342 })
343 } else {
344 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800345 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400346 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay03dca702020-04-25 15:08:05 -0700347 Type::RustVec(_) => Some(quote!(#call.into_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400348 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
349 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800350 Type::Ident(ident) if ident == RustString => Some(quote!(#call.as_string())),
David Tolnay77989692020-04-25 15:57:47 -0700351 Type::RustVec(_) => Some(quote!(#call.as_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400352 _ => None,
353 },
354 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700355 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400356 _ => None,
357 })
Myron Ahn84849302020-03-25 22:00:58 +0700358 }
359 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700360 let function_shim = quote! {
361 #doc
362 pub fn #ident(#(#all_args,)*) #ret {
363 extern "C" {
364 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400365 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700366 #trampolines
367 unsafe {
368 #setup
369 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400370 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700371 }
372 };
373 match &efn.receiver {
374 None => function_shim,
375 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700376 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700377 quote!(impl #receiver_type { #function_shim })
378 }
David Tolnay7db73692019-10-20 14:51:12 -0400379 }
380}
381
David Tolnay75dca2e2020-03-25 20:17:52 -0700382fn expand_function_pointer_trampoline(
383 namespace: &Namespace,
384 efn: &ExternFn,
385 var: &Ident,
386 sig: &Signature,
387 types: &Types,
388) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700389 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
390 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700391 let local_name = parse_quote!(__);
392 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
393 let shim = expand_rust_function_shim_impl(
394 sig,
395 types,
396 &r_trampoline,
397 local_name,
398 catch_unwind_label,
399 None,
400 );
401
402 quote! {
403 let #var = ::cxx::private::FatFunction {
404 trampoline: {
405 extern "C" {
406 #[link_name = #c_trampoline]
407 fn trampoline();
408 }
409 #shim
410 trampoline as usize as *const ()
411 },
412 ptr: #var as usize as *const (),
413 };
414 }
415}
416
David Tolnay7db73692019-10-20 14:51:12 -0400417fn expand_rust_type(ety: &ExternType) -> TokenStream {
418 let ident = &ety.ident;
419 quote! {
420 use super::#ident;
421 }
422}
423
424fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
425 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700426 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700427 let local_name = format_ident!("__{}", ident);
428 let catch_unwind_label = format!("::{}", ident);
429 let invoke = Some(ident);
430 expand_rust_function_shim_impl(
431 efn,
432 types,
433 &link_name,
434 local_name,
435 catch_unwind_label,
436 invoke,
437 )
438}
439
440fn expand_rust_function_shim_impl(
441 sig: &Signature,
442 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700443 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700444 local_name: Ident,
445 catch_unwind_label: String,
446 invoke: Option<&Ident>,
447) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700448 let receiver_var = sig
449 .receiver
450 .as_ref()
451 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700452 let receiver = sig.receiver.as_ref().map(|receiver| {
453 let receiver_type = receiver.ty();
454 quote!(#receiver_var: #receiver_type)
455 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700456 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800457 let ident = &arg.ident;
458 let ty = expand_extern_type(&arg.ty);
459 if types.needs_indirect_abi(&arg.ty) {
460 quote!(#ident: *mut #ty)
461 } else {
462 quote!(#ident: #ty)
463 }
464 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700465 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700466
David Tolnay3a45f2d2020-04-20 01:51:12 -0700467 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400468 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800469 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800470 Type::Ident(i) if i == RustString => {
471 quote!(::std::mem::take((*#ident).as_mut_string()))
472 }
David Tolnay40226ab2020-03-03 00:05:35 -0800473 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay83c69e92020-04-25 16:59:39 -0700474 Type::RustVec(_) => quote!(::std::mem::take((*#ident).as_mut_vec())),
David Tolnay40226ab2020-03-03 00:05:35 -0800475 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800476 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800477 Type::Ident(i) if i == RustString => quote!(#ident.as_string()),
David Tolnay83c69e92020-04-25 16:59:39 -0700478 Type::RustVec(_) => quote!(#ident.as_vec()),
David Tolnay40226ab2020-03-03 00:05:35 -0800479 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800480 },
David Tolnay40226ab2020-03-03 00:05:35 -0800481 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700482 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800483 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
484 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400485 }
486 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700487 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700488
489 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700490 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700491 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700492 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700493 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700494 quote!(#receiver_type::#ident)
495 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700496 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700497 None => quote!(__extern),
498 };
499 call.extend(quote! { (#(#vars),*) });
500
501 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400502 .ret
503 .as_ref()
504 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800505 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400506 Some(quote!(::cxx::private::RustString::from(#call)))
507 }
508 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
David Tolnay83c69e92020-04-25 16:59:39 -0700509 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400510 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
511 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800512 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400513 Some(quote!(::cxx::private::RustString::from_ref(#call)))
514 }
David Tolnay83c69e92020-04-25 16:59:39 -0700515 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from_ref(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400516 _ => None,
517 },
518 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700519 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400520 _ => None,
521 })
522 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700523
524 let mut outparam = None;
525 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700526 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700527 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
528 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700529 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700530 if sig.throws {
531 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700532 Some(_) => quote!(__return),
533 None => quote!(&mut ()),
534 };
535 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700536 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400537 expr = quote!(::std::ptr::write(__return, #expr));
538 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700539
David Tolnay1e548172020-03-16 13:37:09 -0700540 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700541
542 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700543 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700544 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700545 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700546 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700547
548 let pointer = match invoke {
549 None => Some(quote!(__extern: #sig)),
550 Some(_) => None,
551 };
552
David Tolnay7db73692019-10-20 14:51:12 -0400553 quote! {
554 #[doc(hidden)]
555 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700556 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400557 let __fn = concat!(module_path!(), #catch_unwind_label);
558 #expr
559 }
560 }
561}
562
563fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700564 let link_prefix = format!("cxxbridge03$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400565 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400566 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400567
568 let local_prefix = format_ident!("{}__box_", ident);
569 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400570 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400571
572 let span = ident.span();
573 quote_spanned! {span=>
574 #[doc(hidden)]
575 #[export_name = #link_uninit]
576 unsafe extern "C" fn #local_uninit(
577 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
578 ) {
579 ::std::ptr::write(
580 this,
581 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
582 );
583 }
584 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400585 #[export_name = #link_drop]
586 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
587 ::std::ptr::drop_in_place(this);
588 }
David Tolnay7db73692019-10-20 14:51:12 -0400589 }
590}
591
David Tolnay83a7ec92020-04-25 12:37:42 -0700592fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700593 let link_prefix = format!("cxxbridge03$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700594 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700595 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700596 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700597 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700598 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700599
David Tolnay83a7ec92020-04-25 12:37:42 -0700600 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700601 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700602 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700603 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700604 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700605 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700606
David Tolnay83a7ec92020-04-25 12:37:42 -0700607 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700608 quote_spanned! {span=>
609 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700610 #[export_name = #link_new]
611 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
612 ::std::ptr::write(this, ::cxx::private::RustVec::new());
613 }
614 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700615 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700616 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700617 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700618 }
David Tolnay85db5a02020-04-25 13:17:27 -0700619 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700620 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700621 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700622 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700623 }
David Tolnay219c0792020-04-24 20:31:37 -0700624 #[doc(hidden)]
625 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700626 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700627 (*this).as_ptr()
628 }
David Tolnay503d0192020-04-24 22:18:56 -0700629 #[doc(hidden)]
630 #[export_name = #link_stride]
631 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700632 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700633 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700634 }
635}
636
David Tolnay3b40b6f2020-04-24 17:58:24 -0700637fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
638 let name = ident.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700639 let prefix = format!("cxxbridge03$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400640 let link_null = format!("{}null", prefix);
641 let link_new = format!("{}new", prefix);
642 let link_raw = format!("{}raw", prefix);
643 let link_get = format!("{}get", prefix);
644 let link_release = format!("{}release", prefix);
645 let link_drop = format!("{}drop", prefix);
646
David Tolnay3b40b6f2020-04-24 17:58:24 -0700647 let new_method = if types.structs.contains_key(ident) {
648 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700649 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
650 extern "C" {
651 #[link_name = #link_new]
652 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
653 }
654 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
655 unsafe { __new(&mut repr, &mut value) }
656 repr
657 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700658 })
659 } else {
660 None
David Tolnay53838912020-04-09 20:56:44 -0700661 };
662
David Tolnay7db73692019-10-20 14:51:12 -0400663 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700664 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
665 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400666 fn __null() -> *mut ::std::ffi::c_void {
667 extern "C" {
668 #[link_name = #link_null]
669 fn __null(this: *mut *mut ::std::ffi::c_void);
670 }
671 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
672 unsafe { __null(&mut repr) }
673 repr
674 }
David Tolnay53838912020-04-09 20:56:44 -0700675 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400676 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
677 extern "C" {
678 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700679 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400680 }
681 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
682 __raw(&mut repr, raw);
683 repr
684 }
685 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
686 extern "C" {
687 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700688 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400689 }
690 __get(&repr)
691 }
692 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
693 extern "C" {
694 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700695 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400696 }
697 __release(&mut repr)
698 }
699 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
700 extern "C" {
701 #[link_name = #link_drop]
702 fn __drop(this: *mut *mut ::std::ffi::c_void);
703 }
704 __drop(&mut repr);
705 }
706 }
707 }
708}
709
David Tolnay17631102020-04-24 19:01:30 -0700710fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700711 let name = elem.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700712 let prefix = format!("cxxbridge03$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700713 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700714 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay69601622020-04-29 18:48:36 -0700715 let unique_ptr_prefix = format!("cxxbridge03$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700716 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
717 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
718 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
719 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
720 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700721
722 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700723 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700724 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700725 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700726 extern "C" {
727 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700728 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700729 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700730 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700731 }
David Tolnay17631102020-04-24 19:01:30 -0700732 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700733 extern "C" {
734 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700735 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700736 }
David Tolnay9626d082020-04-24 14:52:45 -0700737 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700738 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700739 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
740 extern "C" {
741 #[link_name = #link_unique_ptr_null]
742 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
743 }
744 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
745 unsafe { __unique_ptr_null(&mut repr) }
746 repr
747 }
748 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
749 extern "C" {
750 #[link_name = #link_unique_ptr_raw]
751 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
752 }
753 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
754 __unique_ptr_raw(&mut repr, raw);
755 repr
756 }
757 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
758 extern "C" {
759 #[link_name = #link_unique_ptr_get]
760 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
761 }
762 __unique_ptr_get(&repr)
763 }
764 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
765 extern "C" {
766 #[link_name = #link_unique_ptr_release]
767 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
768 }
769 __unique_ptr_release(&mut repr)
770 }
771 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
772 extern "C" {
773 #[link_name = #link_unique_ptr_drop]
774 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
775 }
776 __unique_ptr_drop(&mut repr);
777 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700778 }
779 }
780}
781
David Tolnay7db73692019-10-20 14:51:12 -0400782fn expand_return_type(ret: &Option<Type>) -> TokenStream {
783 match ret {
784 Some(ret) => quote!(-> #ret),
785 None => TokenStream::new(),
786 }
787}
788
David Tolnay75dca2e2020-03-25 20:17:52 -0700789fn indirect_return(sig: &Signature, types: &Types) -> bool {
790 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700791 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700792 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400793}
794
795fn expand_extern_type(ty: &Type) -> TokenStream {
796 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800797 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400798 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700799 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400800 quote!(*mut #inner)
801 }
David Tolnaye6d50212020-04-25 15:38:23 -0700802 Type::RustVec(ty) => {
803 let elem = expand_extern_type(&ty.inner);
804 quote!(::cxx::private::RustVec<#elem>)
805 }
David Tolnay7db73692019-10-20 14:51:12 -0400806 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800807 Type::Ident(ident) if ident == RustString => quote!(&::cxx::private::RustString),
Myron Ahneba35cf2020-02-05 19:41:51 +0700808 Type::RustVec(ty) => {
809 let inner = expand_extern_type(&ty.inner);
David Tolnay6c6b7e02020-04-24 11:42:59 -0700810 quote!(&::cxx::private::RustVec<#inner>)
Myron Ahneba35cf2020-02-05 19:41:51 +0700811 }
David Tolnay7db73692019-10-20 14:51:12 -0400812 _ => quote!(#ty),
813 },
814 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700815 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400816 _ => quote!(#ty),
817 }
818}
819
820fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
821 let ret = match ret {
822 Some(ret) if !types.needs_indirect_abi(ret) => ret,
823 _ => return TokenStream::new(),
824 };
825 let ty = expand_extern_type(ret);
826 quote!(-> #ty)
827}