blob: d1af2cfa582590bc870909d82e15fb9eb93cae19 [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 Tolnay2ec14632020-05-04 00:47:10 -070018 let ref apis = syntax::parse_items(content.1)?;
19 let ref types = Types::collect(apis)?;
David Tolnay0dd85ff2020-05-03 23:43:33 -070020 check::typecheck(errors, namespace, apis, types);
21 errors.propagate()?;
David Tolnay7db73692019-10-20 14:51:12 -040022
David Tolnaycbf3f032020-05-03 23:51:14 -070023 Ok(expand(namespace, ffi, apis, types))
24}
25
26fn expand(namespace: &Namespace, ffi: ItemMod, apis: &[Api], types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -040027 let mut expanded = TokenStream::new();
28 let mut hidden = TokenStream::new();
David Tolnay7db73692019-10-20 14:51:12 -040029
David Tolnay2ec14632020-05-04 00:47:10 -070030 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040031 if let Api::RustType(ety) = api {
32 expanded.extend(expand_rust_type(ety));
David Tolnay7db73692019-10-20 14:51:12 -040033 let ident = &ety.ident;
David Tolnay4b07ab92020-04-22 19:50:21 -070034 let span = ident.span();
David Tolnay1044d442020-04-22 19:58:30 -070035 hidden.extend(quote_spanned! {span=>
36 let _ = ::std::ptr::read::<#ident>;
37 });
David Tolnay7db73692019-10-20 14:51:12 -040038 }
39 }
40
David Tolnay2ec14632020-05-04 00:47:10 -070041 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040042 match api {
43 Api::Include(_) | Api::RustType(_) => {}
44 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
Joel Galensonc03402a2020-04-23 17:31:09 -070045 Api::Enum(enm) => expanded.extend(expand_enum(enm)),
David Tolnay7db73692019-10-20 14:51:12 -040046 Api::CxxType(ety) => expanded.extend(expand_cxx_type(ety)),
47 Api::CxxFunction(efn) => {
48 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
49 }
50 Api::RustFunction(efn) => {
51 hidden.extend(expand_rust_function_shim(namespace, efn, types))
52 }
53 }
54 }
55
56 for ty in types {
57 if let Type::RustBox(ty) = ty {
58 if let Type::Ident(ident) = &ty.inner {
59 if Atom::from(ident).is_none() {
60 hidden.extend(expand_rust_box(namespace, ident));
61 }
62 }
Myron Ahneba35cf2020-02-05 19:41:51 +070063 } else if let Type::RustVec(ty) = ty {
64 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070065 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070066 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070067 }
Myron Ahneba35cf2020-02-05 19:41:51 +070068 }
David Tolnay7db73692019-10-20 14:51:12 -040069 } else if let Type::UniquePtr(ptr) = ty {
70 if let Type::Ident(ident) = &ptr.inner {
71 if Atom::from(ident).is_none() {
David Tolnay3b40b6f2020-04-24 17:58:24 -070072 expanded.extend(expand_unique_ptr(namespace, ident, types));
Myron Ahneba35cf2020-02-05 19:41:51 +070073 }
74 }
David Tolnay4377a9e2020-04-24 15:20:26 -070075 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070076 if let Type::Ident(ident) = &ptr.inner {
77 if Atom::from(ident).is_none() {
David Tolnay17631102020-04-24 19:01:30 -070078 // Generate impl for CxxVector<T> if T is a struct or opaque
79 // C++ type. Impl for primitives is already provided by cxx
80 // crate.
81 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040082 }
83 }
84 }
85 }
86
87 // Work around https://github.com/rust-lang/rust/issues/67851.
88 if !hidden.is_empty() {
89 expanded.extend(quote! {
90 #[doc(hidden)]
91 const _: () = {
92 #hidden
93 };
94 });
95 }
96
97 let attrs = ffi
98 .attrs
99 .into_iter()
100 .filter(|attr| attr.path.is_ident("doc"));
101 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700102 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400103
David Tolnaycbf3f032020-05-03 23:51:14 -0700104 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400105 #(#attrs)*
106 #[deny(improper_ctypes)]
107 #[allow(non_snake_case)]
108 #vis mod #ident {
109 #expanded
110 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700111 }
David Tolnay7db73692019-10-20 14:51:12 -0400112}
113
114fn expand_struct(strct: &Struct) -> TokenStream {
115 let ident = &strct.ident;
116 let doc = &strct.doc;
117 let derives = &strct.derives;
118 let fields = strct.fields.iter().map(|field| {
119 // This span on the pub makes "private type in public interface" errors
120 // appear in the right place.
121 let vis = Token![pub](field.ident.span());
122 quote!(#vis #field)
123 });
124 quote! {
125 #doc
126 #[derive(#(#derives),*)]
127 #[repr(C)]
128 pub struct #ident {
129 #(#fields,)*
130 }
131 }
132}
133
Joel Galensonc03402a2020-04-23 17:31:09 -0700134fn expand_enum(enm: &Enum) -> TokenStream {
135 let ident = &enm.ident;
136 let doc = &enm.doc;
Joel Galenson04fa0962020-05-01 10:00:31 -0700137 let variants = enm
138 .variants
139 .iter()
140 .scan(None, |prev_discriminant, variant| {
141 let variant_ident = &variant.ident;
142 let discriminant = variant
143 .discriminant
144 .unwrap_or_else(|| prev_discriminant.map_or(0, |n| n + 1));
145 *prev_discriminant = Some(discriminant);
146 Some(quote! {
Joel Galensondb1ec312020-05-01 13:57:32 -0700147 pub const #variant_ident: Self = #ident { repr: #discriminant };
Joel Galenson04fa0962020-05-01 10:00:31 -0700148 })
149 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700150 quote! {
151 #doc
152 #[derive(Copy, Clone, PartialEq, Eq)]
153 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700154 pub struct #ident {
155 pub repr: u32,
156 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700157
158 #[allow(non_upper_case_globals)]
159 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700160 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700161 }
162 }
163}
164
David Tolnay7db73692019-10-20 14:51:12 -0400165fn expand_cxx_type(ety: &ExternType) -> TokenStream {
166 let ident = &ety.ident;
167 let doc = &ety.doc;
168 quote! {
169 #doc
170 #[repr(C)]
171 pub struct #ident {
172 _private: ::cxx::private::Opaque,
173 }
174 }
175}
176
177fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
178 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700179 let receiver = efn.receiver.iter().map(|receiver| {
180 let receiver_type = receiver.ty();
181 quote!(_: #receiver_type)
182 });
David Tolnay39d575f2020-03-03 00:10:56 -0800183 let args = efn.args.iter().map(|arg| {
184 let ident = &arg.ident;
185 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800186 if arg.ty == RustString {
187 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700188 } else if let Type::RustVec(_) = arg.ty {
189 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700190 } else if let Type::Fn(_) = arg.ty {
191 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800192 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800193 quote!(#ident: *mut #ty)
194 } else {
195 quote!(#ident: #ty)
196 }
197 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700198 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700199 let ret = if efn.throws {
200 quote!(-> ::cxx::private::Result)
201 } else {
202 expand_extern_return_type(&efn.ret, types)
203 };
David Tolnay7db73692019-10-20 14:51:12 -0400204 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700205 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400206 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
207 outparam = Some(quote!(__return: *mut #ret));
208 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700209 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400210 let local_name = format_ident!("__{}", ident);
211 quote! {
212 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700213 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400214 }
215}
216
217fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
218 let ident = &efn.ident;
219 let doc = &efn.doc;
220 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700221 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700222 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700223 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700224 let var = receiver.var;
225 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700226 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700227 let args = efn.args.iter().map(|arg| quote!(#arg));
228 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700229 let ret = if efn.throws {
230 let ok = match &efn.ret {
231 Some(ret) => quote!(#ret),
232 None => quote!(()),
233 };
234 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
235 } else {
236 expand_return_type(&efn.ret)
237 };
David Tolnay1e548172020-03-16 13:37:09 -0700238 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700239 let receiver_var = efn
240 .receiver
241 .iter()
242 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700243 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400244 let var = &arg.ident;
245 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800246 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800247 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400248 }
249 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
250 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700251 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400252 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800253 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400254 quote!(::cxx::private::RustString::from_ref(#var))
255 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700256 Type::RustVec(_) => quote!(::cxx::private::RustVec::from_ref(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400257 _ => quote!(#var),
258 },
259 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700260 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400261 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
262 _ => quote!(#var),
263 }
264 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700265 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700266 let trampolines = efn
267 .args
268 .iter()
269 .filter_map(|arg| {
270 if let Type::Fn(f) = &arg.ty {
271 let var = &arg.ident;
272 Some(expand_function_pointer_trampoline(
273 namespace, efn, var, f, types,
274 ))
275 } else {
276 None
277 }
278 })
279 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400280 let mut setup = efn
281 .args
282 .iter()
283 .filter(|arg| types.needs_indirect_abi(&arg.ty))
284 .map(|arg| {
285 let var = &arg.ident;
286 // These are arguments for which C++ has taken ownership of the data
287 // behind the mut reference it received.
288 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700289 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400290 }
291 })
292 .collect::<TokenStream>();
293 let local_name = format_ident!("__{}", ident);
294 let call = if indirect_return {
295 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
296 setup.extend(quote! {
297 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400298 });
David Tolnayebef4a22020-03-17 15:33:47 -0700299 if efn.throws {
300 setup.extend(quote! {
301 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
302 });
303 quote!(::std::result::Result::Ok(__return.assume_init()))
304 } else {
305 setup.extend(quote! {
306 #local_name(#(#vars,)* __return.as_mut_ptr());
307 });
308 quote!(__return.assume_init())
309 }
310 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400311 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700312 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400313 }
314 } else {
315 quote! {
316 #local_name(#(#vars),*)
317 }
318 };
Myron Ahn84849302020-03-25 22:00:58 +0700319 let expr = if efn.throws {
320 efn.ret.as_ref().and_then(|ret| match ret {
321 Type::Ident(ident) if ident == RustString => {
322 Some(quote!(#call.map(|r| r.into_string())))
323 }
324 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay8b9d1762020-04-25 16:05:46 -0700325 Type::RustVec(_) => Some(quote!(#call.map(|r| r.into_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700326 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
327 Type::Ref(ty) => match &ty.inner {
328 Type::Ident(ident) if ident == RustString => {
329 Some(quote!(#call.map(|r| r.as_string())))
330 }
David Tolnay77989692020-04-25 15:57:47 -0700331 Type::RustVec(_) => Some(quote!(#call.map(|r| r.as_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700332 _ => None,
333 },
334 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700335 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700336 _ => None,
337 })
338 } else {
339 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800340 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400341 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay03dca702020-04-25 15:08:05 -0700342 Type::RustVec(_) => Some(quote!(#call.into_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400343 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
344 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800345 Type::Ident(ident) if ident == RustString => Some(quote!(#call.as_string())),
David Tolnay77989692020-04-25 15:57:47 -0700346 Type::RustVec(_) => Some(quote!(#call.as_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400347 _ => None,
348 },
349 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700350 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400351 _ => None,
352 })
Myron Ahn84849302020-03-25 22:00:58 +0700353 }
354 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700355 let function_shim = quote! {
356 #doc
357 pub fn #ident(#(#all_args,)*) #ret {
358 extern "C" {
359 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400360 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700361 #trampolines
362 unsafe {
363 #setup
364 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400365 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700366 }
367 };
368 match &efn.receiver {
369 None => function_shim,
370 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700371 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700372 quote!(impl #receiver_type { #function_shim })
373 }
David Tolnay7db73692019-10-20 14:51:12 -0400374 }
375}
376
David Tolnay75dca2e2020-03-25 20:17:52 -0700377fn expand_function_pointer_trampoline(
378 namespace: &Namespace,
379 efn: &ExternFn,
380 var: &Ident,
381 sig: &Signature,
382 types: &Types,
383) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700384 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
385 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700386 let local_name = parse_quote!(__);
387 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
388 let shim = expand_rust_function_shim_impl(
389 sig,
390 types,
391 &r_trampoline,
392 local_name,
393 catch_unwind_label,
394 None,
395 );
396
397 quote! {
398 let #var = ::cxx::private::FatFunction {
399 trampoline: {
400 extern "C" {
401 #[link_name = #c_trampoline]
402 fn trampoline();
403 }
404 #shim
405 trampoline as usize as *const ()
406 },
407 ptr: #var as usize as *const (),
408 };
409 }
410}
411
David Tolnay7db73692019-10-20 14:51:12 -0400412fn expand_rust_type(ety: &ExternType) -> TokenStream {
413 let ident = &ety.ident;
414 quote! {
415 use super::#ident;
416 }
417}
418
419fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
420 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700421 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700422 let local_name = format_ident!("__{}", ident);
423 let catch_unwind_label = format!("::{}", ident);
424 let invoke = Some(ident);
425 expand_rust_function_shim_impl(
426 efn,
427 types,
428 &link_name,
429 local_name,
430 catch_unwind_label,
431 invoke,
432 )
433}
434
435fn expand_rust_function_shim_impl(
436 sig: &Signature,
437 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700438 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700439 local_name: Ident,
440 catch_unwind_label: String,
441 invoke: Option<&Ident>,
442) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700443 let receiver_var = sig
444 .receiver
445 .as_ref()
446 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700447 let receiver = sig.receiver.as_ref().map(|receiver| {
448 let receiver_type = receiver.ty();
449 quote!(#receiver_var: #receiver_type)
450 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700451 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800452 let ident = &arg.ident;
453 let ty = expand_extern_type(&arg.ty);
454 if types.needs_indirect_abi(&arg.ty) {
455 quote!(#ident: *mut #ty)
456 } else {
457 quote!(#ident: #ty)
458 }
459 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700460 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700461
David Tolnay3a45f2d2020-04-20 01:51:12 -0700462 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400463 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800464 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800465 Type::Ident(i) if i == RustString => {
466 quote!(::std::mem::take((*#ident).as_mut_string()))
467 }
David Tolnay40226ab2020-03-03 00:05:35 -0800468 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay83c69e92020-04-25 16:59:39 -0700469 Type::RustVec(_) => quote!(::std::mem::take((*#ident).as_mut_vec())),
David Tolnay40226ab2020-03-03 00:05:35 -0800470 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800471 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800472 Type::Ident(i) if i == RustString => quote!(#ident.as_string()),
David Tolnay83c69e92020-04-25 16:59:39 -0700473 Type::RustVec(_) => quote!(#ident.as_vec()),
David Tolnay40226ab2020-03-03 00:05:35 -0800474 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800475 },
David Tolnay40226ab2020-03-03 00:05:35 -0800476 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700477 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800478 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
479 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400480 }
481 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700482 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700483
484 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700485 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700486 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700487 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700488 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700489 quote!(#receiver_type::#ident)
490 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700491 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700492 None => quote!(__extern),
493 };
494 call.extend(quote! { (#(#vars),*) });
495
496 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400497 .ret
498 .as_ref()
499 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800500 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400501 Some(quote!(::cxx::private::RustString::from(#call)))
502 }
503 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
David Tolnay83c69e92020-04-25 16:59:39 -0700504 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400505 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
506 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800507 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400508 Some(quote!(::cxx::private::RustString::from_ref(#call)))
509 }
David Tolnay83c69e92020-04-25 16:59:39 -0700510 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from_ref(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400511 _ => None,
512 },
513 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700514 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400515 _ => None,
516 })
517 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700518
519 let mut outparam = None;
520 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700521 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700522 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
523 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700524 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700525 if sig.throws {
526 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700527 Some(_) => quote!(__return),
528 None => quote!(&mut ()),
529 };
530 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700531 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400532 expr = quote!(::std::ptr::write(__return, #expr));
533 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700534
David Tolnay1e548172020-03-16 13:37:09 -0700535 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700536
537 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700538 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700539 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700540 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700541 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700542
543 let pointer = match invoke {
544 None => Some(quote!(__extern: #sig)),
545 Some(_) => None,
546 };
547
David Tolnay7db73692019-10-20 14:51:12 -0400548 quote! {
549 #[doc(hidden)]
550 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700551 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400552 let __fn = concat!(module_path!(), #catch_unwind_label);
553 #expr
554 }
555 }
556}
557
558fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700559 let link_prefix = format!("cxxbridge03$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400560 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400561 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400562
563 let local_prefix = format_ident!("{}__box_", ident);
564 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400565 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400566
567 let span = ident.span();
568 quote_spanned! {span=>
569 #[doc(hidden)]
570 #[export_name = #link_uninit]
571 unsafe extern "C" fn #local_uninit(
572 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
573 ) {
574 ::std::ptr::write(
575 this,
576 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
577 );
578 }
579 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400580 #[export_name = #link_drop]
581 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
582 ::std::ptr::drop_in_place(this);
583 }
David Tolnay7db73692019-10-20 14:51:12 -0400584 }
585}
586
David Tolnay83a7ec92020-04-25 12:37:42 -0700587fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700588 let link_prefix = format!("cxxbridge03$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700589 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700590 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700591 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700592 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700593 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700594
David Tolnay83a7ec92020-04-25 12:37:42 -0700595 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700596 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700597 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700598 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700599 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700600 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700601
David Tolnay83a7ec92020-04-25 12:37:42 -0700602 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700603 quote_spanned! {span=>
604 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700605 #[export_name = #link_new]
606 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
607 ::std::ptr::write(this, ::cxx::private::RustVec::new());
608 }
609 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700610 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700611 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700612 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700613 }
David Tolnay85db5a02020-04-25 13:17:27 -0700614 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700615 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700616 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700617 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700618 }
David Tolnay219c0792020-04-24 20:31:37 -0700619 #[doc(hidden)]
620 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700621 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700622 (*this).as_ptr()
623 }
David Tolnay503d0192020-04-24 22:18:56 -0700624 #[doc(hidden)]
625 #[export_name = #link_stride]
626 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700627 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700628 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700629 }
630}
631
David Tolnay3b40b6f2020-04-24 17:58:24 -0700632fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
633 let name = ident.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700634 let prefix = format!("cxxbridge03$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400635 let link_null = format!("{}null", prefix);
636 let link_new = format!("{}new", prefix);
637 let link_raw = format!("{}raw", prefix);
638 let link_get = format!("{}get", prefix);
639 let link_release = format!("{}release", prefix);
640 let link_drop = format!("{}drop", prefix);
641
David Tolnay3b40b6f2020-04-24 17:58:24 -0700642 let new_method = if types.structs.contains_key(ident) {
643 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700644 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
645 extern "C" {
646 #[link_name = #link_new]
647 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
648 }
649 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
650 unsafe { __new(&mut repr, &mut value) }
651 repr
652 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700653 })
654 } else {
655 None
David Tolnay53838912020-04-09 20:56:44 -0700656 };
657
David Tolnay7db73692019-10-20 14:51:12 -0400658 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700659 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
660 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400661 fn __null() -> *mut ::std::ffi::c_void {
662 extern "C" {
663 #[link_name = #link_null]
664 fn __null(this: *mut *mut ::std::ffi::c_void);
665 }
666 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
667 unsafe { __null(&mut repr) }
668 repr
669 }
David Tolnay53838912020-04-09 20:56:44 -0700670 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400671 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
672 extern "C" {
673 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700674 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400675 }
676 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
677 __raw(&mut repr, raw);
678 repr
679 }
680 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
681 extern "C" {
682 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700683 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400684 }
685 __get(&repr)
686 }
687 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
688 extern "C" {
689 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700690 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400691 }
692 __release(&mut repr)
693 }
694 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
695 extern "C" {
696 #[link_name = #link_drop]
697 fn __drop(this: *mut *mut ::std::ffi::c_void);
698 }
699 __drop(&mut repr);
700 }
701 }
702 }
703}
704
David Tolnay17631102020-04-24 19:01:30 -0700705fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700706 let name = elem.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700707 let prefix = format!("cxxbridge03$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700708 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700709 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay69601622020-04-29 18:48:36 -0700710 let unique_ptr_prefix = format!("cxxbridge03$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700711 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
712 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
713 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
714 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
715 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700716
717 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700718 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700719 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700720 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700721 extern "C" {
722 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700723 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700724 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700725 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700726 }
David Tolnay17631102020-04-24 19:01:30 -0700727 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700728 extern "C" {
729 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700730 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700731 }
David Tolnay9626d082020-04-24 14:52:45 -0700732 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700733 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700734 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
735 extern "C" {
736 #[link_name = #link_unique_ptr_null]
737 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
738 }
739 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
740 unsafe { __unique_ptr_null(&mut repr) }
741 repr
742 }
743 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
744 extern "C" {
745 #[link_name = #link_unique_ptr_raw]
746 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
747 }
748 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
749 __unique_ptr_raw(&mut repr, raw);
750 repr
751 }
752 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
753 extern "C" {
754 #[link_name = #link_unique_ptr_get]
755 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
756 }
757 __unique_ptr_get(&repr)
758 }
759 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
760 extern "C" {
761 #[link_name = #link_unique_ptr_release]
762 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
763 }
764 __unique_ptr_release(&mut repr)
765 }
766 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
767 extern "C" {
768 #[link_name = #link_unique_ptr_drop]
769 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
770 }
771 __unique_ptr_drop(&mut repr);
772 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700773 }
774 }
775}
776
David Tolnay7db73692019-10-20 14:51:12 -0400777fn expand_return_type(ret: &Option<Type>) -> TokenStream {
778 match ret {
779 Some(ret) => quote!(-> #ret),
780 None => TokenStream::new(),
781 }
782}
783
David Tolnay75dca2e2020-03-25 20:17:52 -0700784fn indirect_return(sig: &Signature, types: &Types) -> bool {
785 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700786 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700787 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400788}
789
790fn expand_extern_type(ty: &Type) -> TokenStream {
791 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800792 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400793 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700794 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400795 quote!(*mut #inner)
796 }
David Tolnaye6d50212020-04-25 15:38:23 -0700797 Type::RustVec(ty) => {
798 let elem = expand_extern_type(&ty.inner);
799 quote!(::cxx::private::RustVec<#elem>)
800 }
David Tolnay7db73692019-10-20 14:51:12 -0400801 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800802 Type::Ident(ident) if ident == RustString => quote!(&::cxx::private::RustString),
Myron Ahneba35cf2020-02-05 19:41:51 +0700803 Type::RustVec(ty) => {
804 let inner = expand_extern_type(&ty.inner);
David Tolnay6c6b7e02020-04-24 11:42:59 -0700805 quote!(&::cxx::private::RustVec<#inner>)
Myron Ahneba35cf2020-02-05 19:41:51 +0700806 }
David Tolnay7db73692019-10-20 14:51:12 -0400807 _ => quote!(#ty),
808 },
809 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700810 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400811 _ => quote!(#ty),
812 }
813}
814
815fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
816 let ret = match ret {
817 Some(ret) if !types.needs_indirect_abi(ret) => ret,
818 _ => return TokenStream::new(),
819 };
820 let ty = expand_extern_type(ret);
821 quote!(-> #ty)
822}