blob: 7503f3a4e125c07eef33d6f729000d289adba7ed [file] [log] [blame]
David Tolnaya52602b2020-03-06 10:24:34 -08001use crate::syntax::atom::Atom::{self, *};
David Tolnayf12e9832020-04-24 18:46:44 -07002use crate::syntax::mangled::to_mangled;
David Tolnay08419302020-04-19 20:38:20 -07003use crate::syntax::namespace::Namespace;
David Tolnay891061b2020-04-19 22:42:33 -07004use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07005use crate::syntax::{
6 self, check, mangle, Api, ExternFn, ExternType, Signature, Struct, Type, Types,
7};
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};
Myron Ahneba35cf2020-02-05 19:41:51 +070010use syn::{parse_quote, spanned::Spanned, Error, ItemMod, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040011
12pub fn bridge(namespace: &Namespace, ffi: ItemMod) -> Result<TokenStream> {
13 let ident = &ffi.ident;
14 let content = ffi.content.ok_or(Error::new(
15 Span::call_site(),
16 "#[cxx::bridge] module must have inline contents",
17 ))?;
18 let apis = syntax::parse_items(content.1)?;
19 let ref types = Types::collect(&apis)?;
20 check::typecheck(&apis, types)?;
21
22 let mut expanded = TokenStream::new();
23 let mut hidden = TokenStream::new();
David Tolnay7db73692019-10-20 14:51:12 -040024
25 for api in &apis {
26 if let Api::RustType(ety) = api {
27 expanded.extend(expand_rust_type(ety));
David Tolnay7db73692019-10-20 14:51:12 -040028 let ident = &ety.ident;
David Tolnay4b07ab92020-04-22 19:50:21 -070029 let span = ident.span();
David Tolnay1044d442020-04-22 19:58:30 -070030 hidden.extend(quote_spanned! {span=>
31 let _ = ::std::ptr::read::<#ident>;
32 });
David Tolnay7db73692019-10-20 14:51:12 -040033 }
34 }
35
36 for api in &apis {
37 match api {
38 Api::Include(_) | Api::RustType(_) => {}
39 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
40 Api::CxxType(ety) => expanded.extend(expand_cxx_type(ety)),
41 Api::CxxFunction(efn) => {
42 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
43 }
44 Api::RustFunction(efn) => {
45 hidden.extend(expand_rust_function_shim(namespace, efn, types))
46 }
47 }
48 }
49
50 for ty in types {
51 if let Type::RustBox(ty) = ty {
52 if let Type::Ident(ident) = &ty.inner {
53 if Atom::from(ident).is_none() {
54 hidden.extend(expand_rust_box(namespace, ident));
55 }
56 }
Myron Ahneba35cf2020-02-05 19:41:51 +070057 } else if let Type::RustVec(ty) = ty {
58 if let Type::Ident(ident) = &ty.inner {
59 hidden.extend(expand_rust_vec(namespace, &ty.inner, ident));
60 }
David Tolnay7db73692019-10-20 14:51:12 -040061 } else if let Type::UniquePtr(ptr) = ty {
62 if let Type::Ident(ident) = &ptr.inner {
63 if Atom::from(ident).is_none() {
David Tolnay3b40b6f2020-04-24 17:58:24 -070064 expanded.extend(expand_unique_ptr(namespace, ident, types));
Myron Ahneba35cf2020-02-05 19:41:51 +070065 }
66 }
David Tolnay4377a9e2020-04-24 15:20:26 -070067 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070068 if let Type::Ident(ident) = &ptr.inner {
69 if Atom::from(ident).is_none() {
David Tolnay17631102020-04-24 19:01:30 -070070 // Generate impl for CxxVector<T> if T is a struct or opaque
71 // C++ type. Impl for primitives is already provided by cxx
72 // crate.
73 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040074 }
75 }
76 }
77 }
78
79 // Work around https://github.com/rust-lang/rust/issues/67851.
80 if !hidden.is_empty() {
81 expanded.extend(quote! {
82 #[doc(hidden)]
83 const _: () = {
84 #hidden
85 };
86 });
87 }
88
89 let attrs = ffi
90 .attrs
91 .into_iter()
92 .filter(|attr| attr.path.is_ident("doc"));
93 let vis = &ffi.vis;
94
95 Ok(quote! {
96 #(#attrs)*
97 #[deny(improper_ctypes)]
98 #[allow(non_snake_case)]
99 #vis mod #ident {
100 #expanded
101 }
102 })
103}
104
105fn expand_struct(strct: &Struct) -> TokenStream {
106 let ident = &strct.ident;
107 let doc = &strct.doc;
108 let derives = &strct.derives;
109 let fields = strct.fields.iter().map(|field| {
110 // This span on the pub makes "private type in public interface" errors
111 // appear in the right place.
112 let vis = Token![pub](field.ident.span());
113 quote!(#vis #field)
114 });
115 quote! {
116 #doc
117 #[derive(#(#derives),*)]
118 #[repr(C)]
119 pub struct #ident {
120 #(#fields,)*
121 }
122 }
123}
124
125fn expand_cxx_type(ety: &ExternType) -> TokenStream {
126 let ident = &ety.ident;
127 let doc = &ety.doc;
128 quote! {
129 #doc
130 #[repr(C)]
131 pub struct #ident {
132 _private: ::cxx::private::Opaque,
133 }
134 }
135}
136
137fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
138 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700139 let receiver = efn.receiver.iter().map(|receiver| {
140 let receiver_type = receiver.ty();
141 quote!(_: #receiver_type)
142 });
David Tolnay39d575f2020-03-03 00:10:56 -0800143 let args = efn.args.iter().map(|arg| {
144 let ident = &arg.ident;
145 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800146 if arg.ty == RustString {
147 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700148 } else if let Type::Fn(_) = arg.ty {
149 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800150 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800151 quote!(#ident: *mut #ty)
152 } else {
153 quote!(#ident: #ty)
154 }
155 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700156 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700157 let ret = if efn.throws {
158 quote!(-> ::cxx::private::Result)
159 } else {
160 expand_extern_return_type(&efn.ret, types)
161 };
David Tolnay7db73692019-10-20 14:51:12 -0400162 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700163 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400164 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
165 outparam = Some(quote!(__return: *mut #ret));
166 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700167 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400168 let local_name = format_ident!("__{}", ident);
169 quote! {
170 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700171 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400172 }
173}
174
175fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
176 let ident = &efn.ident;
177 let doc = &efn.doc;
178 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700179 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700180 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700181 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700182 let var = receiver.var;
183 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700184 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700185 let args = efn.args.iter().map(|arg| quote!(#arg));
186 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700187 let ret = if efn.throws {
188 let ok = match &efn.ret {
189 Some(ret) => quote!(#ret),
190 None => quote!(()),
191 };
192 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
193 } else {
194 expand_return_type(&efn.ret)
195 };
David Tolnay1e548172020-03-16 13:37:09 -0700196 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700197 let receiver_var = efn
198 .receiver
199 .iter()
200 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700201 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400202 let var = &arg.ident;
203 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800204 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800205 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400206 }
207 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
208 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay6c6b7e02020-04-24 11:42:59 -0700209 Type::RustVec(_) => quote!(::cxx::private::RustVec::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400210 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800211 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400212 quote!(::cxx::private::RustString::from_ref(#var))
213 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700214 Type::RustVec(_) => quote!(::cxx::private::RustVec::from_ref(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400215 _ => quote!(#var),
216 },
217 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700218 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400219 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
220 _ => quote!(#var),
221 }
222 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700223 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700224 let trampolines = efn
225 .args
226 .iter()
227 .filter_map(|arg| {
228 if let Type::Fn(f) = &arg.ty {
229 let var = &arg.ident;
230 Some(expand_function_pointer_trampoline(
231 namespace, efn, var, f, types,
232 ))
233 } else {
234 None
235 }
236 })
237 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400238 let mut setup = efn
239 .args
240 .iter()
241 .filter(|arg| types.needs_indirect_abi(&arg.ty))
242 .map(|arg| {
243 let var = &arg.ident;
244 // These are arguments for which C++ has taken ownership of the data
245 // behind the mut reference it received.
246 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700247 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400248 }
249 })
250 .collect::<TokenStream>();
251 let local_name = format_ident!("__{}", ident);
252 let call = if indirect_return {
253 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
254 setup.extend(quote! {
255 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400256 });
David Tolnayebef4a22020-03-17 15:33:47 -0700257 if efn.throws {
258 setup.extend(quote! {
259 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
260 });
261 quote!(::std::result::Result::Ok(__return.assume_init()))
262 } else {
263 setup.extend(quote! {
264 #local_name(#(#vars,)* __return.as_mut_ptr());
265 });
266 quote!(__return.assume_init())
267 }
268 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400269 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700270 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400271 }
272 } else {
273 quote! {
274 #local_name(#(#vars),*)
275 }
276 };
Myron Ahn84849302020-03-25 22:00:58 +0700277 let expr = if efn.throws {
278 efn.ret.as_ref().and_then(|ret| match ret {
279 Type::Ident(ident) if ident == RustString => {
280 Some(quote!(#call.map(|r| r.into_string())))
281 }
282 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
283 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
284 Type::Ref(ty) => match &ty.inner {
285 Type::Ident(ident) if ident == RustString => {
286 Some(quote!(#call.map(|r| r.as_string())))
287 }
288 _ => None,
289 },
290 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700291 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700292 _ => None,
293 })
294 } else {
295 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800296 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400297 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
298 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
299 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800300 Type::Ident(ident) if ident == RustString => Some(quote!(#call.as_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400301 _ => None,
302 },
303 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700304 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400305 _ => None,
306 })
Myron Ahn84849302020-03-25 22:00:58 +0700307 }
308 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700309 let function_shim = quote! {
310 #doc
311 pub fn #ident(#(#all_args,)*) #ret {
312 extern "C" {
313 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400314 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700315 #trampolines
316 unsafe {
317 #setup
318 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400319 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700320 }
321 };
322 match &efn.receiver {
323 None => function_shim,
324 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700325 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700326 quote!(impl #receiver_type { #function_shim })
327 }
David Tolnay7db73692019-10-20 14:51:12 -0400328 }
329}
330
David Tolnay75dca2e2020-03-25 20:17:52 -0700331fn expand_function_pointer_trampoline(
332 namespace: &Namespace,
333 efn: &ExternFn,
334 var: &Ident,
335 sig: &Signature,
336 types: &Types,
337) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700338 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
339 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700340 let local_name = parse_quote!(__);
341 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
342 let shim = expand_rust_function_shim_impl(
343 sig,
344 types,
345 &r_trampoline,
346 local_name,
347 catch_unwind_label,
348 None,
349 );
350
351 quote! {
352 let #var = ::cxx::private::FatFunction {
353 trampoline: {
354 extern "C" {
355 #[link_name = #c_trampoline]
356 fn trampoline();
357 }
358 #shim
359 trampoline as usize as *const ()
360 },
361 ptr: #var as usize as *const (),
362 };
363 }
364}
365
David Tolnay7db73692019-10-20 14:51:12 -0400366fn expand_rust_type(ety: &ExternType) -> TokenStream {
367 let ident = &ety.ident;
368 quote! {
369 use super::#ident;
370 }
371}
372
373fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
374 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700375 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700376 let local_name = format_ident!("__{}", ident);
377 let catch_unwind_label = format!("::{}", ident);
378 let invoke = Some(ident);
379 expand_rust_function_shim_impl(
380 efn,
381 types,
382 &link_name,
383 local_name,
384 catch_unwind_label,
385 invoke,
386 )
387}
388
389fn expand_rust_function_shim_impl(
390 sig: &Signature,
391 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700392 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700393 local_name: Ident,
394 catch_unwind_label: String,
395 invoke: Option<&Ident>,
396) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700397 let receiver_var = sig
398 .receiver
399 .as_ref()
400 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700401 let receiver = sig.receiver.as_ref().map(|receiver| {
402 let receiver_type = receiver.ty();
403 quote!(#receiver_var: #receiver_type)
404 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700405 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800406 let ident = &arg.ident;
407 let ty = expand_extern_type(&arg.ty);
408 if types.needs_indirect_abi(&arg.ty) {
409 quote!(#ident: *mut #ty)
410 } else {
411 quote!(#ident: #ty)
412 }
413 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700414 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700415
David Tolnay3a45f2d2020-04-20 01:51:12 -0700416 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400417 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800418 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800419 Type::Ident(i) if i == RustString => {
420 quote!(::std::mem::take((*#ident).as_mut_string()))
421 }
David Tolnay40226ab2020-03-03 00:05:35 -0800422 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
423 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800424 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800425 Type::Ident(i) if i == RustString => quote!(#ident.as_string()),
David Tolnay40226ab2020-03-03 00:05:35 -0800426 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800427 },
David Tolnay40226ab2020-03-03 00:05:35 -0800428 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700429 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800430 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
431 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400432 }
433 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700434 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700435
436 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700437 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700438 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700439 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700440 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700441 quote!(#receiver_type::#ident)
442 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700443 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700444 None => quote!(__extern),
445 };
446 call.extend(quote! { (#(#vars),*) });
447
448 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400449 .ret
450 .as_ref()
451 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800452 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400453 Some(quote!(::cxx::private::RustString::from(#call)))
454 }
455 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
456 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
457 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800458 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400459 Some(quote!(::cxx::private::RustString::from_ref(#call)))
460 }
461 _ => None,
462 },
463 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700464 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400465 _ => None,
466 })
467 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700468
469 let mut outparam = None;
470 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700471 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700472 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
473 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700474 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700475 if sig.throws {
476 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700477 Some(_) => quote!(__return),
478 None => quote!(&mut ()),
479 };
480 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700481 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400482 expr = quote!(::std::ptr::write(__return, #expr));
483 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700484
David Tolnay1e548172020-03-16 13:37:09 -0700485 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700486
487 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700488 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700489 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700490 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700491 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700492
493 let pointer = match invoke {
494 None => Some(quote!(__extern: #sig)),
495 Some(_) => None,
496 };
497
David Tolnay7db73692019-10-20 14:51:12 -0400498 quote! {
499 #[doc(hidden)]
500 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700501 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400502 let __fn = concat!(module_path!(), #catch_unwind_label);
503 #expr
504 }
505 }
506}
507
508fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay8c730492020-03-13 01:29:06 -0700509 let link_prefix = format!("cxxbridge02$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400510 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400511 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400512
513 let local_prefix = format_ident!("{}__box_", ident);
514 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400515 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400516
517 let span = ident.span();
518 quote_spanned! {span=>
519 #[doc(hidden)]
520 #[export_name = #link_uninit]
521 unsafe extern "C" fn #local_uninit(
522 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
523 ) {
524 ::std::ptr::write(
525 this,
526 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
527 );
528 }
529 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400530 #[export_name = #link_drop]
531 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
532 ::std::ptr::drop_in_place(this);
533 }
David Tolnay7db73692019-10-20 14:51:12 -0400534 }
535}
536
Myron Ahneba35cf2020-02-05 19:41:51 +0700537fn expand_rust_vec(namespace: &Namespace, ty: &Type, ident: &Ident) -> TokenStream {
538 let inner = ty;
David Tolnayf12e9832020-04-24 18:46:44 -0700539 let mangled = to_mangled(namespace, ty) + "$";
Myron Ahneba35cf2020-02-05 19:41:51 +0700540 let link_prefix = format!("cxxbridge02$rust_vec${}", mangled);
541 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700542 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700543 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700544 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700545
546 let local_prefix = format_ident!("{}__vec_", ident);
547 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700548 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700549 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700550 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700551
552 let span = ty.span();
553 quote_spanned! {span=>
554 #[doc(hidden)]
555 #[export_name = #link_drop]
David Tolnay6c6b7e02020-04-24 11:42:59 -0700556 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#inner>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700557 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700558 }
David Tolnay85db5a02020-04-25 13:17:27 -0700559 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700560 #[export_name = #link_len]
David Tolnay6c6b7e02020-04-24 11:42:59 -0700561 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#inner>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700562 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700563 }
David Tolnay219c0792020-04-24 20:31:37 -0700564 #[doc(hidden)]
565 #[export_name = #link_data]
566 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#inner>) -> *const #inner {
567 (*this).as_ptr()
568 }
David Tolnay503d0192020-04-24 22:18:56 -0700569 #[doc(hidden)]
570 #[export_name = #link_stride]
571 unsafe extern "C" fn #local_stride() -> usize {
572 ::std::mem::size_of::<#inner>()
573 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700574 }
575}
576
David Tolnay3b40b6f2020-04-24 17:58:24 -0700577fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
578 let name = ident.to_string();
579 let prefix = format!("cxxbridge02$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400580 let link_null = format!("{}null", prefix);
581 let link_new = format!("{}new", prefix);
582 let link_raw = format!("{}raw", prefix);
583 let link_get = format!("{}get", prefix);
584 let link_release = format!("{}release", prefix);
585 let link_drop = format!("{}drop", prefix);
586
David Tolnay3b40b6f2020-04-24 17:58:24 -0700587 let new_method = if types.structs.contains_key(ident) {
588 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700589 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
590 extern "C" {
591 #[link_name = #link_new]
592 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
593 }
594 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
595 unsafe { __new(&mut repr, &mut value) }
596 repr
597 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700598 })
599 } else {
600 None
David Tolnay53838912020-04-09 20:56:44 -0700601 };
602
David Tolnay7db73692019-10-20 14:51:12 -0400603 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700604 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
605 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400606 fn __null() -> *mut ::std::ffi::c_void {
607 extern "C" {
608 #[link_name = #link_null]
609 fn __null(this: *mut *mut ::std::ffi::c_void);
610 }
611 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
612 unsafe { __null(&mut repr) }
613 repr
614 }
David Tolnay53838912020-04-09 20:56:44 -0700615 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400616 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
617 extern "C" {
618 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700619 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400620 }
621 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
622 __raw(&mut repr, raw);
623 repr
624 }
625 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
626 extern "C" {
627 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700628 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400629 }
630 __get(&repr)
631 }
632 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
633 extern "C" {
634 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700635 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400636 }
637 __release(&mut repr)
638 }
639 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
640 extern "C" {
641 #[link_name = #link_drop]
642 fn __drop(this: *mut *mut ::std::ffi::c_void);
643 }
644 __drop(&mut repr);
645 }
646 }
647 }
648}
649
David Tolnay17631102020-04-24 19:01:30 -0700650fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700651 let name = elem.to_string();
David Tolnay17631102020-04-24 19:01:30 -0700652 let prefix = format!("cxxbridge02$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700653 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700654 let link_get_unchecked = format!("{}get_unchecked", prefix);
655 let link_push_back = format!("{}push_back", prefix);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700656 let unique_ptr_prefix = format!("cxxbridge02$unique_ptr$std$vector${}{}$", namespace, elem);
657 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
658 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
659 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
660 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
661 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700662
663 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700664 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700665 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700666 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700667 extern "C" {
668 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700669 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700670 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700671 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700672 }
David Tolnay17631102020-04-24 19:01:30 -0700673 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700674 extern "C" {
675 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700676 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700677 }
David Tolnay9626d082020-04-24 14:52:45 -0700678 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700679 }
David Tolnay17631102020-04-24 19:01:30 -0700680 fn __push_back(v: &::cxx::CxxVector<Self>, item: &Self) {
David Tolnayfa2119c2020-04-24 14:42:58 -0700681 extern "C" {
682 #[link_name = #link_push_back]
David Tolnay17631102020-04-24 19:01:30 -0700683 fn __push_back(_: &::cxx::CxxVector<#elem>, _: &#elem);
Myron Ahneba35cf2020-02-05 19:41:51 +0700684 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700685 unsafe { __push_back(v, item) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700686 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700687 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
688 extern "C" {
689 #[link_name = #link_unique_ptr_null]
690 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
691 }
692 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
693 unsafe { __unique_ptr_null(&mut repr) }
694 repr
695 }
696 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
697 extern "C" {
698 #[link_name = #link_unique_ptr_raw]
699 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
700 }
701 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
702 __unique_ptr_raw(&mut repr, raw);
703 repr
704 }
705 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
706 extern "C" {
707 #[link_name = #link_unique_ptr_get]
708 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
709 }
710 __unique_ptr_get(&repr)
711 }
712 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
713 extern "C" {
714 #[link_name = #link_unique_ptr_release]
715 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
716 }
717 __unique_ptr_release(&mut repr)
718 }
719 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
720 extern "C" {
721 #[link_name = #link_unique_ptr_drop]
722 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
723 }
724 __unique_ptr_drop(&mut repr);
725 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700726 }
727 }
728}
729
David Tolnay7db73692019-10-20 14:51:12 -0400730fn expand_return_type(ret: &Option<Type>) -> TokenStream {
731 match ret {
732 Some(ret) => quote!(-> #ret),
733 None => TokenStream::new(),
734 }
735}
736
David Tolnay75dca2e2020-03-25 20:17:52 -0700737fn indirect_return(sig: &Signature, types: &Types) -> bool {
738 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700739 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700740 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400741}
742
743fn expand_extern_type(ty: &Type) -> TokenStream {
744 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800745 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400746 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700747 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400748 quote!(*mut #inner)
749 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700750 Type::RustVec(ty) => quote!(::cxx::private::RustVec<#ty>),
David Tolnay7db73692019-10-20 14:51:12 -0400751 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800752 Type::Ident(ident) if ident == RustString => quote!(&::cxx::private::RustString),
Myron Ahneba35cf2020-02-05 19:41:51 +0700753 Type::RustVec(ty) => {
754 let inner = expand_extern_type(&ty.inner);
David Tolnay6c6b7e02020-04-24 11:42:59 -0700755 quote!(&::cxx::private::RustVec<#inner>)
Myron Ahneba35cf2020-02-05 19:41:51 +0700756 }
David Tolnay7db73692019-10-20 14:51:12 -0400757 _ => quote!(#ty),
758 },
759 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700760 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400761 _ => quote!(#ty),
762 }
763}
764
765fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
766 let ret = match ret {
767 Some(ret) if !types.needs_indirect_abi(ret) => ret,
768 _ => return TokenStream::new(),
769 };
770 let ty = expand_extern_type(ret);
771 quote!(-> #ty)
772}