blob: 0869c2c587262fbfcad0beb229121799159ceab9 [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 Tolnayc8361022020-08-25 21:57:53 -070034 hidden.extend(expand_rust_type_assert_sized(ety));
David Tolnay7db73692019-10-20 14:51:12 -040035 }
36 }
37
David Tolnay2ec14632020-05-04 00:47:10 -070038 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040039 match api {
40 Api::Include(_) | Api::RustType(_) => {}
41 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
Joel Galensonc03402a2020-04-23 17:31:09 -070042 Api::Enum(enm) => expanded.extend(expand_enum(enm)),
Joel Galenson905eb2e2020-05-04 14:58:14 -070043 Api::CxxType(ety) => {
Joel Galenson0f654ff2020-05-04 20:04:21 -070044 if !types.enums.contains_key(&ety.ident) {
David Tolnay5f9e8ca2020-05-07 17:21:05 -070045 expanded.extend(expand_cxx_type(namespace, ety));
Joel Galenson905eb2e2020-05-04 14:58:14 -070046 }
47 }
David Tolnay7db73692019-10-20 14:51:12 -040048 Api::CxxFunction(efn) => {
49 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
50 }
51 Api::RustFunction(efn) => {
52 hidden.extend(expand_rust_function_shim(namespace, efn, types))
53 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -070054 Api::TypeAlias(alias) => {
55 expanded.extend(expand_type_alias(alias));
56 hidden.extend(expand_type_alias_verify(namespace, alias));
57 }
David Tolnay7db73692019-10-20 14:51:12 -040058 }
59 }
60
61 for ty in types {
62 if let Type::RustBox(ty) = ty {
63 if let Type::Ident(ident) = &ty.inner {
64 if Atom::from(ident).is_none() {
65 hidden.extend(expand_rust_box(namespace, ident));
66 }
67 }
Myron Ahneba35cf2020-02-05 19:41:51 +070068 } else if let Type::RustVec(ty) = ty {
69 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070070 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070071 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070072 }
Myron Ahneba35cf2020-02-05 19:41:51 +070073 }
David Tolnay7db73692019-10-20 14:51:12 -040074 } else if let Type::UniquePtr(ptr) = ty {
75 if let Type::Ident(ident) = &ptr.inner {
David Tolnay2b821e62020-05-07 22:55:28 -070076 if Atom::from(ident).is_none() && !types.aliases.contains_key(ident) {
David Tolnay3b40b6f2020-04-24 17:58:24 -070077 expanded.extend(expand_unique_ptr(namespace, ident, types));
Myron Ahneba35cf2020-02-05 19:41:51 +070078 }
79 }
David Tolnay4377a9e2020-04-24 15:20:26 -070080 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070081 if let Type::Ident(ident) = &ptr.inner {
David Tolnay2b821e62020-05-07 22:55:28 -070082 if Atom::from(ident).is_none() && !types.aliases.contains_key(ident) {
David Tolnay17631102020-04-24 19:01:30 -070083 // Generate impl for CxxVector<T> if T is a struct or opaque
84 // C++ type. Impl for primitives is already provided by cxx
85 // crate.
86 expanded.extend(expand_cxx_vector(namespace, ident));
David Tolnay7db73692019-10-20 14:51:12 -040087 }
88 }
89 }
90 }
91
92 // Work around https://github.com/rust-lang/rust/issues/67851.
93 if !hidden.is_empty() {
94 expanded.extend(quote! {
95 #[doc(hidden)]
96 const _: () = {
97 #hidden
98 };
99 });
100 }
101
102 let attrs = ffi
103 .attrs
104 .into_iter()
105 .filter(|attr| attr.path.is_ident("doc"));
106 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700107 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400108
David Tolnaycbf3f032020-05-03 23:51:14 -0700109 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400110 #(#attrs)*
111 #[deny(improper_ctypes)]
112 #[allow(non_snake_case)]
113 #vis mod #ident {
114 #expanded
115 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700116 }
David Tolnay7db73692019-10-20 14:51:12 -0400117}
118
119fn expand_struct(strct: &Struct) -> TokenStream {
120 let ident = &strct.ident;
121 let doc = &strct.doc;
122 let derives = &strct.derives;
123 let fields = strct.fields.iter().map(|field| {
124 // This span on the pub makes "private type in public interface" errors
125 // appear in the right place.
126 let vis = Token![pub](field.ident.span());
127 quote!(#vis #field)
128 });
129 quote! {
130 #doc
131 #[derive(#(#derives),*)]
132 #[repr(C)]
133 pub struct #ident {
134 #(#fields,)*
135 }
136 }
137}
138
Joel Galensonc03402a2020-04-23 17:31:09 -0700139fn expand_enum(enm: &Enum) -> TokenStream {
140 let ident = &enm.ident;
141 let doc = &enm.doc;
David Tolnayc605e6f2020-05-10 23:37:12 -0700142 let repr = enm.repr;
Joel Galenson88547732020-05-05 08:23:42 -0700143 let variants = enm.variants.iter().map(|variant| {
144 let variant_ident = &variant.ident;
145 let discriminant = &variant.discriminant;
146 Some(quote! {
147 pub const #variant_ident: Self = #ident { repr: #discriminant };
148 })
149 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700150 quote! {
151 #doc
152 #[derive(Copy, Clone, PartialEq, Eq)]
153 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700154 pub struct #ident {
David Tolnayc605e6f2020-05-10 23:37:12 -0700155 pub repr: #repr,
Joel Galensondb1ec312020-05-01 13:57:32 -0700156 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700157
158 #[allow(non_upper_case_globals)]
159 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700160 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700161 }
162 }
163}
164
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700165fn expand_cxx_type(namespace: &Namespace, ety: &ExternType) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400166 let ident = &ety.ident;
167 let doc = &ety.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700168 let type_id = type_id(namespace, ident);
169
David Tolnay7db73692019-10-20 14:51:12 -0400170 quote! {
171 #doc
172 #[repr(C)]
173 pub struct #ident {
174 _private: ::cxx::private::Opaque,
175 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700176
177 unsafe impl ::cxx::ExternType for #ident {
178 type Id = #type_id;
179 }
David Tolnay7db73692019-10-20 14:51:12 -0400180 }
181}
182
183fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
184 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700185 let receiver = efn.receiver.iter().map(|receiver| {
186 let receiver_type = receiver.ty();
187 quote!(_: #receiver_type)
188 });
David Tolnay39d575f2020-03-03 00:10:56 -0800189 let args = efn.args.iter().map(|arg| {
190 let ident = &arg.ident;
191 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800192 if arg.ty == RustString {
193 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700194 } else if let Type::RustVec(_) = arg.ty {
195 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700196 } else if let Type::Fn(_) = arg.ty {
197 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800198 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800199 quote!(#ident: *mut #ty)
200 } else {
201 quote!(#ident: #ty)
202 }
203 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700204 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700205 let ret = if efn.throws {
206 quote!(-> ::cxx::private::Result)
207 } else {
208 expand_extern_return_type(&efn.ret, types)
209 };
David Tolnay7db73692019-10-20 14:51:12 -0400210 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700211 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400212 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
213 outparam = Some(quote!(__return: *mut #ret));
214 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700215 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400216 let local_name = format_ident!("__{}", ident);
217 quote! {
218 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700219 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400220 }
221}
222
223fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
224 let ident = &efn.ident;
225 let doc = &efn.doc;
226 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700227 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700228 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700229 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700230 let var = receiver.var;
231 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700232 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700233 let args = efn.args.iter().map(|arg| quote!(#arg));
234 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700235 let ret = if efn.throws {
236 let ok = match &efn.ret {
237 Some(ret) => quote!(#ret),
238 None => quote!(()),
239 };
240 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
241 } else {
242 expand_return_type(&efn.ret)
243 };
David Tolnay1e548172020-03-16 13:37:09 -0700244 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700245 let receiver_var = efn
246 .receiver
247 .iter()
248 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700249 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400250 let var = &arg.ident;
251 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800252 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800253 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400254 }
255 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
256 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700257 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400258 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700259 Type::Ident(ident) if ident == RustString => match ty.mutability {
260 None => quote!(::cxx::private::RustString::from_ref(#var)),
261 Some(_) => quote!(::cxx::private::RustString::from_mut(#var)),
262 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700263 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
264 None => quote!(::cxx::private::RustVec::from_ref_vec_string(#var)),
265 Some(_) => quote!(::cxx::private::RustVec::from_mut_vec_string(#var)),
266 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700267 Type::RustVec(_) => match ty.mutability {
268 None => quote!(::cxx::private::RustVec::from_ref(#var)),
269 Some(_) => quote!(::cxx::private::RustVec::from_mut(#var)),
270 },
David Tolnay7db73692019-10-20 14:51:12 -0400271 _ => quote!(#var),
272 },
273 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700274 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400275 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
276 _ => quote!(#var),
277 }
278 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700279 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700280 let trampolines = efn
281 .args
282 .iter()
283 .filter_map(|arg| {
284 if let Type::Fn(f) = &arg.ty {
285 let var = &arg.ident;
286 Some(expand_function_pointer_trampoline(
287 namespace, efn, var, f, types,
288 ))
289 } else {
290 None
291 }
292 })
293 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400294 let mut setup = efn
295 .args
296 .iter()
297 .filter(|arg| types.needs_indirect_abi(&arg.ty))
298 .map(|arg| {
299 let var = &arg.ident;
300 // These are arguments for which C++ has taken ownership of the data
301 // behind the mut reference it received.
302 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700303 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400304 }
305 })
306 .collect::<TokenStream>();
307 let local_name = format_ident!("__{}", ident);
308 let call = if indirect_return {
309 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
310 setup.extend(quote! {
311 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400312 });
David Tolnayebef4a22020-03-17 15:33:47 -0700313 if efn.throws {
314 setup.extend(quote! {
315 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
316 });
317 quote!(::std::result::Result::Ok(__return.assume_init()))
318 } else {
319 setup.extend(quote! {
320 #local_name(#(#vars,)* __return.as_mut_ptr());
321 });
322 quote!(__return.assume_init())
323 }
324 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400325 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700326 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400327 }
328 } else {
329 quote! {
330 #local_name(#(#vars),*)
331 }
332 };
Myron Ahn84849302020-03-25 22:00:58 +0700333 let expr = if efn.throws {
334 efn.ret.as_ref().and_then(|ret| match ret {
335 Type::Ident(ident) if ident == RustString => {
336 Some(quote!(#call.map(|r| r.into_string())))
337 }
338 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700339 Type::RustVec(vec) => {
340 if vec.inner == RustString {
341 Some(quote!(#call.map(|r| r.into_vec_string())))
342 } else {
343 Some(quote!(#call.map(|r| r.into_vec())))
344 }
345 }
Myron Ahn84849302020-03-25 22:00:58 +0700346 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
347 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700348 Type::Ident(ident) if ident == RustString => match ty.mutability {
349 None => Some(quote!(#call.map(|r| r.as_string()))),
350 Some(_) => Some(quote!(#call.map(|r| r.as_mut_string()))),
351 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700352 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
353 None => Some(quote!(#call.map(|r| r.as_vec_string()))),
354 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec_string()))),
355 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700356 Type::RustVec(_) => match ty.mutability {
357 None => Some(quote!(#call.map(|r| r.as_vec()))),
358 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec()))),
359 },
Myron Ahn84849302020-03-25 22:00:58 +0700360 _ => None,
361 },
362 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700363 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700364 _ => None,
365 })
366 } else {
367 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800368 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400369 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700370 Type::RustVec(vec) => {
371 if vec.inner == RustString {
372 Some(quote!(#call.into_vec_string()))
373 } else {
374 Some(quote!(#call.into_vec()))
375 }
376 }
David Tolnay7db73692019-10-20 14:51:12 -0400377 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
378 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700379 Type::Ident(ident) if ident == RustString => match ty.mutability {
380 None => Some(quote!(#call.as_string())),
381 Some(_) => Some(quote!(#call.as_mut_string())),
382 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700383 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
384 None => Some(quote!(#call.as_vec_string())),
385 Some(_) => Some(quote!(#call.as_mut_vec_string())),
386 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700387 Type::RustVec(_) => match ty.mutability {
388 None => Some(quote!(#call.as_vec())),
389 Some(_) => Some(quote!(#call.as_mut_vec())),
390 },
David Tolnay7db73692019-10-20 14:51:12 -0400391 _ => None,
392 },
393 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700394 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400395 _ => None,
396 })
Myron Ahn84849302020-03-25 22:00:58 +0700397 }
398 .unwrap_or(call);
David Tolnayc66cdbb2020-04-20 01:41:15 -0700399 let function_shim = quote! {
400 #doc
401 pub fn #ident(#(#all_args,)*) #ret {
402 extern "C" {
403 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400404 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700405 #trampolines
406 unsafe {
407 #setup
408 #expr
David Tolnay7db73692019-10-20 14:51:12 -0400409 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700410 }
411 };
412 match &efn.receiver {
413 None => function_shim,
414 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700415 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700416 quote!(impl #receiver_type { #function_shim })
417 }
David Tolnay7db73692019-10-20 14:51:12 -0400418 }
419}
420
David Tolnay75dca2e2020-03-25 20:17:52 -0700421fn expand_function_pointer_trampoline(
422 namespace: &Namespace,
423 efn: &ExternFn,
424 var: &Ident,
425 sig: &Signature,
426 types: &Types,
427) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700428 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
429 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700430 let local_name = parse_quote!(__);
431 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
432 let shim = expand_rust_function_shim_impl(
433 sig,
434 types,
435 &r_trampoline,
436 local_name,
437 catch_unwind_label,
438 None,
439 );
440
441 quote! {
442 let #var = ::cxx::private::FatFunction {
443 trampoline: {
444 extern "C" {
445 #[link_name = #c_trampoline]
446 fn trampoline();
447 }
448 #shim
449 trampoline as usize as *const ()
450 },
451 ptr: #var as usize as *const (),
452 };
453 }
454}
455
David Tolnay7db73692019-10-20 14:51:12 -0400456fn expand_rust_type(ety: &ExternType) -> TokenStream {
457 let ident = &ety.ident;
458 quote! {
459 use super::#ident;
460 }
461}
462
David Tolnayc8361022020-08-25 21:57:53 -0700463fn expand_rust_type_assert_sized(ety: &ExternType) -> TokenStream {
464 // Rustc will render as follows if not sized:
465 //
466 // type TheirType;
467 // -----^^^^^^^^^-
468 // | |
469 // | doesn't have a size known at compile-time
470 // required by this bound in `ffi::_::__AssertSized`
471
472 let ident = &ety.ident;
473 let begin_span = Token![::](ety.type_token.span);
474 let sized = quote_spanned! {ety.semi_token.span=>
475 #begin_span std::marker::Sized
476 };
477 quote_spanned! {ident.span()=>
478 let _ = {
479 fn __AssertSized<T: ?#sized + #sized>() {}
480 __AssertSized::<#ident>
481 };
482 }
483}
484
David Tolnay7db73692019-10-20 14:51:12 -0400485fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
486 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700487 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700488 let local_name = format_ident!("__{}", ident);
489 let catch_unwind_label = format!("::{}", ident);
490 let invoke = Some(ident);
491 expand_rust_function_shim_impl(
492 efn,
493 types,
494 &link_name,
495 local_name,
496 catch_unwind_label,
497 invoke,
498 )
499}
500
501fn expand_rust_function_shim_impl(
502 sig: &Signature,
503 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700504 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700505 local_name: Ident,
506 catch_unwind_label: String,
507 invoke: Option<&Ident>,
508) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700509 let receiver_var = sig
510 .receiver
511 .as_ref()
512 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700513 let receiver = sig.receiver.as_ref().map(|receiver| {
514 let receiver_type = receiver.ty();
515 quote!(#receiver_var: #receiver_type)
516 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700517 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800518 let ident = &arg.ident;
519 let ty = expand_extern_type(&arg.ty);
520 if types.needs_indirect_abi(&arg.ty) {
521 quote!(#ident: *mut #ty)
522 } else {
523 quote!(#ident: #ty)
524 }
525 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700526 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700527
David Tolnay3a45f2d2020-04-20 01:51:12 -0700528 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400529 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800530 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800531 Type::Ident(i) if i == RustString => {
532 quote!(::std::mem::take((*#ident).as_mut_string()))
533 }
David Tolnay40226ab2020-03-03 00:05:35 -0800534 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay33f56ad2020-08-27 17:06:35 -0700535 Type::RustVec(vec) => {
536 if vec.inner == RustString {
537 quote!(::std::mem::take((*#ident).as_mut_vec_string()))
538 } else {
539 quote!(::std::mem::take((*#ident).as_mut_vec()))
540 }
541 }
David Tolnay40226ab2020-03-03 00:05:35 -0800542 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800543 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700544 Type::Ident(i) if i == RustString => match ty.mutability {
545 None => quote!(#ident.as_string()),
546 Some(_) => quote!(#ident.as_mut_string()),
547 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700548 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
549 None => quote!(#ident.as_vec_string()),
550 Some(_) => quote!(#ident.as_mut_vec_string()),
551 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700552 Type::RustVec(_) => match ty.mutability {
553 None => quote!(#ident.as_vec()),
554 Some(_) => quote!(#ident.as_mut_vec()),
555 },
David Tolnay40226ab2020-03-03 00:05:35 -0800556 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800557 },
David Tolnay40226ab2020-03-03 00:05:35 -0800558 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700559 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800560 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
561 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400562 }
563 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700564 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700565
566 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700567 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700568 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700569 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700570 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700571 quote!(#receiver_type::#ident)
572 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700573 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700574 None => quote!(__extern),
575 };
576 call.extend(quote! { (#(#vars),*) });
577
578 let mut expr = sig
David Tolnay7db73692019-10-20 14:51:12 -0400579 .ret
580 .as_ref()
581 .and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800582 Type::Ident(ident) if ident == RustString => {
David Tolnay7db73692019-10-20 14:51:12 -0400583 Some(quote!(::cxx::private::RustString::from(#call)))
584 }
585 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw(#call))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700586 Type::RustVec(vec) => {
587 if vec.inner == RustString {
588 Some(quote!(::cxx::private::RustVec::from_vec_string(#call)))
589 } else {
590 Some(quote!(::cxx::private::RustVec::from(#call)))
591 }
592 }
David Tolnay7db73692019-10-20 14:51:12 -0400593 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw(#call))),
594 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700595 Type::Ident(ident) if ident == RustString => match ty.mutability {
596 None => Some(quote!(::cxx::private::RustString::from_ref(#call))),
597 Some(_) => Some(quote!(::cxx::private::RustString::from_mut(#call))),
598 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700599 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
600 None => Some(quote!(::cxx::private::RustVec::from_ref_vec_string(#call))),
601 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut_vec_string(#call))),
602 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700603 Type::RustVec(_) => match ty.mutability {
604 None => Some(quote!(::cxx::private::RustVec::from_ref(#call))),
605 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut(#call))),
606 },
David Tolnay7db73692019-10-20 14:51:12 -0400607 _ => None,
608 },
609 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from(#call))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700610 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from(#call))),
David Tolnay7db73692019-10-20 14:51:12 -0400611 _ => None,
612 })
613 .unwrap_or(call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700614
615 let mut outparam = None;
616 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700617 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700618 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
619 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700620 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700621 if sig.throws {
622 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700623 Some(_) => quote!(__return),
624 None => quote!(&mut ()),
625 };
626 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700627 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400628 expr = quote!(::std::ptr::write(__return, #expr));
629 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700630
David Tolnay1e548172020-03-16 13:37:09 -0700631 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700632
633 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700634 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700635 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700636 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700637 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700638
639 let pointer = match invoke {
640 None => Some(quote!(__extern: #sig)),
641 Some(_) => None,
642 };
643
David Tolnay7db73692019-10-20 14:51:12 -0400644 quote! {
645 #[doc(hidden)]
646 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700647 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400648 let __fn = concat!(module_path!(), #catch_unwind_label);
649 #expr
650 }
651 }
652}
653
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700654fn expand_type_alias(alias: &TypeAlias) -> TokenStream {
655 let ident = &alias.ident;
656 let ty = &alias.ty;
657 quote! {
658 pub type #ident = #ty;
659 }
660}
661
662fn expand_type_alias_verify(namespace: &Namespace, alias: &TypeAlias) -> TokenStream {
663 let ident = &alias.ident;
664 let type_id = type_id(namespace, ident);
David Tolnay83fe0f02020-05-07 20:00:15 -0700665 let begin_span = alias.type_token.span;
666 let end_span = alias.semi_token.span;
667 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_type::<);
668 let end = quote_spanned!(end_span=> >);
669
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700670 quote! {
David Tolnay83fe0f02020-05-07 20:00:15 -0700671 const _: fn() = #begin #ident, #type_id #end;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700672 }
673}
674
675fn type_id(namespace: &Namespace, ident: &Ident) -> TokenStream {
676 let mut path = String::new();
677 for name in namespace {
678 path += &name.to_string();
679 path += "::";
680 }
681 path += &ident.to_string();
682
683 quote! {
684 ::cxx::type_id!(#path)
685 }
686}
687
David Tolnay7db73692019-10-20 14:51:12 -0400688fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700689 let link_prefix = format!("cxxbridge03$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400690 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400691 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400692
693 let local_prefix = format_ident!("{}__box_", ident);
694 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400695 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400696
697 let span = ident.span();
698 quote_spanned! {span=>
699 #[doc(hidden)]
700 #[export_name = #link_uninit]
701 unsafe extern "C" fn #local_uninit(
702 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
703 ) {
704 ::std::ptr::write(
705 this,
706 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
707 );
708 }
709 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400710 #[export_name = #link_drop]
711 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
712 ::std::ptr::drop_in_place(this);
713 }
David Tolnay7db73692019-10-20 14:51:12 -0400714 }
715}
716
David Tolnay83a7ec92020-04-25 12:37:42 -0700717fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay69601622020-04-29 18:48:36 -0700718 let link_prefix = format!("cxxbridge03$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700719 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700720 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700721 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700722 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700723 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700724
David Tolnay83a7ec92020-04-25 12:37:42 -0700725 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700726 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700727 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700728 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700729 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700730 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700731
David Tolnay83a7ec92020-04-25 12:37:42 -0700732 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700733 quote_spanned! {span=>
734 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700735 #[export_name = #link_new]
736 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
737 ::std::ptr::write(this, ::cxx::private::RustVec::new());
738 }
739 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700740 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700741 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700742 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700743 }
David Tolnay85db5a02020-04-25 13:17:27 -0700744 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700745 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700746 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700747 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700748 }
David Tolnay219c0792020-04-24 20:31:37 -0700749 #[doc(hidden)]
750 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700751 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700752 (*this).as_ptr()
753 }
David Tolnay503d0192020-04-24 22:18:56 -0700754 #[doc(hidden)]
755 #[export_name = #link_stride]
756 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700757 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700758 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700759 }
760}
761
David Tolnay3b40b6f2020-04-24 17:58:24 -0700762fn expand_unique_ptr(namespace: &Namespace, ident: &Ident, types: &Types) -> TokenStream {
763 let name = ident.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700764 let prefix = format!("cxxbridge03$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400765 let link_null = format!("{}null", prefix);
766 let link_new = format!("{}new", prefix);
767 let link_raw = format!("{}raw", prefix);
768 let link_get = format!("{}get", prefix);
769 let link_release = format!("{}release", prefix);
770 let link_drop = format!("{}drop", prefix);
771
David Tolnay3b40b6f2020-04-24 17:58:24 -0700772 let new_method = if types.structs.contains_key(ident) {
773 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700774 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
775 extern "C" {
776 #[link_name = #link_new]
777 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
778 }
779 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
780 unsafe { __new(&mut repr, &mut value) }
781 repr
782 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700783 })
784 } else {
785 None
David Tolnay53838912020-04-09 20:56:44 -0700786 };
787
David Tolnay7db73692019-10-20 14:51:12 -0400788 quote! {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700789 unsafe impl ::cxx::private::UniquePtrTarget for #ident {
790 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400791 fn __null() -> *mut ::std::ffi::c_void {
792 extern "C" {
793 #[link_name = #link_null]
794 fn __null(this: *mut *mut ::std::ffi::c_void);
795 }
796 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
797 unsafe { __null(&mut repr) }
798 repr
799 }
David Tolnay53838912020-04-09 20:56:44 -0700800 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400801 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
802 extern "C" {
803 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700804 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400805 }
806 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
807 __raw(&mut repr, raw);
808 repr
809 }
810 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
811 extern "C" {
812 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700813 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400814 }
815 __get(&repr)
816 }
817 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
818 extern "C" {
819 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700820 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400821 }
822 __release(&mut repr)
823 }
824 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
825 extern "C" {
826 #[link_name = #link_drop]
827 fn __drop(this: *mut *mut ::std::ffi::c_void);
828 }
829 __drop(&mut repr);
830 }
831 }
832 }
833}
834
David Tolnay17631102020-04-24 19:01:30 -0700835fn expand_cxx_vector(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700836 let name = elem.to_string();
David Tolnay69601622020-04-29 18:48:36 -0700837 let prefix = format!("cxxbridge03$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700838 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700839 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay69601622020-04-29 18:48:36 -0700840 let unique_ptr_prefix = format!("cxxbridge03$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700841 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
842 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
843 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
844 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
845 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700846
847 quote! {
David Tolnay17631102020-04-24 19:01:30 -0700848 unsafe impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700849 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700850 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700851 extern "C" {
852 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700853 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700854 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700855 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700856 }
David Tolnay17631102020-04-24 19:01:30 -0700857 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> &Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700858 extern "C" {
859 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700860 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700861 }
David Tolnay9626d082020-04-24 14:52:45 -0700862 &*__get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700863 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700864 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
865 extern "C" {
866 #[link_name = #link_unique_ptr_null]
867 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
868 }
869 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
870 unsafe { __unique_ptr_null(&mut repr) }
871 repr
872 }
873 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
874 extern "C" {
875 #[link_name = #link_unique_ptr_raw]
876 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
877 }
878 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
879 __unique_ptr_raw(&mut repr, raw);
880 repr
881 }
882 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
883 extern "C" {
884 #[link_name = #link_unique_ptr_get]
885 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
886 }
887 __unique_ptr_get(&repr)
888 }
889 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
890 extern "C" {
891 #[link_name = #link_unique_ptr_release]
892 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
893 }
894 __unique_ptr_release(&mut repr)
895 }
896 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
897 extern "C" {
898 #[link_name = #link_unique_ptr_drop]
899 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
900 }
901 __unique_ptr_drop(&mut repr);
902 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700903 }
904 }
905}
906
David Tolnay7db73692019-10-20 14:51:12 -0400907fn expand_return_type(ret: &Option<Type>) -> TokenStream {
908 match ret {
909 Some(ret) => quote!(-> #ret),
910 None => TokenStream::new(),
911 }
912}
913
David Tolnay75dca2e2020-03-25 20:17:52 -0700914fn indirect_return(sig: &Signature, types: &Types) -> bool {
915 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700916 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700917 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400918}
919
920fn expand_extern_type(ty: &Type) -> TokenStream {
921 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800922 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400923 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700924 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400925 quote!(*mut #inner)
926 }
David Tolnaye6d50212020-04-25 15:38:23 -0700927 Type::RustVec(ty) => {
928 let elem = expand_extern_type(&ty.inner);
929 quote!(::cxx::private::RustVec<#elem>)
930 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700931 Type::Ref(ty) => {
932 let mutability = ty.mutability;
933 match &ty.inner {
934 Type::Ident(ident) if ident == RustString => {
935 quote!(&#mutability ::cxx::private::RustString)
936 }
937 Type::RustVec(ty) => {
938 let inner = expand_extern_type(&ty.inner);
939 quote!(&#mutability ::cxx::private::RustVec<#inner>)
940 }
941 _ => quote!(#ty),
Myron Ahneba35cf2020-02-05 19:41:51 +0700942 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700943 }
David Tolnay7db73692019-10-20 14:51:12 -0400944 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700945 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400946 _ => quote!(#ty),
947 }
948}
949
950fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
951 let ret = match ret {
952 Some(ret) if !types.needs_indirect_abi(ret) => ret,
953 _ => return TokenStream::new(),
954 };
955 let ty = expand_extern_type(ret);
956 quote!(-> #ty)
957}