blob: 2c8cc872c440277a1ca6c12b65bcafd222f47d12 [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 Tolnay891061b2020-04-19 22:42:33 -07003use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07004use crate::syntax::{
5 self, check, mangle, Api, ExternFn, ExternType, Signature, Struct, Type, Types,
6};
David Tolnay7db73692019-10-20 14:51:12 -04007use proc_macro2::{Ident, Span, TokenStream};
David Tolnayf9ffb932020-04-20 02:22:57 -07008use quote::{format_ident, quote, quote_spanned, ToTokens};
David Tolnay83a7ec92020-04-25 12:37:42 -07009use syn::{parse_quote, Error, ItemMod, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040010
11pub fn bridge(namespace: &Namespace, ffi: ItemMod) -> Result<TokenStream> {
12 let ident = &ffi.ident;
13 let content = ffi.content.ok_or(Error::new(
14 Span::call_site(),
15 "#[cxx::bridge] module must have inline contents",
16 ))?;
17 let apis = syntax::parse_items(content.1)?;
18 let ref types = Types::collect(&apis)?;
19 check::typecheck(&apis, types)?;
20
21 let mut expanded = TokenStream::new();
22 let mut hidden = TokenStream::new();
David Tolnay7db73692019-10-20 14:51:12 -040023
24 for api in &apis {
25 if let Api::RustType(ety) = api {
26 expanded.extend(expand_rust_type(ety));
David Tolnay7db73692019-10-20 14:51:12 -040027 let ident = &ety.ident;
David Tolnay4b07ab92020-04-22 19:50:21 -070028 let span = ident.span();
David Tolnay1044d442020-04-22 19:58:30 -070029 hidden.extend(quote_spanned! {span=>
30 let _ = ::std::ptr::read::<#ident>;
31 });
David Tolnay7db73692019-10-20 14:51:12 -040032 }
33 }
34
35 for api in &apis {
36 match api {
37 Api::Include(_) | Api::RustType(_) => {}
38 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
39 Api::CxxType(ety) => expanded.extend(expand_cxx_type(ety)),
40 Api::CxxFunction(efn) => {
41 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
42 }
43 Api::RustFunction(efn) => {
44 hidden.extend(expand_rust_function_shim(namespace, efn, types))
45 }
46 }
47 }
48
49 for ty in types {
50 if let Type::RustBox(ty) = ty {
51 if let Type::Ident(ident) = &ty.inner {
52 if Atom::from(ident).is_none() {
53 hidden.extend(expand_rust_box(namespace, ident));
54 }
55 }
Myron Ahneba35cf2020-02-05 19:41:51 +070056 } else if let Type::RustVec(ty) = ty {
57 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070058 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070059 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070060 }
Myron Ahneba35cf2020-02-05 19:41:51 +070061 }
David Tolnay7db73692019-10-20 14:51:12 -040062 } else if let Type::UniquePtr(ptr) = ty {
63 if let Type::Ident(ident) = &ptr.inner {
64 if Atom::from(ident).is_none() {
David Tolnay3b40b6f2020-04-24 17:58:24 -070065 expanded.extend(expand_unique_ptr(namespace, ident, types));
Myron Ahneba35cf2020-02-05 19:41:51 +070066 }
67 }
David Tolnay4377a9e2020-04-24 15:20:26 -070068 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070069 if let Type::Ident(ident) = &ptr.inner {
70 if Atom::from(ident).is_none() {
David Tolnay17631102020-04-24 19:01:30 -070071 // Generate impl for CxxVector<T> if T is a struct or opaque
72 // C++ type. Impl for primitives is already provided by cxx
73 // crate.
74 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040075 }
76 }
77 }
78 }
79
80 // Work around https://github.com/rust-lang/rust/issues/67851.
81 if !hidden.is_empty() {
82 expanded.extend(quote! {
83 #[doc(hidden)]
84 const _: () = {
85 #hidden
86 };
87 });
88 }
89
90 let attrs = ffi
91 .attrs
92 .into_iter()
93 .filter(|attr| attr.path.is_ident("doc"));
94 let vis = &ffi.vis;
95
96 Ok(quote! {
97 #(#attrs)*
98 #[deny(improper_ctypes)]
99 #[allow(non_snake_case)]
100 #vis mod #ident {
101 #expanded
102 }
103 })
104}
105
106fn expand_struct(strct: &Struct) -> TokenStream {
107 let ident = &strct.ident;
108 let doc = &strct.doc;
109 let derives = &strct.derives;
110 let fields = strct.fields.iter().map(|field| {
111 // This span on the pub makes "private type in public interface" errors
112 // appear in the right place.
113 let vis = Token![pub](field.ident.span());
114 quote!(#vis #field)
115 });
116 quote! {
117 #doc
118 #[derive(#(#derives),*)]
119 #[repr(C)]
120 pub struct #ident {
121 #(#fields,)*
122 }
123 }
124}
125
126fn expand_cxx_type(ety: &ExternType) -> TokenStream {
127 let ident = &ety.ident;
128 let doc = &ety.doc;
129 quote! {
130 #doc
131 #[repr(C)]
132 pub struct #ident {
133 _private: ::cxx::private::Opaque,
134 }
135 }
136}
137
138fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
139 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700140 let receiver = efn.receiver.iter().map(|receiver| {
141 let receiver_type = receiver.ty();
142 quote!(_: #receiver_type)
143 });
David Tolnay39d575f2020-03-03 00:10:56 -0800144 let args = efn.args.iter().map(|arg| {
145 let ident = &arg.ident;
146 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800147 if arg.ty == RustString {
148 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700149 } else if let Type::RustVec(_) = arg.ty {
150 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700151 } else if let Type::Fn(_) = arg.ty {
152 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800153 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800154 quote!(#ident: *mut #ty)
155 } else {
156 quote!(#ident: #ty)
157 }
158 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700159 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700160 let ret = if efn.throws {
161 quote!(-> ::cxx::private::Result)
162 } else {
163 expand_extern_return_type(&efn.ret, types)
164 };
David Tolnay7db73692019-10-20 14:51:12 -0400165 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700166 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400167 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
168 outparam = Some(quote!(__return: *mut #ret));
169 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700170 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400171 let local_name = format_ident!("__{}", ident);
172 quote! {
173 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700174 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400175 }
176}
177
178fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
179 let ident = &efn.ident;
180 let doc = &efn.doc;
181 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700182 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700183 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700184 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700185 let var = receiver.var;
186 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700187 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700188 let args = efn.args.iter().map(|arg| quote!(#arg));
189 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700190 let ret = if efn.throws {
191 let ok = match &efn.ret {
192 Some(ret) => quote!(#ret),
193 None => quote!(()),
194 };
195 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
196 } else {
197 expand_return_type(&efn.ret)
198 };
David Tolnay1e548172020-03-16 13:37:09 -0700199 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700200 let receiver_var = efn
201 .receiver
202 .iter()
203 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700204 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400205 let var = &arg.ident;
206 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800207 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800208 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400209 }
210 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
211 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700212 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400213 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800214 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400215 quote!(::cxx::private::RustString::from_ref(#var))
216 }
David Tolnay6c6b7e02020-04-24 11:42:59 -0700217 Type::RustVec(_) => quote!(::cxx::private::RustVec::from_ref(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400218 _ => quote!(#var),
219 },
220 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700221 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400222 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
223 _ => quote!(#var),
224 }
225 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700226 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700227 let trampolines = efn
228 .args
229 .iter()
230 .filter_map(|arg| {
231 if let Type::Fn(f) = &arg.ty {
232 let var = &arg.ident;
233 Some(expand_function_pointer_trampoline(
234 namespace, efn, var, f, types,
235 ))
236 } else {
237 None
238 }
239 })
240 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400241 let mut setup = efn
242 .args
243 .iter()
244 .filter(|arg| types.needs_indirect_abi(&arg.ty))
245 .map(|arg| {
246 let var = &arg.ident;
247 // These are arguments for which C++ has taken ownership of the data
248 // behind the mut reference it received.
249 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700250 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400251 }
252 })
253 .collect::<TokenStream>();
254 let local_name = format_ident!("__{}", ident);
255 let call = if indirect_return {
256 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
257 setup.extend(quote! {
258 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400259 });
David Tolnayebef4a22020-03-17 15:33:47 -0700260 if efn.throws {
261 setup.extend(quote! {
262 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
263 });
264 quote!(::std::result::Result::Ok(__return.assume_init()))
265 } else {
266 setup.extend(quote! {
267 #local_name(#(#vars,)* __return.as_mut_ptr());
268 });
269 quote!(__return.assume_init())
270 }
271 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400272 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700273 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400274 }
275 } else {
276 quote! {
277 #local_name(#(#vars),*)
278 }
279 };
Myron Ahn84849302020-03-25 22:00:58 +0700280 let expr = if efn.throws {
281 efn.ret.as_ref().and_then(|ret| match ret {
282 Type::Ident(ident) if ident == RustString => {
283 Some(quote!(#call.map(|r| r.into_string())))
284 }
285 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay8b9d1762020-04-25 16:05:46 -0700286 Type::RustVec(_) => Some(quote!(#call.map(|r| r.into_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700287 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
288 Type::Ref(ty) => match &ty.inner {
289 Type::Ident(ident) if ident == RustString => {
290 Some(quote!(#call.map(|r| r.as_string())))
291 }
David Tolnay77989692020-04-25 15:57:47 -0700292 Type::RustVec(_) => Some(quote!(#call.map(|r| r.as_vec()))),
Myron Ahn84849302020-03-25 22:00:58 +0700293 _ => None,
294 },
295 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700296 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700297 _ => None,
298 })
299 } else {
300 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800301 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400302 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay03dca702020-04-25 15:08:05 -0700303 Type::RustVec(_) => Some(quote!(#call.into_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400304 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
305 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800306 Type::Ident(ident) if ident == RustString => Some(quote!(#call.as_string())),
David Tolnay77989692020-04-25 15:57:47 -0700307 Type::RustVec(_) => Some(quote!(#call.as_vec())),
David Tolnay7db73692019-10-20 14:51:12 -0400308 _ => None,
309 },
310 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700311 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400312 _ => None,
313 })
Myron Ahn84849302020-03-25 22:00:58 +0700314 }
315 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700316 let function_shim = quote! {
317 #doc
318 pub fn #ident(#(#all_args,)*) #ret {
319 extern "C" {
320 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400321 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700322 #trampolines
323 unsafe {
324 #setup
325 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400326 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700327 }
328 };
329 match &efn.receiver {
330 None => function_shim,
331 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700332 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700333 quote!(impl #receiver_type { #function_shim })
334 }
David Tolnay7db73692019-10-20 14:51:12 -0400335 }
336}
337
David Tolnay75dca2e2020-03-25 20:17:52 -0700338fn expand_function_pointer_trampoline(
339 namespace: &Namespace,
340 efn: &ExternFn,
341 var: &Ident,
342 sig: &Signature,
343 types: &Types,
344) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700345 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
346 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700347 let local_name = parse_quote!(__);
348 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
349 let shim = expand_rust_function_shim_impl(
350 sig,
351 types,
352 &r_trampoline,
353 local_name,
354 catch_unwind_label,
355 None,
356 );
357
358 quote! {
359 let #var = ::cxx::private::FatFunction {
360 trampoline: {
361 extern "C" {
362 #[link_name = #c_trampoline]
363 fn trampoline();
364 }
365 #shim
366 trampoline as usize as *const ()
367 },
368 ptr: #var as usize as *const (),
369 };
370 }
371}
372
David Tolnay7db73692019-10-20 14:51:12 -0400373fn expand_rust_type(ety: &ExternType) -> TokenStream {
374 let ident = &ety.ident;
375 quote! {
376 use super::#ident;
377 }
378}
379
380fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
381 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700382 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700383 let local_name = format_ident!("__{}", ident);
384 let catch_unwind_label = format!("::{}", ident);
385 let invoke = Some(ident);
386 expand_rust_function_shim_impl(
387 efn,
388 types,
389 &link_name,
390 local_name,
391 catch_unwind_label,
392 invoke,
393 )
394}
395
396fn expand_rust_function_shim_impl(
397 sig: &Signature,
398 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700399 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700400 local_name: Ident,
401 catch_unwind_label: String,
402 invoke: Option<&Ident>,
403) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700404 let receiver_var = sig
405 .receiver
406 .as_ref()
407 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700408 let receiver = sig.receiver.as_ref().map(|receiver| {
409 let receiver_type = receiver.ty();
410 quote!(#receiver_var: #receiver_type)
411 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700412 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800413 let ident = &arg.ident;
414 let ty = expand_extern_type(&arg.ty);
415 if types.needs_indirect_abi(&arg.ty) {
416 quote!(#ident: *mut #ty)
417 } else {
418 quote!(#ident: #ty)
419 }
420 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700421 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700422
David Tolnay3a45f2d2020-04-20 01:51:12 -0700423 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400424 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800425 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800426 Type::Ident(i) if i == RustString => {
427 quote!(::std::mem::take((*#ident).as_mut_string()))
428 }
David Tolnay40226ab2020-03-03 00:05:35 -0800429 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
430 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800431 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800432 Type::Ident(i) if i == RustString => quote!(#ident.as_string()),
David Tolnay40226ab2020-03-03 00:05:35 -0800433 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800434 },
David Tolnay40226ab2020-03-03 00:05:35 -0800435 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700436 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800437 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
438 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400439 }
440 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700441 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700442
443 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700444 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700445 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700446 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700447 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700448 quote!(#receiver_type::#ident)
449 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700450 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700451 None => quote!(__extern),
452 };
453 call.extend(quote! { (#(#vars),*) });
454
455 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400456 .ret
457 .as_ref()
458 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800459 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400460 Some(quote!(::cxx::private::RustString::from(#call)))
461 }
462 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
463 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
464 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800465 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400466 Some(quote!(::cxx::private::RustString::from_ref(#call)))
467 }
468 _ => None,
469 },
470 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700471 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400472 _ => None,
473 })
474 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700475
476 let mut outparam = None;
477 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700478 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700479 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
480 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700481 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700482 if sig.throws {
483 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700484 Some(_) => quote!(__return),
485 None => quote!(&mut ()),
486 };
487 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700488 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400489 expr = quote!(::std::ptr::write(__return, #expr));
490 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700491
David Tolnay1e548172020-03-16 13:37:09 -0700492 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700493
494 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700495 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700496 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700497 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700498 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700499
500 let pointer = match invoke {
501 None => Some(quote!(__extern: #sig)),
502 Some(_) => None,
503 };
504
David Tolnay7db73692019-10-20 14:51:12 -0400505 quote! {
506 #[doc(hidden)]
507 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700508 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400509 let __fn = concat!(module_path!(), #catch_unwind_label);
510 #expr
511 }
512 }
513}
514
515fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay8c730492020-03-13 01:29:06 -0700516 let link_prefix = format!("cxxbridge02$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400517 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400518 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400519
520 let local_prefix = format_ident!("{}__box_", ident);
521 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400522 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400523
524 let span = ident.span();
525 quote_spanned! {span=>
526 #[doc(hidden)]
527 #[export_name = #link_uninit]
528 unsafe extern "C" fn #local_uninit(
529 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
530 ) {
531 ::std::ptr::write(
532 this,
533 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
534 );
535 }
536 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400537 #[export_name = #link_drop]
538 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
539 ::std::ptr::drop_in_place(this);
540 }
David Tolnay7db73692019-10-20 14:51:12 -0400541 }
542}
543
David Tolnay83a7ec92020-04-25 12:37:42 -0700544fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
545 let link_prefix = format!("cxxbridge02$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700546 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700547 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700548 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700549 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700550 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700551
David Tolnay83a7ec92020-04-25 12:37:42 -0700552 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700553 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700554 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700555 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700556 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700557 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700558
David Tolnay83a7ec92020-04-25 12:37:42 -0700559 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700560 quote_spanned! {span=>
561 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700562 #[export_name = #link_new]
563 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
564 ::std::ptr::write(this, ::cxx::private::RustVec::new());
565 }
566 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700567 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700568 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700569 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700570 }
David Tolnay85db5a02020-04-25 13:17:27 -0700571 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700572 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700573 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700574 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700575 }
David Tolnay219c0792020-04-24 20:31:37 -0700576 #[doc(hidden)]
577 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700578 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700579 (*this).as_ptr()
580 }
David Tolnay503d0192020-04-24 22:18:56 -0700581 #[doc(hidden)]
582 #[export_name = #link_stride]
583 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700584 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700585 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700586 }
587}
588
David Tolnay3b40b6f2020-04-24 17:58:24 -0700589fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
590 let name = ident.to_string();
591 let prefix = format!("cxxbridge02$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400592 let link_null = format!("{}null", prefix);
593 let link_new = format!("{}new", prefix);
594 let link_raw = format!("{}raw", prefix);
595 let link_get = format!("{}get", prefix);
596 let link_release = format!("{}release", prefix);
597 let link_drop = format!("{}drop", prefix);
598
David Tolnay3b40b6f2020-04-24 17:58:24 -0700599 let new_method = if types.structs.contains_key(ident) {
600 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700601 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
602 extern "C" {
603 #[link_name = #link_new]
604 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
605 }
606 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
607 unsafe { __new(&mut repr, &mut value) }
608 repr
609 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700610 })
611 } else {
612 None
David Tolnay53838912020-04-09 20:56:44 -0700613 };
614
David Tolnay7db73692019-10-20 14:51:12 -0400615 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700616 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
617 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400618 fn __null() -> *mut ::std::ffi::c_void {
619 extern "C" {
620 #[link_name = #link_null]
621 fn __null(this: *mut *mut ::std::ffi::c_void);
622 }
623 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
624 unsafe { __null(&mut repr) }
625 repr
626 }
David Tolnay53838912020-04-09 20:56:44 -0700627 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400628 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
629 extern "C" {
630 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700631 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400632 }
633 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
634 __raw(&mut repr, raw);
635 repr
636 }
637 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
638 extern "C" {
639 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700640 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400641 }
642 __get(&repr)
643 }
644 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
645 extern "C" {
646 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700647 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400648 }
649 __release(&mut repr)
650 }
651 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
652 extern "C" {
653 #[link_name = #link_drop]
654 fn __drop(this: *mut *mut ::std::ffi::c_void);
655 }
656 __drop(&mut repr);
657 }
658 }
659 }
660}
661
David Tolnay17631102020-04-24 19:01:30 -0700662fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700663 let name = elem.to_string();
David Tolnay17631102020-04-24 19:01:30 -0700664 let prefix = format!("cxxbridge02$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700665 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700666 let link_get_unchecked = format!("{}get_unchecked", prefix);
667 let link_push_back = format!("{}push_back", prefix);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700668 let unique_ptr_prefix = format!("cxxbridge02$unique_ptr$std$vector${}{}$", namespace, elem);
669 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
670 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
671 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
672 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
673 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700674
675 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700676 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700677 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700678 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700679 extern "C" {
680 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700681 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700682 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700683 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700684 }
David Tolnay17631102020-04-24 19:01:30 -0700685 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700686 extern "C" {
687 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700688 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700689 }
David Tolnay9626d082020-04-24 14:52:45 -0700690 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700691 }
David Tolnay17631102020-04-24 19:01:30 -0700692 fn __push_back(v: &::cxx::CxxVector<Self>, item: &Self) {
David Tolnayfa2119c2020-04-24 14:42:58 -0700693 extern "C" {
694 #[link_name = #link_push_back]
David Tolnay17631102020-04-24 19:01:30 -0700695 fn __push_back(_: &::cxx::CxxVector<#elem>, _: &#elem);
Myron Ahneba35cf2020-02-05 19:41:51 +0700696 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700697 unsafe { __push_back(v, item) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700698 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700699 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
700 extern "C" {
701 #[link_name = #link_unique_ptr_null]
702 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
703 }
704 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
705 unsafe { __unique_ptr_null(&mut repr) }
706 repr
707 }
708 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
709 extern "C" {
710 #[link_name = #link_unique_ptr_raw]
711 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
712 }
713 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
714 __unique_ptr_raw(&mut repr, raw);
715 repr
716 }
717 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
718 extern "C" {
719 #[link_name = #link_unique_ptr_get]
720 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
721 }
722 __unique_ptr_get(&repr)
723 }
724 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
725 extern "C" {
726 #[link_name = #link_unique_ptr_release]
727 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
728 }
729 __unique_ptr_release(&mut repr)
730 }
731 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
732 extern "C" {
733 #[link_name = #link_unique_ptr_drop]
734 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
735 }
736 __unique_ptr_drop(&mut repr);
737 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700738 }
739 }
740}
741
David Tolnay7db73692019-10-20 14:51:12 -0400742fn expand_return_type(ret: &Option<Type>) -> TokenStream {
743 match ret {
744 Some(ret) => quote!(-> #ret),
745 None => TokenStream::new(),
746 }
747}
748
David Tolnay75dca2e2020-03-25 20:17:52 -0700749fn indirect_return(sig: &Signature, types: &Types) -> bool {
750 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700751 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700752 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400753}
754
755fn expand_extern_type(ty: &Type) -> TokenStream {
756 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800757 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400758 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700759 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400760 quote!(*mut #inner)
761 }
David Tolnaye6d50212020-04-25 15:38:23 -0700762 Type::RustVec(ty) => {
763 let elem = expand_extern_type(&ty.inner);
764 quote!(::cxx::private::RustVec<#elem>)
765 }
David Tolnay7db73692019-10-20 14:51:12 -0400766 Type::Ref(ty) => match &ty.inner {
David Tolnaya52602b2020-03-06 10:24:34 -0800767 Type::Ident(ident) if ident == RustString => quote!(&::cxx::private::RustString),
Myron Ahneba35cf2020-02-05 19:41:51 +0700768 Type::RustVec(ty) => {
769 let inner = expand_extern_type(&ty.inner);
David Tolnay6c6b7e02020-04-24 11:42:59 -0700770 quote!(&::cxx::private::RustVec<#inner>)
Myron Ahneba35cf2020-02-05 19:41:51 +0700771 }
David Tolnay7db73692019-10-20 14:51:12 -0400772 _ => quote!(#ty),
773 },
774 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700775 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400776 _ => quote!(#ty),
777 }
778}
779
780fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
781 let ret = match ret {
782 Some(ret) if !types.needs_indirect_abi(ret) => ret,
783 _ => return TokenStream::new(),
784 };
785 let ty = expand_extern_type(ret);
786 quote!(-> #ty)
787}