blob: f28d955f1101d553355f7a2c1b4eba7c2400d712 [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::{
David Tolnay5f9e8ca2020-05-07 17:21:05 -07006 self, check, mangle, Api, Enum, ExternFn, ExternType, Signature, Struct, Type, TypeAlias, 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) {
David Tolnay5f9e8ca2020-05-07 17:21:05 -070049 expanded.extend(expand_cxx_type(namespace, ety));
Joel Galenson905eb2e2020-05-04 14:58:14 -070050 }
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 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -070058 Api::TypeAlias(alias) => {
59 expanded.extend(expand_type_alias(alias));
60 hidden.extend(expand_type_alias_verify(namespace, alias));
61 }
David Tolnay7db73692019-10-20 14:51:12 -040062 }
63 }
64
65 for ty in types {
66 if let Type::RustBox(ty) = ty {
67 if let Type::Ident(ident) = &ty.inner {
68 if Atom::from(ident).is_none() {
69 hidden.extend(expand_rust_box(namespace, ident));
70 }
71 }
Myron Ahneba35cf2020-02-05 19:41:51 +070072 } else if let Type::RustVec(ty) = ty {
73 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070074 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070075 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070076 }
Myron Ahneba35cf2020-02-05 19:41:51 +070077 }
David Tolnay7db73692019-10-20 14:51:12 -040078 } else if let Type::UniquePtr(ptr) = ty {
79 if let Type::Ident(ident) = &ptr.inner {
David Tolnay2b821e62020-05-07 22:55:28 -070080 if Atom::from(ident).is_none() && !types.aliases.contains_key(ident) {
David Tolnay3b40b6f2020-04-24 17:58:24 -070081 expanded.extend(expand_unique_ptr(namespace, ident, types));
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 {
David Tolnay2b821e62020-05-07 22:55:28 -070086 if Atom::from(ident).is_none() && !types.aliases.contains_key(ident) {
David Tolnay17631102020-04-24 19:01:30 -070087 // Generate impl for CxxVector<T> if T is a struct or opaque
88 // C++ type. Impl for primitives is already provided by cxx
89 // crate.
90 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040091 }
92 }
93 }
94 }
95
96 // Work around https://github.com/rust-lang/rust/issues/67851.
97 if !hidden.is_empty() {
98 expanded.extend(quote! {
99 #[doc(hidden)]
100 const _: () = {
101 #hidden
102 };
103 });
104 }
105
106 let attrs = ffi
107 .attrs
108 .into_iter()
109 .filter(|attr| attr.path.is_ident("doc"));
110 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700111 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400112
David Tolnaycbf3f032020-05-03 23:51:14 -0700113 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400114 #(#attrs)*
115 #[deny(improper_ctypes)]
116 #[allow(non_snake_case)]
117 #vis mod #ident {
118 #expanded
119 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700120 }
David Tolnay7db73692019-10-20 14:51:12 -0400121}
122
123fn expand_struct(strct: &Struct) -> TokenStream {
124 let ident = &strct.ident;
125 let doc = &strct.doc;
126 let derives = &strct.derives;
127 let fields = strct.fields.iter().map(|field| {
128 // This span on the pub makes "private type in public interface" errors
129 // appear in the right place.
130 let vis = Token![pub](field.ident.span());
131 quote!(#vis #field)
132 });
133 quote! {
134 #doc
135 #[derive(#(#derives),*)]
136 #[repr(C)]
137 pub struct #ident {
138 #(#fields,)*
139 }
140 }
141}
142
Joel Galensonc03402a2020-04-23 17:31:09 -0700143fn expand_enum(enm: &Enum) -> TokenStream {
144 let ident = &enm.ident;
145 let doc = &enm.doc;
David Tolnayc605e6f2020-05-10 23:37:12 -0700146 let repr = enm.repr;
Joel Galenson88547732020-05-05 08:23:42 -0700147 let variants = enm.variants.iter().map(|variant| {
148 let variant_ident = &variant.ident;
149 let discriminant = &variant.discriminant;
150 Some(quote! {
151 pub const #variant_ident: Self = #ident { repr: #discriminant };
152 })
153 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700154 quote! {
155 #doc
156 #[derive(Copy, Clone, PartialEq, Eq)]
157 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700158 pub struct #ident {
David Tolnayc605e6f2020-05-10 23:37:12 -0700159 pub repr: #repr,
Joel Galensondb1ec312020-05-01 13:57:32 -0700160 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700161
162 #[allow(non_upper_case_globals)]
163 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700164 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700165 }
166 }
167}
168
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700169fn expand_cxx_type(namespace: &Namespace, ety: &ExternType) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400170 let ident = &ety.ident;
171 let doc = &ety.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700172 let type_id = type_id(namespace, ident);
173
David Tolnay7db73692019-10-20 14:51:12 -0400174 quote! {
175 #doc
176 #[repr(C)]
177 pub struct #ident {
178 _private: ::cxx::private::Opaque,
179 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700180
181 unsafe impl ::cxx::ExternType for #ident {
182 type Id = #type_id;
183 }
David Tolnay7db73692019-10-20 14:51:12 -0400184 }
185}
186
187fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
188 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700189 let receiver = efn.receiver.iter().map(|receiver| {
190 let receiver_type = receiver.ty();
191 quote!(_: #receiver_type)
192 });
David Tolnay39d575f2020-03-03 00:10:56 -0800193 let args = efn.args.iter().map(|arg| {
194 let ident = &arg.ident;
195 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800196 if arg.ty == RustString {
197 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700198 } else if let Type::RustVec(_) = arg.ty {
199 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700200 } else if let Type::Fn(_) = arg.ty {
201 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800202 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800203 quote!(#ident: *mut #ty)
204 } else {
205 quote!(#ident: #ty)
206 }
207 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700208 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700209 let ret = if efn.throws {
210 quote!(-> ::cxx::private::Result)
211 } else {
212 expand_extern_return_type(&efn.ret, types)
213 };
David Tolnay7db73692019-10-20 14:51:12 -0400214 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700215 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400216 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
217 outparam = Some(quote!(__return: *mut #ret));
218 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700219 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400220 let local_name = format_ident!("__{}", ident);
221 quote! {
222 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700223 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400224 }
225}
226
227fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
228 let ident = &efn.ident;
229 let doc = &efn.doc;
230 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700231 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700232 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700233 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700234 let var = receiver.var;
235 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700236 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700237 let args = efn.args.iter().map(|arg| quote!(#arg));
238 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700239 let ret = if efn.throws {
240 let ok = match &efn.ret {
241 Some(ret) => quote!(#ret),
242 None => quote!(()),
243 };
244 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
245 } else {
246 expand_return_type(&efn.ret)
247 };
David Tolnay1e548172020-03-16 13:37:09 -0700248 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700249 let receiver_var = efn
250 .receiver
251 .iter()
252 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700253 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400254 let var = &arg.ident;
255 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800256 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800257 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400258 }
259 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
260 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700261 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400262 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800263 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400264 quote!(::cxx::private::RustString::from_ref(#var))
265 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700266 Type::RustVec(_) => quote!(::cxx::private::RustVec::from_ref(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400267 _ => quote!(#var),
268 },
269 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700270 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400271 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
272 _ => quote!(#var),
273 }
274 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700275 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700276 let trampolines = efn
277 .args
278 .iter()
279 .filter_map(|arg| {
280 if let Type::Fn(f) = &arg.ty {
281 let var = &arg.ident;
282 Some(expand_function_pointer_trampoline(
283 namespace, efn, var, f, types,
284 ))
285 } else {
286 None
287 }
288 })
289 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400290 let mut setup = efn
291 .args
292 .iter()
293 .filter(|arg| types.needs_indirect_abi(&arg.ty))
294 .map(|arg| {
295 let var = &arg.ident;
296 // These are arguments for which C++ has taken ownership of the data
297 // behind the mut reference it received.
298 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700299 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400300 }
301 })
302 .collect::<TokenStream>();
303 let local_name = format_ident!("__{}", ident);
304 let call = if indirect_return {
305 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
306 setup.extend(quote! {
307 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400308 });
David Tolnayebef4a22020-03-17 15:33:47 -0700309 if efn.throws {
310 setup.extend(quote! {
311 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
312 });
313 quote!(::std::result::Result::Ok(__return.assume_init()))
314 } else {
315 setup.extend(quote! {
316 #local_name(#(#vars,)* __return.as_mut_ptr());
317 });
318 quote!(__return.assume_init())
319 }
320 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400321 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700322 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400323 }
324 } else {
325 quote! {
326 #local_name(#(#vars),*)
327 }
328 };
Myron Ahn84849302020-03-25 22:00:58 +0700329 let expr = if efn.throws {
330 efn.ret.as_ref().and_then(|ret| match ret {
331 Type::Ident(ident) if ident == RustString => {
332 Some(quote!(#call.map(|r| r.into_string())))
333 }
334 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay8b9d1762020-04-25 16:05:46 -0700335 Type::RustVec(_) => Some(quote!(#call.map(|r| r.into_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700336 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
337 Type::Ref(ty) => match &ty.inner {
338 Type::Ident(ident) if ident == RustString => {
339 Some(quote!(#call.map(|r| r.as_string())))
340 }
David Tolnay77989692020-04-25 15:57:47 -0700341 Type::RustVec(_) => Some(quote!(#call.map(|r| r.as_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700342 _ => None,
343 },
344 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700345 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700346 _ => None,
347 })
348 } else {
349 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800350 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400351 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay03dca702020-04-25 15:08:05 -0700352 Type::RustVec(_) => Some(quote!(#call.into_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400353 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
354 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800355 Type::Ident(ident) if ident == RustString => Some(quote!(#call.as_string())),
David Tolnay77989692020-04-25 15:57:47 -0700356 Type::RustVec(_) => Some(quote!(#call.as_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400357 _ => None,
358 },
359 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700360 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400361 _ => None,
362 })
Myron Ahn84849302020-03-25 22:00:58 +0700363 }
364 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700365 let function_shim = quote! {
366 #doc
367 pub fn #ident(#(#all_args,)*) #ret {
368 extern "C" {
369 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400370 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700371 #trampolines
372 unsafe {
373 #setup
374 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400375 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700376 }
377 };
378 match &efn.receiver {
379 None => function_shim,
380 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700381 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700382 quote!(impl #receiver_type { #function_shim })
383 }
David Tolnay7db73692019-10-20 14:51:12 -0400384 }
385}
386
David Tolnay75dca2e2020-03-25 20:17:52 -0700387fn expand_function_pointer_trampoline(
388 namespace: &Namespace,
389 efn: &ExternFn,
390 var: &Ident,
391 sig: &Signature,
392 types: &Types,
393) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700394 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
395 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700396 let local_name = parse_quote!(__);
397 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
398 let shim = expand_rust_function_shim_impl(
399 sig,
400 types,
401 &r_trampoline,
402 local_name,
403 catch_unwind_label,
404 None,
405 );
406
407 quote! {
408 let #var = ::cxx::private::FatFunction {
409 trampoline: {
410 extern "C" {
411 #[link_name = #c_trampoline]
412 fn trampoline();
413 }
414 #shim
415 trampoline as usize as *const ()
416 },
417 ptr: #var as usize as *const (),
418 };
419 }
420}
421
David Tolnay7db73692019-10-20 14:51:12 -0400422fn expand_rust_type(ety: &ExternType) -> TokenStream {
423 let ident = &ety.ident;
424 quote! {
425 use super::#ident;
426 }
427}
428
429fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
430 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700431 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700432 let local_name = format_ident!("__{}", ident);
433 let catch_unwind_label = format!("::{}", ident);
434 let invoke = Some(ident);
435 expand_rust_function_shim_impl(
436 efn,
437 types,
438 &link_name,
439 local_name,
440 catch_unwind_label,
441 invoke,
442 )
443}
444
445fn expand_rust_function_shim_impl(
446 sig: &Signature,
447 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700448 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700449 local_name: Ident,
450 catch_unwind_label: String,
451 invoke: Option<&Ident>,
452) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700453 let receiver_var = sig
454 .receiver
455 .as_ref()
456 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700457 let receiver = sig.receiver.as_ref().map(|receiver| {
458 let receiver_type = receiver.ty();
459 quote!(#receiver_var: #receiver_type)
460 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700461 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800462 let ident = &arg.ident;
463 let ty = expand_extern_type(&arg.ty);
464 if types.needs_indirect_abi(&arg.ty) {
465 quote!(#ident: *mut #ty)
466 } else {
467 quote!(#ident: #ty)
468 }
469 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700470 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700471
David Tolnay3a45f2d2020-04-20 01:51:12 -0700472 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400473 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800474 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800475 Type::Ident(i) if i == RustString => {
476 quote!(::std::mem::take((*#ident).as_mut_string()))
477 }
David Tolnay40226ab2020-03-03 00:05:35 -0800478 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay83c69e92020-04-25 16:59:39 -0700479 Type::RustVec(_) => quote!(::std::mem::take((*#ident).as_mut_vec())),
David Tolnay40226ab2020-03-03 00:05:35 -0800480 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800481 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800482 Type::Ident(i) if i == RustString => quote!(#ident.as_string()),
David Tolnay83c69e92020-04-25 16:59:39 -0700483 Type::RustVec(_) => quote!(#ident.as_vec()),
David Tolnay40226ab2020-03-03 00:05:35 -0800484 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800485 },
David Tolnay40226ab2020-03-03 00:05:35 -0800486 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700487 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800488 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
489 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400490 }
491 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700492 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700493
494 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700495 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700496 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700497 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700498 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700499 quote!(#receiver_type::#ident)
500 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700501 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700502 None => quote!(__extern),
503 };
504 call.extend(quote! { (#(#vars),*) });
505
506 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400507 .ret
508 .as_ref()
509 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800510 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400511 Some(quote!(::cxx::private::RustString::from(#call)))
512 }
513 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
David Tolnay83c69e92020-04-25 16:59:39 -0700514 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400515 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
516 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800517 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400518 Some(quote!(::cxx::private::RustString::from_ref(#call)))
519 }
David Tolnay83c69e92020-04-25 16:59:39 -0700520 Type::RustVec(_) => Some(quote!(::cxx::private::RustVec::from_ref(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400521 _ => None,
522 },
523 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700524 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400525 _ => None,
526 })
527 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700528
529 let mut outparam = None;
530 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700531 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700532 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
533 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700534 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700535 if sig.throws {
536 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700537 Some(_) => quote!(__return),
538 None => quote!(&mut ()),
539 };
540 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700541 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400542 expr = quote!(::std::ptr::write(__return, #expr));
543 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700544
David Tolnay1e548172020-03-16 13:37:09 -0700545 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700546
547 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700548 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700549 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700550 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700551 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700552
553 let pointer = match invoke {
554 None => Some(quote!(__extern: #sig)),
555 Some(_) => None,
556 };
557
David Tolnay7db73692019-10-20 14:51:12 -0400558 quote! {
559 #[doc(hidden)]
560 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700561 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400562 let __fn = concat!(module_path!(), #catch_unwind_label);
563 #expr
564 }
565 }
566}
567
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700568fn expand_type_alias(alias: &TypeAlias) -> TokenStream {
569 let ident = &alias.ident;
570 let ty = &alias.ty;
571 quote! {
572 pub type #ident = #ty;
573 }
574}
575
576fn expand_type_alias_verify(namespace: &Namespace, alias: &TypeAlias) -> TokenStream {
577 let ident = &alias.ident;
578 let type_id = type_id(namespace, ident);
David Tolnay83fe0f02020-05-07 20:00:15 -0700579 let begin_span = alias.type_token.span;
580 let end_span = alias.semi_token.span;
581 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_type::<);
582 let end = quote_spanned!(end_span=> >);
583
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700584 quote! {
David Tolnay83fe0f02020-05-07 20:00:15 -0700585 const _: fn() = #begin #ident, #type_id #end;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700586 }
587}
588
589fn type_id(namespace: &Namespace, ident: &Ident) -> TokenStream {
590 let mut path = String::new();
591 for name in namespace {
592 path += &name.to_string();
593 path += "::";
594 }
595 path += &ident.to_string();
596
597 quote! {
598 ::cxx::type_id!(#path)
599 }
600}
601
David Tolnay7db73692019-10-20 14:51:12 -0400602fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700603 let link_prefix = format!("cxxbridge03$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400604 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400605 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400606
607 let local_prefix = format_ident!("{}__box_", ident);
608 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400609 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400610
611 let span = ident.span();
612 quote_spanned! {span=>
613 #[doc(hidden)]
614 #[export_name = #link_uninit]
615 unsafe extern "C" fn #local_uninit(
616 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
617 ) {
618 ::std::ptr::write(
619 this,
620 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
621 );
622 }
623 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400624 #[export_name = #link_drop]
625 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
626 ::std::ptr::drop_in_place(this);
627 }
David Tolnay7db73692019-10-20 14:51:12 -0400628 }
629}
630
David Tolnay83a7ec92020-04-25 12:37:42 -0700631fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700632 let link_prefix = format!("cxxbridge03$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700633 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700634 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700635 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700636 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700637 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700638
David Tolnay83a7ec92020-04-25 12:37:42 -0700639 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700640 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700641 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700642 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700643 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700644 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700645
David Tolnay83a7ec92020-04-25 12:37:42 -0700646 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700647 quote_spanned! {span=>
648 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700649 #[export_name = #link_new]
650 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
651 ::std::ptr::write(this, ::cxx::private::RustVec::new());
652 }
653 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700654 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700655 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700656 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700657 }
David Tolnay85db5a02020-04-25 13:17:27 -0700658 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700659 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700660 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700661 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700662 }
David Tolnay219c0792020-04-24 20:31:37 -0700663 #[doc(hidden)]
664 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700665 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700666 (*this).as_ptr()
667 }
David Tolnay503d0192020-04-24 22:18:56 -0700668 #[doc(hidden)]
669 #[export_name = #link_stride]
670 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700671 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700672 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700673 }
674}
675
David Tolnay3b40b6f2020-04-24 17:58:24 -0700676fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
677 let name = ident.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700678 let prefix = format!("cxxbridge03$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400679 let link_null = format!("{}null", prefix);
680 let link_new = format!("{}new", prefix);
681 let link_raw = format!("{}raw", prefix);
682 let link_get = format!("{}get", prefix);
683 let link_release = format!("{}release", prefix);
684 let link_drop = format!("{}drop", prefix);
685
David Tolnay3b40b6f2020-04-24 17:58:24 -0700686 let new_method = if types.structs.contains_key(ident) {
687 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700688 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
689 extern "C" {
690 #[link_name = #link_new]
691 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
692 }
693 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
694 unsafe { __new(&mut repr, &mut value) }
695 repr
696 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700697 })
698 } else {
699 None
David Tolnay53838912020-04-09 20:56:44 -0700700 };
701
David Tolnay7db73692019-10-20 14:51:12 -0400702 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700703 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
704 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400705 fn __null() -> *mut ::std::ffi::c_void {
706 extern "C" {
707 #[link_name = #link_null]
708 fn __null(this: *mut *mut ::std::ffi::c_void);
709 }
710 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
711 unsafe { __null(&mut repr) }
712 repr
713 }
David Tolnay53838912020-04-09 20:56:44 -0700714 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400715 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
716 extern "C" {
717 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700718 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400719 }
720 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
721 __raw(&mut repr, raw);
722 repr
723 }
724 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
725 extern "C" {
726 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700727 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400728 }
729 __get(&repr)
730 }
731 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
732 extern "C" {
733 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700734 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400735 }
736 __release(&mut repr)
737 }
738 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
739 extern "C" {
740 #[link_name = #link_drop]
741 fn __drop(this: *mut *mut ::std::ffi::c_void);
742 }
743 __drop(&mut repr);
744 }
745 }
746 }
747}
748
David Tolnay17631102020-04-24 19:01:30 -0700749fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700750 let name = elem.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700751 let prefix = format!("cxxbridge03$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700752 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700753 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay69601622020-04-29 18:48:36 -0700754 let unique_ptr_prefix = format!("cxxbridge03$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700755 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
756 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
757 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
758 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
759 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700760
761 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700762 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700763 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700764 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700765 extern "C" {
766 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700767 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700768 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700769 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700770 }
David Tolnay17631102020-04-24 19:01:30 -0700771 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700772 extern "C" {
773 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700774 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700775 }
David Tolnay9626d082020-04-24 14:52:45 -0700776 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700777 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700778 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
779 extern "C" {
780 #[link_name = #link_unique_ptr_null]
781 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
782 }
783 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
784 unsafe { __unique_ptr_null(&mut repr) }
785 repr
786 }
787 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
788 extern "C" {
789 #[link_name = #link_unique_ptr_raw]
790 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
791 }
792 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
793 __unique_ptr_raw(&mut repr, raw);
794 repr
795 }
796 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
797 extern "C" {
798 #[link_name = #link_unique_ptr_get]
799 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
800 }
801 __unique_ptr_get(&repr)
802 }
803 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
804 extern "C" {
805 #[link_name = #link_unique_ptr_release]
806 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
807 }
808 __unique_ptr_release(&mut repr)
809 }
810 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
811 extern "C" {
812 #[link_name = #link_unique_ptr_drop]
813 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
814 }
815 __unique_ptr_drop(&mut repr);
816 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700817 }
818 }
819}
820
David Tolnay7db73692019-10-20 14:51:12 -0400821fn expand_return_type(ret: &Option<Type>) -> TokenStream {
822 match ret {
823 Some(ret) => quote!(-> #ret),
824 None => TokenStream::new(),
825 }
826}
827
David Tolnay75dca2e2020-03-25 20:17:52 -0700828fn indirect_return(sig: &Signature, types: &Types) -> bool {
829 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700830 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700831 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400832}
833
834fn expand_extern_type(ty: &Type) -> TokenStream {
835 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800836 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400837 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700838 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400839 quote!(*mut #inner)
840 }
David Tolnaye6d50212020-04-25 15:38:23 -0700841 Type::RustVec(ty) => {
842 let elem = expand_extern_type(&ty.inner);
843 quote!(::cxx::private::RustVec<#elem>)
844 }
David Tolnay7db73692019-10-20 14:51:12 -0400845 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800846 Type::Ident(ident) if ident == RustString => quote!(&::cxx::private::RustString),
Myron Ahneba35cf2020-02-05 19:41:51 +0700847 Type::RustVec(ty) => {
848 let inner = expand_extern_type(&ty.inner);
David Tolnay6c6b7e02020-04-24 11:42:59 -0700849 quote!(&::cxx::private::RustVec<#inner>)
Myron Ahneba35cf2020-02-05 19:41:51 +0700850 }
David Tolnay7db73692019-10-20 14:51:12 -0400851 _ => quote!(#ty),
852 },
853 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700854 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400855 _ => quote!(#ty),
856 }
857}
858
859fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
860 let ret = match ret {
861 Some(ret) if !types.needs_indirect_abi(ret) => ret,
862 _ => return TokenStream::new(),
863 };
864 let ty = expand_extern_type(ret);
865 quote!(-> #ty)
866}