blob: 13134d070a8e60dde007c7fac91a46a360ae797d [file] [log] [blame]
David Tolnaya52602b2020-03-06 10:24:34 -08001use crate::syntax::atom::Atom::{self, *};
David Tolnay05ef6ff2020-08-29 11:27:05 -07002use crate::syntax::file::Module;
David Tolnay08419302020-04-19 20:38:20 -07003use crate::syntax::namespace::Namespace;
David Tolnay0dd85ff2020-05-03 23:43:33 -07004use crate::syntax::report::Errors;
David Tolnay891061b2020-04-19 22:42:33 -07005use crate::syntax::symbol::Symbol;
David Tolnay3caa50a2020-04-19 21:25:34 -07006use crate::syntax::{
David Tolnay0531f432020-10-03 23:50:28 -07007 self, check, mangle, Api, Enum, ExternFn, ExternType, Impl, Signature, Struct, Type, TypeAlias,
8 Types,
David Tolnay3caa50a2020-04-19 21:25:34 -07009};
David Tolnay0531f432020-10-03 23:50:28 -070010use proc_macro2::{Ident, Span, TokenStream};
David Tolnayf9ffb932020-04-20 02:22:57 -070011use quote::{format_ident, quote, quote_spanned, ToTokens};
David Tolnay05ef6ff2020-08-29 11:27:05 -070012use std::mem;
13use syn::{parse_quote, Result, Token};
David Tolnay7db73692019-10-20 14:51:12 -040014
David Tolnay3c64a4e2020-08-29 14:07:38 -070015pub fn bridge(mut ffi: Module) -> Result<TokenStream> {
David Tolnay0dd85ff2020-05-03 23:43:33 -070016 let ref mut errors = Errors::new();
David Tolnay05ef6ff2020-08-29 11:27:05 -070017 let content = mem::take(&mut ffi.content);
David Tolnay805dca32020-08-29 19:09:55 -070018 let trusted = ffi.unsafety.is_some();
19 let ref apis = syntax::parse_items(errors, content, trusted);
David Tolnay52759782020-05-03 23:59:40 -070020 let ref types = Types::collect(errors, apis);
21 errors.propagate()?;
David Tolnay3c64a4e2020-08-29 14:07:38 -070022 let namespace = &ffi.namespace;
David Tolnay0dd85ff2020-05-03 23:43:33 -070023 check::typecheck(errors, namespace, apis, types);
24 errors.propagate()?;
David Tolnay7db73692019-10-20 14:51:12 -040025
David Tolnay3c64a4e2020-08-29 14:07:38 -070026 Ok(expand(ffi, apis, types))
David Tolnaycbf3f032020-05-03 23:51:14 -070027}
28
David Tolnay3c64a4e2020-08-29 14:07:38 -070029fn expand(ffi: Module, apis: &[Api], types: &Types) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -040030 let mut expanded = TokenStream::new();
31 let mut hidden = TokenStream::new();
David Tolnay3c64a4e2020-08-29 14:07:38 -070032 let namespace = &ffi.namespace;
David Tolnay7db73692019-10-20 14:51:12 -040033
David Tolnay2ec14632020-05-04 00:47:10 -070034 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040035 if let Api::RustType(ety) = api {
36 expanded.extend(expand_rust_type(ety));
David Tolnayc8361022020-08-25 21:57:53 -070037 hidden.extend(expand_rust_type_assert_sized(ety));
David Tolnay7db73692019-10-20 14:51:12 -040038 }
39 }
40
David Tolnay2ec14632020-05-04 00:47:10 -070041 for api in apis {
David Tolnay7db73692019-10-20 14:51:12 -040042 match api {
David Tolnay7e69f892020-10-03 22:20:22 -070043 Api::Include(_) | Api::RustType(_) | Api::Impl(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -040044 Api::Struct(strct) => expanded.extend(expand_struct(strct)),
Joel Galensonc03402a2020-04-23 17:31:09 -070045 Api::Enum(enm) => expanded.extend(expand_enum(enm)),
Joel Galenson905eb2e2020-05-04 14:58:14 -070046 Api::CxxType(ety) => {
David Tolnaya593d6e2020-08-29 19:48:08 -070047 let ident = &ety.ident;
48 if !types.structs.contains_key(ident) && !types.enums.contains_key(ident) {
David Tolnay5f9e8ca2020-05-07 17:21:05 -070049 expanded.extend(expand_cxx_type(namespace, ety));
Joel Galenson905eb2e2020-05-04 14:58:14 -070050 }
51 }
David Tolnay7db73692019-10-20 14:51:12 -040052 Api::CxxFunction(efn) => {
53 expanded.extend(expand_cxx_function_shim(namespace, efn, types));
54 }
55 Api::RustFunction(efn) => {
56 hidden.extend(expand_rust_function_shim(namespace, efn, types))
57 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -070058 Api::TypeAlias(alias) => {
59 expanded.extend(expand_type_alias(alias));
David Tolnay63a0e4e2020-10-03 18:52:32 -070060 hidden.extend(expand_type_alias_verify(namespace, alias, types));
David Tolnay5f9e8ca2020-05-07 17:21:05 -070061 }
David Tolnay7db73692019-10-20 14:51:12 -040062 }
63 }
64
65 for ty in types {
David Tolnay0531f432020-10-03 23:50:28 -070066 let explicit_impl = types.explicit_impls.get(ty);
David Tolnay7db73692019-10-20 14:51:12 -040067 if let Type::RustBox(ty) = ty {
68 if let Type::Ident(ident) = &ty.inner {
69 if Atom::from(ident).is_none() {
70 hidden.extend(expand_rust_box(namespace, ident));
71 }
72 }
Myron Ahneba35cf2020-02-05 19:41:51 +070073 } else if let Type::RustVec(ty) = ty {
74 if let Type::Ident(ident) = &ty.inner {
David Tolnaya006bca2020-04-25 11:28:13 -070075 if Atom::from(ident).is_none() {
David Tolnay83a7ec92020-04-25 12:37:42 -070076 hidden.extend(expand_rust_vec(namespace, ident));
David Tolnaya006bca2020-04-25 11:28:13 -070077 }
Myron Ahneba35cf2020-02-05 19:41:51 +070078 }
David Tolnay7db73692019-10-20 14:51:12 -040079 } else if let Type::UniquePtr(ptr) = ty {
80 if let Type::Ident(ident) = &ptr.inner {
David Tolnay7e69f892020-10-03 22:20:22 -070081 if Atom::from(ident).is_none()
David Tolnay0531f432020-10-03 23:50:28 -070082 && (explicit_impl.is_some() || !types.aliases.contains_key(ident))
David Tolnay7e69f892020-10-03 22:20:22 -070083 {
David Tolnay0531f432020-10-03 23:50:28 -070084 expanded.extend(expand_unique_ptr(namespace, ident, types, explicit_impl));
Myron Ahneba35cf2020-02-05 19:41:51 +070085 }
86 }
David Tolnay4377a9e2020-04-24 15:20:26 -070087 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +070088 if let Type::Ident(ident) = &ptr.inner {
David Tolnay7e69f892020-10-03 22:20:22 -070089 if Atom::from(ident).is_none()
David Tolnay0531f432020-10-03 23:50:28 -070090 && (explicit_impl.is_some() || !types.aliases.contains_key(ident))
David Tolnay7e69f892020-10-03 22:20:22 -070091 {
David Tolnay17631102020-04-24 19:01:30 -070092 // Generate impl for CxxVector<T> if T is a struct or opaque
93 // C++ type. Impl for primitives is already provided by cxx
94 // crate.
David Tolnay0531f432020-10-03 23:50:28 -070095 expanded.extend(expand_cxx_vector(namespace, ident, explicit_impl));
David Tolnay7db73692019-10-20 14:51:12 -040096 }
97 }
98 }
99 }
100
101 // Work around https://github.com/rust-lang/rust/issues/67851.
102 if !hidden.is_empty() {
103 expanded.extend(quote! {
104 #[doc(hidden)]
105 const _: () = {
106 #hidden
107 };
108 });
109 }
110
111 let attrs = ffi
112 .attrs
113 .into_iter()
114 .filter(|attr| attr.path.is_ident("doc"));
115 let vis = &ffi.vis;
David Tolnaycbf3f032020-05-03 23:51:14 -0700116 let ident = &ffi.ident;
David Tolnay7db73692019-10-20 14:51:12 -0400117
David Tolnaycbf3f032020-05-03 23:51:14 -0700118 quote! {
David Tolnay7db73692019-10-20 14:51:12 -0400119 #(#attrs)*
120 #[deny(improper_ctypes)]
121 #[allow(non_snake_case)]
122 #vis mod #ident {
123 #expanded
124 }
David Tolnaycbf3f032020-05-03 23:51:14 -0700125 }
David Tolnay7db73692019-10-20 14:51:12 -0400126}
127
128fn expand_struct(strct: &Struct) -> TokenStream {
129 let ident = &strct.ident;
130 let doc = &strct.doc;
131 let derives = &strct.derives;
132 let fields = strct.fields.iter().map(|field| {
133 // This span on the pub makes "private type in public interface" errors
134 // appear in the right place.
135 let vis = Token![pub](field.ident.span());
136 quote!(#vis #field)
137 });
138 quote! {
139 #doc
140 #[derive(#(#derives),*)]
141 #[repr(C)]
142 pub struct #ident {
143 #(#fields,)*
144 }
145 }
146}
147
Joel Galensonc03402a2020-04-23 17:31:09 -0700148fn expand_enum(enm: &Enum) -> TokenStream {
149 let ident = &enm.ident;
150 let doc = &enm.doc;
David Tolnayc605e6f2020-05-10 23:37:12 -0700151 let repr = enm.repr;
Joel Galenson88547732020-05-05 08:23:42 -0700152 let variants = enm.variants.iter().map(|variant| {
153 let variant_ident = &variant.ident;
154 let discriminant = &variant.discriminant;
155 Some(quote! {
156 pub const #variant_ident: Self = #ident { repr: #discriminant };
157 })
158 });
Joel Galensonc03402a2020-04-23 17:31:09 -0700159 quote! {
160 #doc
161 #[derive(Copy, Clone, PartialEq, Eq)]
162 #[repr(transparent)]
Joel Galensondb1ec312020-05-01 13:57:32 -0700163 pub struct #ident {
David Tolnayc605e6f2020-05-10 23:37:12 -0700164 pub repr: #repr,
Joel Galensondb1ec312020-05-01 13:57:32 -0700165 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700166
167 #[allow(non_upper_case_globals)]
168 impl #ident {
David Tolnayd7984c22020-04-30 20:09:31 -0700169 #(#variants)*
Joel Galensonc03402a2020-04-23 17:31:09 -0700170 }
171 }
172}
173
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700174fn expand_cxx_type(namespace: &Namespace, ety: &ExternType) -> TokenStream {
David Tolnay7db73692019-10-20 14:51:12 -0400175 let ident = &ety.ident;
176 let doc = &ety.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700177 let type_id = type_id(namespace, ident);
178
David Tolnay7db73692019-10-20 14:51:12 -0400179 quote! {
180 #doc
181 #[repr(C)]
182 pub struct #ident {
183 _private: ::cxx::private::Opaque,
184 }
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700185
186 unsafe impl ::cxx::ExternType for #ident {
187 type Id = #type_id;
David Tolnay38f5ad62020-10-03 18:06:44 -0700188 type Kind = ::cxx::kind::Opaque;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700189 }
David Tolnay7db73692019-10-20 14:51:12 -0400190 }
191}
192
193fn expand_cxx_function_decl(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
194 let ident = &efn.ident;
David Tolnay18ba92c2020-04-22 16:17:30 -0700195 let receiver = efn.receiver.iter().map(|receiver| {
196 let receiver_type = receiver.ty();
197 quote!(_: #receiver_type)
198 });
David Tolnay39d575f2020-03-03 00:10:56 -0800199 let args = efn.args.iter().map(|arg| {
200 let ident = &arg.ident;
201 let ty = expand_extern_type(&arg.ty);
David Tolnaya46a2372020-03-06 10:03:48 -0800202 if arg.ty == RustString {
203 quote!(#ident: *const #ty)
David Tolnay313b10e2020-04-25 16:30:51 -0700204 } else if let Type::RustVec(_) = arg.ty {
205 quote!(#ident: *const #ty)
David Tolnay75dca2e2020-03-25 20:17:52 -0700206 } else if let Type::Fn(_) = arg.ty {
207 quote!(#ident: ::cxx::private::FatFunction)
David Tolnaya46a2372020-03-06 10:03:48 -0800208 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay39d575f2020-03-03 00:10:56 -0800209 quote!(#ident: *mut #ty)
210 } else {
211 quote!(#ident: #ty)
212 }
213 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700214 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700215 let ret = if efn.throws {
216 quote!(-> ::cxx::private::Result)
217 } else {
218 expand_extern_return_type(&efn.ret, types)
219 };
David Tolnay7db73692019-10-20 14:51:12 -0400220 let mut outparam = None;
David Tolnay1e548172020-03-16 13:37:09 -0700221 if indirect_return(efn, types) {
David Tolnay7db73692019-10-20 14:51:12 -0400222 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
223 outparam = Some(quote!(__return: *mut #ret));
224 }
David Tolnay3caa50a2020-04-19 21:25:34 -0700225 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay7db73692019-10-20 14:51:12 -0400226 let local_name = format_ident!("__{}", ident);
227 quote! {
228 #[link_name = #link_name]
Joel Galenson3d4f6122020-04-07 15:54:05 -0700229 fn #local_name(#(#all_args,)* #outparam) #ret;
David Tolnay7db73692019-10-20 14:51:12 -0400230 }
231}
232
233fn expand_cxx_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
234 let ident = &efn.ident;
235 let doc = &efn.doc;
236 let decl = expand_cxx_function_decl(namespace, efn, types);
David Tolnayfb6e3862020-04-20 01:33:23 -0700237 let receiver = efn.receiver.iter().map(|receiver| {
David Tolnayf9ffb932020-04-20 02:22:57 -0700238 let ampersand = receiver.ampersand;
David Tolnayfb6e3862020-04-20 01:33:23 -0700239 let mutability = receiver.mutability;
David Tolnayf9ffb932020-04-20 02:22:57 -0700240 let var = receiver.var;
241 quote!(#ampersand #mutability #var)
David Tolnayfb6e3862020-04-20 01:33:23 -0700242 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700243 let args = efn.args.iter().map(|arg| quote!(#arg));
244 let all_args = receiver.chain(args);
David Tolnayebef4a22020-03-17 15:33:47 -0700245 let ret = if efn.throws {
246 let ok = match &efn.ret {
247 Some(ret) => quote!(#ret),
248 None => quote!(()),
249 };
250 quote!(-> ::std::result::Result<#ok, ::cxx::Exception>)
251 } else {
252 expand_return_type(&efn.ret)
253 };
David Tolnay1e548172020-03-16 13:37:09 -0700254 let indirect_return = indirect_return(efn, types);
David Tolnayf9ffb932020-04-20 02:22:57 -0700255 let receiver_var = efn
256 .receiver
257 .iter()
258 .map(|receiver| receiver.var.to_token_stream());
Joel Galenson3d4f6122020-04-07 15:54:05 -0700259 let arg_vars = efn.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400260 let var = &arg.ident;
261 match &arg.ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800262 Type::Ident(ident) if ident == RustString => {
David Tolnaya46a2372020-03-06 10:03:48 -0800263 quote!(#var.as_mut_ptr() as *const ::cxx::private::RustString)
David Tolnay7db73692019-10-20 14:51:12 -0400264 }
265 Type::RustBox(_) => quote!(::std::boxed::Box::into_raw(#var)),
266 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::into_raw(#var)),
David Tolnay313b10e2020-04-25 16:30:51 -0700267 Type::RustVec(_) => quote!(#var.as_mut_ptr() as *const ::cxx::private::RustVec<_>),
David Tolnay7db73692019-10-20 14:51:12 -0400268 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700269 Type::Ident(ident) if ident == RustString => match ty.mutability {
270 None => quote!(::cxx::private::RustString::from_ref(#var)),
271 Some(_) => quote!(::cxx::private::RustString::from_mut(#var)),
272 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700273 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
274 None => quote!(::cxx::private::RustVec::from_ref_vec_string(#var)),
275 Some(_) => quote!(::cxx::private::RustVec::from_mut_vec_string(#var)),
276 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700277 Type::RustVec(_) => match ty.mutability {
278 None => quote!(::cxx::private::RustVec::from_ref(#var)),
279 Some(_) => quote!(::cxx::private::RustVec::from_mut(#var)),
280 },
David Tolnay7db73692019-10-20 14:51:12 -0400281 _ => quote!(#var),
282 },
283 Type::Str(_) => quote!(::cxx::private::RustStr::from(#var)),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700284 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8::from(#var)),
David Tolnay7db73692019-10-20 14:51:12 -0400285 ty if types.needs_indirect_abi(ty) => quote!(#var.as_mut_ptr()),
286 _ => quote!(#var),
287 }
288 });
Joel Galenson3d4f6122020-04-07 15:54:05 -0700289 let vars = receiver_var.chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700290 let trampolines = efn
291 .args
292 .iter()
293 .filter_map(|arg| {
294 if let Type::Fn(f) = &arg.ty {
295 let var = &arg.ident;
296 Some(expand_function_pointer_trampoline(
297 namespace, efn, var, f, types,
298 ))
299 } else {
300 None
301 }
302 })
303 .collect::<TokenStream>();
David Tolnay7db73692019-10-20 14:51:12 -0400304 let mut setup = efn
305 .args
306 .iter()
307 .filter(|arg| types.needs_indirect_abi(&arg.ty))
308 .map(|arg| {
309 let var = &arg.ident;
310 // These are arguments for which C++ has taken ownership of the data
311 // behind the mut reference it received.
312 quote! {
David Tolnaycb4ee4b2020-04-24 11:28:08 -0700313 let mut #var = ::std::mem::MaybeUninit::new(#var);
David Tolnay7db73692019-10-20 14:51:12 -0400314 }
315 })
316 .collect::<TokenStream>();
317 let local_name = format_ident!("__{}", ident);
318 let call = if indirect_return {
319 let ret = expand_extern_type(efn.ret.as_ref().unwrap());
320 setup.extend(quote! {
321 let mut __return = ::std::mem::MaybeUninit::<#ret>::uninit();
David Tolnay7db73692019-10-20 14:51:12 -0400322 });
David Tolnayebef4a22020-03-17 15:33:47 -0700323 if efn.throws {
324 setup.extend(quote! {
325 #local_name(#(#vars,)* __return.as_mut_ptr()).exception()?;
326 });
327 quote!(::std::result::Result::Ok(__return.assume_init()))
328 } else {
329 setup.extend(quote! {
330 #local_name(#(#vars,)* __return.as_mut_ptr());
331 });
332 quote!(__return.assume_init())
333 }
334 } else if efn.throws {
David Tolnay7db73692019-10-20 14:51:12 -0400335 quote! {
David Tolnayebef4a22020-03-17 15:33:47 -0700336 #local_name(#(#vars),*).exception()
David Tolnay7db73692019-10-20 14:51:12 -0400337 }
338 } else {
339 quote! {
340 #local_name(#(#vars),*)
341 }
342 };
Myron Ahn84849302020-03-25 22:00:58 +0700343 let expr = if efn.throws {
344 efn.ret.as_ref().and_then(|ret| match ret {
345 Type::Ident(ident) if ident == RustString => {
346 Some(quote!(#call.map(|r| r.into_string())))
347 }
348 Type::RustBox(_) => Some(quote!(#call.map(|r| ::std::boxed::Box::from_raw(r)))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700349 Type::RustVec(vec) => {
350 if vec.inner == RustString {
351 Some(quote!(#call.map(|r| r.into_vec_string())))
352 } else {
353 Some(quote!(#call.map(|r| r.into_vec())))
354 }
355 }
Myron Ahn84849302020-03-25 22:00:58 +0700356 Type::UniquePtr(_) => Some(quote!(#call.map(|r| ::cxx::UniquePtr::from_raw(r)))),
357 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700358 Type::Ident(ident) if ident == RustString => match ty.mutability {
359 None => Some(quote!(#call.map(|r| r.as_string()))),
360 Some(_) => Some(quote!(#call.map(|r| r.as_mut_string()))),
361 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700362 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
363 None => Some(quote!(#call.map(|r| r.as_vec_string()))),
364 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec_string()))),
365 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700366 Type::RustVec(_) => match ty.mutability {
367 None => Some(quote!(#call.map(|r| r.as_vec()))),
368 Some(_) => Some(quote!(#call.map(|r| r.as_mut_vec()))),
369 },
Myron Ahn84849302020-03-25 22:00:58 +0700370 _ => None,
371 },
372 Type::Str(_) => Some(quote!(#call.map(|r| r.as_str()))),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700373 Type::SliceRefU8(_) => Some(quote!(#call.map(|r| r.as_slice()))),
Myron Ahn84849302020-03-25 22:00:58 +0700374 _ => None,
375 })
376 } else {
377 efn.ret.as_ref().and_then(|ret| match ret {
David Tolnaya52602b2020-03-06 10:24:34 -0800378 Type::Ident(ident) if ident == RustString => Some(quote!(#call.into_string())),
David Tolnay7db73692019-10-20 14:51:12 -0400379 Type::RustBox(_) => Some(quote!(::std::boxed::Box::from_raw(#call))),
David Tolnay33f56ad2020-08-27 17:06:35 -0700380 Type::RustVec(vec) => {
381 if vec.inner == RustString {
382 Some(quote!(#call.into_vec_string()))
383 } else {
384 Some(quote!(#call.into_vec()))
385 }
386 }
David Tolnay7db73692019-10-20 14:51:12 -0400387 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::from_raw(#call))),
388 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700389 Type::Ident(ident) if ident == RustString => match ty.mutability {
390 None => Some(quote!(#call.as_string())),
391 Some(_) => Some(quote!(#call.as_mut_string())),
392 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700393 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
394 None => Some(quote!(#call.as_vec_string())),
395 Some(_) => Some(quote!(#call.as_mut_vec_string())),
396 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700397 Type::RustVec(_) => match ty.mutability {
398 None => Some(quote!(#call.as_vec())),
399 Some(_) => Some(quote!(#call.as_mut_vec())),
400 },
David Tolnay7db73692019-10-20 14:51:12 -0400401 _ => None,
402 },
403 Type::Str(_) => Some(quote!(#call.as_str())),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700404 Type::SliceRefU8(_) => Some(quote!(#call.as_slice())),
David Tolnay7db73692019-10-20 14:51:12 -0400405 _ => None,
406 })
Myron Ahn84849302020-03-25 22:00:58 +0700407 }
408 .unwrap_or(call);
David Tolnaye67bcf52020-09-06 23:50:44 -0700409 let mut dispatch = quote!(#setup #expr);
410 let unsafety = &efn.sig.unsafety;
411 if unsafety.is_none() {
412 dispatch = quote!(unsafe { #dispatch });
413 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700414 let function_shim = quote! {
415 #doc
David Tolnaye67bcf52020-09-06 23:50:44 -0700416 pub #unsafety fn #ident(#(#all_args,)*) #ret {
David Tolnayc66cdbb2020-04-20 01:41:15 -0700417 extern "C" {
418 #decl
David Tolnay7db73692019-10-20 14:51:12 -0400419 }
David Tolnayc66cdbb2020-04-20 01:41:15 -0700420 #trampolines
David Tolnaye67bcf52020-09-06 23:50:44 -0700421 #dispatch
David Tolnayc66cdbb2020-04-20 01:41:15 -0700422 }
423 };
424 match &efn.receiver {
425 None => function_shim,
426 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700427 let receiver_type = &receiver.ty;
David Tolnayc66cdbb2020-04-20 01:41:15 -0700428 quote!(impl #receiver_type { #function_shim })
429 }
David Tolnay7db73692019-10-20 14:51:12 -0400430 }
431}
432
David Tolnay75dca2e2020-03-25 20:17:52 -0700433fn expand_function_pointer_trampoline(
434 namespace: &Namespace,
435 efn: &ExternFn,
436 var: &Ident,
437 sig: &Signature,
438 types: &Types,
439) -> TokenStream {
David Tolnay891061b2020-04-19 22:42:33 -0700440 let c_trampoline = mangle::c_trampoline(namespace, efn, var);
441 let r_trampoline = mangle::r_trampoline(namespace, efn, var);
David Tolnay75dca2e2020-03-25 20:17:52 -0700442 let local_name = parse_quote!(__);
443 let catch_unwind_label = format!("::{}::{}", efn.ident, var);
444 let shim = expand_rust_function_shim_impl(
445 sig,
446 types,
447 &r_trampoline,
448 local_name,
449 catch_unwind_label,
450 None,
451 );
452
453 quote! {
454 let #var = ::cxx::private::FatFunction {
455 trampoline: {
456 extern "C" {
457 #[link_name = #c_trampoline]
458 fn trampoline();
459 }
460 #shim
461 trampoline as usize as *const ()
462 },
463 ptr: #var as usize as *const (),
464 };
465 }
466}
467
David Tolnay7db73692019-10-20 14:51:12 -0400468fn expand_rust_type(ety: &ExternType) -> TokenStream {
469 let ident = &ety.ident;
470 quote! {
471 use super::#ident;
472 }
473}
474
David Tolnayc8361022020-08-25 21:57:53 -0700475fn expand_rust_type_assert_sized(ety: &ExternType) -> TokenStream {
476 // Rustc will render as follows if not sized:
477 //
478 // type TheirType;
479 // -----^^^^^^^^^-
480 // | |
481 // | doesn't have a size known at compile-time
482 // required by this bound in `ffi::_::__AssertSized`
483
484 let ident = &ety.ident;
485 let begin_span = Token![::](ety.type_token.span);
486 let sized = quote_spanned! {ety.semi_token.span=>
487 #begin_span std::marker::Sized
488 };
489 quote_spanned! {ident.span()=>
490 let _ = {
491 fn __AssertSized<T: ?#sized + #sized>() {}
492 __AssertSized::<#ident>
493 };
494 }
495}
496
David Tolnay7db73692019-10-20 14:51:12 -0400497fn expand_rust_function_shim(namespace: &Namespace, efn: &ExternFn, types: &Types) -> TokenStream {
498 let ident = &efn.ident;
David Tolnay3caa50a2020-04-19 21:25:34 -0700499 let link_name = mangle::extern_fn(namespace, efn);
David Tolnay75dca2e2020-03-25 20:17:52 -0700500 let local_name = format_ident!("__{}", ident);
501 let catch_unwind_label = format!("::{}", ident);
502 let invoke = Some(ident);
503 expand_rust_function_shim_impl(
504 efn,
505 types,
506 &link_name,
507 local_name,
508 catch_unwind_label,
509 invoke,
510 )
511}
512
513fn expand_rust_function_shim_impl(
514 sig: &Signature,
515 types: &Types,
David Tolnay891061b2020-04-19 22:42:33 -0700516 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700517 local_name: Ident,
518 catch_unwind_label: String,
519 invoke: Option<&Ident>,
520) -> TokenStream {
David Tolnayf9ffb932020-04-20 02:22:57 -0700521 let receiver_var = sig
522 .receiver
523 .as_ref()
524 .map(|receiver| quote_spanned!(receiver.var.span=> __self));
David Tolnay18ba92c2020-04-22 16:17:30 -0700525 let receiver = sig.receiver.as_ref().map(|receiver| {
526 let receiver_type = receiver.ty();
527 quote!(#receiver_var: #receiver_type)
528 });
David Tolnay75dca2e2020-03-25 20:17:52 -0700529 let args = sig.args.iter().map(|arg| {
David Tolnay39d575f2020-03-03 00:10:56 -0800530 let ident = &arg.ident;
531 let ty = expand_extern_type(&arg.ty);
532 if types.needs_indirect_abi(&arg.ty) {
533 quote!(#ident: *mut #ty)
534 } else {
535 quote!(#ident: #ty)
536 }
537 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700538 let all_args = receiver.into_iter().chain(args);
David Tolnay75dca2e2020-03-25 20:17:52 -0700539
David Tolnay3a45f2d2020-04-20 01:51:12 -0700540 let arg_vars = sig.args.iter().map(|arg| {
David Tolnay7db73692019-10-20 14:51:12 -0400541 let ident = &arg.ident;
David Tolnay17955e22020-01-20 17:58:24 -0800542 match &arg.ty {
David Tolnaycc3767f2020-03-06 10:41:51 -0800543 Type::Ident(i) if i == RustString => {
544 quote!(::std::mem::take((*#ident).as_mut_string()))
545 }
David Tolnay40226ab2020-03-03 00:05:35 -0800546 Type::RustBox(_) => quote!(::std::boxed::Box::from_raw(#ident)),
David Tolnay33f56ad2020-08-27 17:06:35 -0700547 Type::RustVec(vec) => {
548 if vec.inner == RustString {
549 quote!(::std::mem::take((*#ident).as_mut_vec_string()))
550 } else {
551 quote!(::std::mem::take((*#ident).as_mut_vec()))
552 }
553 }
David Tolnay40226ab2020-03-03 00:05:35 -0800554 Type::UniquePtr(_) => quote!(::cxx::UniquePtr::from_raw(#ident)),
David Tolnay17955e22020-01-20 17:58:24 -0800555 Type::Ref(ty) => match &ty.inner {
David Tolnayf1c7f322020-08-27 00:46:01 -0700556 Type::Ident(i) if i == RustString => match ty.mutability {
557 None => quote!(#ident.as_string()),
558 Some(_) => quote!(#ident.as_mut_string()),
559 },
David Tolnay33f56ad2020-08-27 17:06:35 -0700560 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
561 None => quote!(#ident.as_vec_string()),
562 Some(_) => quote!(#ident.as_mut_vec_string()),
563 },
David Tolnayf1c7f322020-08-27 00:46:01 -0700564 Type::RustVec(_) => match ty.mutability {
565 None => quote!(#ident.as_vec()),
566 Some(_) => quote!(#ident.as_mut_vec()),
567 },
David Tolnay40226ab2020-03-03 00:05:35 -0800568 _ => quote!(#ident),
David Tolnay17955e22020-01-20 17:58:24 -0800569 },
David Tolnay40226ab2020-03-03 00:05:35 -0800570 Type::Str(_) => quote!(#ident.as_str()),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700571 Type::SliceRefU8(_) => quote!(#ident.as_slice()),
David Tolnay40226ab2020-03-03 00:05:35 -0800572 ty if types.needs_indirect_abi(ty) => quote!(::std::ptr::read(#ident)),
573 _ => quote!(#ident),
David Tolnay7db73692019-10-20 14:51:12 -0400574 }
575 });
David Tolnayf9ffb932020-04-20 02:22:57 -0700576 let vars = receiver_var.into_iter().chain(arg_vars);
David Tolnay75dca2e2020-03-25 20:17:52 -0700577
578 let mut call = match invoke {
David Tolnay3a45f2d2020-04-20 01:51:12 -0700579 Some(ident) => match &sig.receiver {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700580 None => quote!(super::#ident),
David Tolnay3a45f2d2020-04-20 01:51:12 -0700581 Some(receiver) => {
David Tolnay05e11cc2020-04-20 02:13:56 -0700582 let receiver_type = &receiver.ty;
David Tolnay3a45f2d2020-04-20 01:51:12 -0700583 quote!(#receiver_type::#ident)
584 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700585 },
David Tolnay75dca2e2020-03-25 20:17:52 -0700586 None => quote!(__extern),
587 };
588 call.extend(quote! { (#(#vars),*) });
589
David Tolnay22602b42020-09-21 18:04:05 -0400590 let conversion = sig.ret.as_ref().and_then(|ret| match ret {
591 Type::Ident(ident) if ident == RustString => Some(quote!(::cxx::private::RustString::from)),
592 Type::RustBox(_) => Some(quote!(::std::boxed::Box::into_raw)),
593 Type::RustVec(vec) => {
594 if vec.inner == RustString {
595 Some(quote!(::cxx::private::RustVec::from_vec_string))
596 } else {
597 Some(quote!(::cxx::private::RustVec::from))
David Tolnay7db73692019-10-20 14:51:12 -0400598 }
David Tolnay22602b42020-09-21 18:04:05 -0400599 }
600 Type::UniquePtr(_) => Some(quote!(::cxx::UniquePtr::into_raw)),
601 Type::Ref(ty) => match &ty.inner {
602 Type::Ident(ident) if ident == RustString => match ty.mutability {
603 None => Some(quote!(::cxx::private::RustString::from_ref)),
604 Some(_) => Some(quote!(::cxx::private::RustString::from_mut)),
David Tolnay7db73692019-10-20 14:51:12 -0400605 },
David Tolnay22602b42020-09-21 18:04:05 -0400606 Type::RustVec(vec) if vec.inner == RustString => match ty.mutability {
607 None => Some(quote!(::cxx::private::RustVec::from_ref_vec_string)),
608 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut_vec_string)),
609 },
610 Type::RustVec(_) => match ty.mutability {
611 None => Some(quote!(::cxx::private::RustVec::from_ref)),
612 Some(_) => Some(quote!(::cxx::private::RustVec::from_mut)),
613 },
David Tolnay7db73692019-10-20 14:51:12 -0400614 _ => None,
David Tolnay22602b42020-09-21 18:04:05 -0400615 },
616 Type::Str(_) => Some(quote!(::cxx::private::RustStr::from)),
617 Type::SliceRefU8(_) => Some(quote!(::cxx::private::RustSliceU8::from)),
618 _ => None,
619 });
620
621 let mut expr = match conversion {
622 None => call,
623 Some(conversion) if !sig.throws => quote!(#conversion(#call)),
624 Some(conversion) => quote!(::std::result::Result::map(#call, #conversion)),
625 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700626
627 let mut outparam = None;
628 let indirect_return = indirect_return(sig, types);
David Tolnay1e548172020-03-16 13:37:09 -0700629 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700630 let ret = expand_extern_type(sig.ret.as_ref().unwrap());
631 outparam = Some(quote!(__return: *mut #ret,));
David Tolnay1e548172020-03-16 13:37:09 -0700632 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700633 if sig.throws {
634 let out = match sig.ret {
David Tolnaycecada62020-03-17 01:45:58 -0700635 Some(_) => quote!(__return),
636 None => quote!(&mut ()),
637 };
638 expr = quote!(::cxx::private::r#try(#out, #expr));
David Tolnay1e548172020-03-16 13:37:09 -0700639 } else if indirect_return {
David Tolnay7db73692019-10-20 14:51:12 -0400640 expr = quote!(::std::ptr::write(__return, #expr));
641 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700642
David Tolnay1e548172020-03-16 13:37:09 -0700643 expr = quote!(::cxx::private::catch_unwind(__fn, move || #expr));
David Tolnay75dca2e2020-03-25 20:17:52 -0700644
645 let ret = if sig.throws {
David Tolnay486b6ec2020-03-17 01:19:57 -0700646 quote!(-> ::cxx::private::Result)
David Tolnay1e548172020-03-16 13:37:09 -0700647 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700648 expand_extern_return_type(&sig.ret, types)
David Tolnay1e548172020-03-16 13:37:09 -0700649 };
David Tolnay75dca2e2020-03-25 20:17:52 -0700650
651 let pointer = match invoke {
652 None => Some(quote!(__extern: #sig)),
653 Some(_) => None,
654 };
655
David Tolnay7db73692019-10-20 14:51:12 -0400656 quote! {
657 #[doc(hidden)]
658 #[export_name = #link_name]
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700659 unsafe extern "C" fn #local_name(#(#all_args,)* #outparam #pointer) #ret {
David Tolnay7db73692019-10-20 14:51:12 -0400660 let __fn = concat!(module_path!(), #catch_unwind_label);
661 #expr
662 }
663 }
664}
665
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700666fn expand_type_alias(alias: &TypeAlias) -> TokenStream {
Bryan Henry890083d2020-09-13 10:34:31 -0700667 let doc = &alias.doc;
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700668 let ident = &alias.ident;
669 let ty = &alias.ty;
670 quote! {
Bryan Henry890083d2020-09-13 10:34:31 -0700671 #doc
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700672 pub type #ident = #ty;
673 }
674}
675
David Tolnay63a0e4e2020-10-03 18:52:32 -0700676fn expand_type_alias_verify(
677 namespace: &Namespace,
678 alias: &TypeAlias,
679 types: &Types,
680) -> TokenStream {
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700681 let ident = &alias.ident;
682 let type_id = type_id(namespace, ident);
David Tolnay83fe0f02020-05-07 20:00:15 -0700683 let begin_span = alias.type_token.span;
684 let end_span = alias.semi_token.span;
685 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_type::<);
686 let end = quote_spanned!(end_span=> >);
687
David Tolnay63a0e4e2020-10-03 18:52:32 -0700688 let mut verify = quote! {
David Tolnay83fe0f02020-05-07 20:00:15 -0700689 const _: fn() = #begin #ident, #type_id #end;
David Tolnay63a0e4e2020-10-03 18:52:32 -0700690 };
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700691
David Tolnay3208fd72020-10-03 20:19:40 -0700692 if types.required_trivial.contains_key(&alias.ident) {
David Tolnay63a0e4e2020-10-03 18:52:32 -0700693 let begin = quote_spanned!(begin_span=> ::cxx::private::verify_extern_kind::<);
694 verify.extend(quote! {
695 const _: fn() = #begin #ident, ::cxx::kind::Trivial #end;
696 });
Adrian Taylorc7043292020-09-25 12:48:36 -0700697 }
David Tolnay63a0e4e2020-10-03 18:52:32 -0700698
699 verify
Adrian Taylorc7043292020-09-25 12:48:36 -0700700}
701
David Tolnay5f9e8ca2020-05-07 17:21:05 -0700702fn type_id(namespace: &Namespace, ident: &Ident) -> TokenStream {
703 let mut path = String::new();
704 for name in namespace {
705 path += &name.to_string();
706 path += "::";
707 }
708 path += &ident.to_string();
709
710 quote! {
711 ::cxx::type_id!(#path)
712 }
713}
714
David Tolnay7db73692019-10-20 14:51:12 -0400715fn expand_rust_box(namespace: &Namespace, ident: &Ident) -> TokenStream {
David Tolnay591dcb62020-09-01 23:00:38 -0700716 let link_prefix = format!("cxxbridge04$box${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400717 let link_uninit = format!("{}uninit", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400718 let link_drop = format!("{}drop", link_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400719
720 let local_prefix = format_ident!("{}__box_", ident);
721 let local_uninit = format_ident!("{}uninit", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400722 let local_drop = format_ident!("{}drop", local_prefix);
David Tolnay7db73692019-10-20 14:51:12 -0400723
724 let span = ident.span();
725 quote_spanned! {span=>
726 #[doc(hidden)]
727 #[export_name = #link_uninit]
728 unsafe extern "C" fn #local_uninit(
729 this: *mut ::std::boxed::Box<::std::mem::MaybeUninit<#ident>>,
730 ) {
731 ::std::ptr::write(
732 this,
733 ::std::boxed::Box::new(::std::mem::MaybeUninit::uninit()),
734 );
735 }
736 #[doc(hidden)]
David Tolnay7db73692019-10-20 14:51:12 -0400737 #[export_name = #link_drop]
738 unsafe extern "C" fn #local_drop(this: *mut ::std::boxed::Box<#ident>) {
739 ::std::ptr::drop_in_place(this);
740 }
David Tolnay7db73692019-10-20 14:51:12 -0400741 }
742}
743
David Tolnay83a7ec92020-04-25 12:37:42 -0700744fn expand_rust_vec(namespace: &Namespace, elem: &Ident) -> TokenStream {
David Tolnay591dcb62020-09-01 23:00:38 -0700745 let link_prefix = format!("cxxbridge04$rust_vec${}{}$", namespace, elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700746 let link_new = format!("{}new", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700747 let link_drop = format!("{}drop", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700748 let link_len = format!("{}len", link_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700749 let link_data = format!("{}data", link_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700750 let link_stride = format!("{}stride", link_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700751
David Tolnay83a7ec92020-04-25 12:37:42 -0700752 let local_prefix = format_ident!("{}__vec_", elem);
David Tolnayf97c2d52020-04-25 16:37:48 -0700753 let local_new = format_ident!("{}new", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700754 let local_drop = format_ident!("{}drop", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700755 let local_len = format_ident!("{}len", local_prefix);
David Tolnay219c0792020-04-24 20:31:37 -0700756 let local_data = format_ident!("{}data", local_prefix);
David Tolnay503d0192020-04-24 22:18:56 -0700757 let local_stride = format_ident!("{}stride", local_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700758
David Tolnay83a7ec92020-04-25 12:37:42 -0700759 let span = elem.span();
Myron Ahneba35cf2020-02-05 19:41:51 +0700760 quote_spanned! {span=>
761 #[doc(hidden)]
David Tolnayf97c2d52020-04-25 16:37:48 -0700762 #[export_name = #link_new]
763 unsafe extern "C" fn #local_new(this: *mut ::cxx::private::RustVec<#elem>) {
764 ::std::ptr::write(this, ::cxx::private::RustVec::new());
765 }
766 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700767 #[export_name = #link_drop]
David Tolnay83a7ec92020-04-25 12:37:42 -0700768 unsafe extern "C" fn #local_drop(this: *mut ::cxx::private::RustVec<#elem>) {
David Tolnay4c64afb2020-04-24 11:30:18 -0700769 ::std::ptr::drop_in_place(this);
Myron Ahneba35cf2020-02-05 19:41:51 +0700770 }
David Tolnay85db5a02020-04-25 13:17:27 -0700771 #[doc(hidden)]
Myron Ahneba35cf2020-02-05 19:41:51 +0700772 #[export_name = #link_len]
David Tolnay83a7ec92020-04-25 12:37:42 -0700773 unsafe extern "C" fn #local_len(this: *const ::cxx::private::RustVec<#elem>) -> usize {
David Tolnay2cef5df2020-04-24 11:33:58 -0700774 (*this).len()
Myron Ahneba35cf2020-02-05 19:41:51 +0700775 }
David Tolnay219c0792020-04-24 20:31:37 -0700776 #[doc(hidden)]
777 #[export_name = #link_data]
David Tolnay83a7ec92020-04-25 12:37:42 -0700778 unsafe extern "C" fn #local_data(this: *const ::cxx::private::RustVec<#elem>) -> *const #elem {
David Tolnay219c0792020-04-24 20:31:37 -0700779 (*this).as_ptr()
780 }
David Tolnay503d0192020-04-24 22:18:56 -0700781 #[doc(hidden)]
782 #[export_name = #link_stride]
783 unsafe extern "C" fn #local_stride() -> usize {
David Tolnay83a7ec92020-04-25 12:37:42 -0700784 ::std::mem::size_of::<#elem>()
David Tolnay503d0192020-04-24 22:18:56 -0700785 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700786 }
787}
788
David Tolnay0531f432020-10-03 23:50:28 -0700789fn expand_unique_ptr(
790 namespace: &Namespace,
791 ident: &Ident,
792 types: &Types,
793 explicit_impl: Option<&Impl>,
794) -> TokenStream {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700795 let name = ident.to_string();
David Tolnay591dcb62020-09-01 23:00:38 -0700796 let prefix = format!("cxxbridge04$unique_ptr${}{}$", namespace, ident);
David Tolnay7db73692019-10-20 14:51:12 -0400797 let link_null = format!("{}null", prefix);
798 let link_new = format!("{}new", prefix);
799 let link_raw = format!("{}raw", prefix);
800 let link_get = format!("{}get", prefix);
801 let link_release = format!("{}release", prefix);
802 let link_drop = format!("{}drop", prefix);
803
David Tolnay3b40b6f2020-04-24 17:58:24 -0700804 let new_method = if types.structs.contains_key(ident) {
805 Some(quote! {
David Tolnay53838912020-04-09 20:56:44 -0700806 fn __new(mut value: Self) -> *mut ::std::ffi::c_void {
807 extern "C" {
808 #[link_name = #link_new]
809 fn __new(this: *mut *mut ::std::ffi::c_void, value: *mut #ident);
810 }
811 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
812 unsafe { __new(&mut repr, &mut value) }
813 repr
814 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700815 })
816 } else {
817 None
David Tolnay53838912020-04-09 20:56:44 -0700818 };
819
David Tolnay0531f432020-10-03 23:50:28 -0700820 let begin_span =
821 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
822 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
823 let unsafe_token = format_ident!("unsafe", span = begin_span);
824
825 quote_spanned! {end_span=>
826 #unsafe_token impl ::cxx::private::UniquePtrTarget for #ident {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700827 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay7db73692019-10-20 14:51:12 -0400828 fn __null() -> *mut ::std::ffi::c_void {
829 extern "C" {
830 #[link_name = #link_null]
831 fn __null(this: *mut *mut ::std::ffi::c_void);
832 }
833 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
834 unsafe { __null(&mut repr) }
835 repr
836 }
David Tolnay53838912020-04-09 20:56:44 -0700837 #new_method
David Tolnay7db73692019-10-20 14:51:12 -0400838 unsafe fn __raw(raw: *mut Self) -> *mut ::std::ffi::c_void {
839 extern "C" {
840 #[link_name = #link_raw]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700841 fn __raw(this: *mut *mut ::std::ffi::c_void, raw: *mut #ident);
David Tolnay7db73692019-10-20 14:51:12 -0400842 }
843 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
844 __raw(&mut repr, raw);
845 repr
846 }
847 unsafe fn __get(repr: *mut ::std::ffi::c_void) -> *const Self {
848 extern "C" {
849 #[link_name = #link_get]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700850 fn __get(this: *const *mut ::std::ffi::c_void) -> *const #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400851 }
852 __get(&repr)
853 }
854 unsafe fn __release(mut repr: *mut ::std::ffi::c_void) -> *mut Self {
855 extern "C" {
856 #[link_name = #link_release]
David Tolnay3b40b6f2020-04-24 17:58:24 -0700857 fn __release(this: *mut *mut ::std::ffi::c_void) -> *mut #ident;
David Tolnay7db73692019-10-20 14:51:12 -0400858 }
859 __release(&mut repr)
860 }
861 unsafe fn __drop(mut repr: *mut ::std::ffi::c_void) {
862 extern "C" {
863 #[link_name = #link_drop]
864 fn __drop(this: *mut *mut ::std::ffi::c_void);
865 }
866 __drop(&mut repr);
867 }
868 }
869 }
870}
871
David Tolnay0531f432020-10-03 23:50:28 -0700872fn expand_cxx_vector(
873 namespace: &Namespace,
874 elem: &Ident,
875 explicit_impl: Option<&Impl>,
876) -> TokenStream {
877 let _ = explicit_impl;
David Tolnay3b40b6f2020-04-24 17:58:24 -0700878 let name = elem.to_string();
David Tolnay591dcb62020-09-01 23:00:38 -0700879 let prefix = format!("cxxbridge04$std$vector${}{}$", namespace, elem);
David Tolnaya83247c2020-04-24 14:36:10 -0700880 let link_size = format!("{}size", prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700881 let link_get_unchecked = format!("{}get_unchecked", prefix);
David Tolnay591dcb62020-09-01 23:00:38 -0700882 let unique_ptr_prefix = format!("cxxbridge04$unique_ptr$std$vector${}{}$", namespace, elem);
David Tolnay3b40b6f2020-04-24 17:58:24 -0700883 let link_unique_ptr_null = format!("{}null", unique_ptr_prefix);
884 let link_unique_ptr_raw = format!("{}raw", unique_ptr_prefix);
885 let link_unique_ptr_get = format!("{}get", unique_ptr_prefix);
886 let link_unique_ptr_release = format!("{}release", unique_ptr_prefix);
887 let link_unique_ptr_drop = format!("{}drop", unique_ptr_prefix);
Myron Ahneba35cf2020-02-05 19:41:51 +0700888
David Tolnay0531f432020-10-03 23:50:28 -0700889 let begin_span =
890 explicit_impl.map_or_else(Span::call_site, |explicit| explicit.impl_token.span);
891 let end_span = explicit_impl.map_or_else(Span::call_site, |explicit| explicit.brace_token.span);
892 let unsafe_token = format_ident!("unsafe", span = begin_span);
893
894 quote_spanned! {end_span=>
895 #unsafe_token impl ::cxx::private::VectorElement for #elem {
David Tolnay3b40b6f2020-04-24 17:58:24 -0700896 const __NAME: &'static dyn ::std::fmt::Display = &#name;
David Tolnay17631102020-04-24 19:01:30 -0700897 fn __vector_size(v: &::cxx::CxxVector<Self>) -> usize {
David Tolnayfa2119c2020-04-24 14:42:58 -0700898 extern "C" {
899 #[link_name = #link_size]
David Tolnay17631102020-04-24 19:01:30 -0700900 fn __vector_size(_: &::cxx::CxxVector<#elem>) -> usize;
Myron Ahneba35cf2020-02-05 19:41:51 +0700901 }
David Tolnayfa2119c2020-04-24 14:42:58 -0700902 unsafe { __vector_size(v) }
Myron Ahneba35cf2020-02-05 19:41:51 +0700903 }
David Tolnay93637ca2020-09-24 15:58:20 -0400904 unsafe fn __get_unchecked(v: &::cxx::CxxVector<Self>, pos: usize) -> *const Self {
David Tolnaycc75ad22020-04-24 14:45:16 -0700905 extern "C" {
906 #[link_name = #link_get_unchecked]
David Tolnay17631102020-04-24 19:01:30 -0700907 fn __get_unchecked(_: &::cxx::CxxVector<#elem>, _: usize) -> *const #elem;
David Tolnaycc75ad22020-04-24 14:45:16 -0700908 }
David Tolnay93637ca2020-09-24 15:58:20 -0400909 __get_unchecked(v, pos)
David Tolnaycc75ad22020-04-24 14:45:16 -0700910 }
David Tolnay3b40b6f2020-04-24 17:58:24 -0700911 fn __unique_ptr_null() -> *mut ::std::ffi::c_void {
912 extern "C" {
913 #[link_name = #link_unique_ptr_null]
914 fn __unique_ptr_null(this: *mut *mut ::std::ffi::c_void);
915 }
916 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
917 unsafe { __unique_ptr_null(&mut repr) }
918 repr
919 }
920 unsafe fn __unique_ptr_raw(raw: *mut ::cxx::CxxVector<Self>) -> *mut ::std::ffi::c_void {
921 extern "C" {
922 #[link_name = #link_unique_ptr_raw]
923 fn __unique_ptr_raw(this: *mut *mut ::std::ffi::c_void, raw: *mut ::cxx::CxxVector<#elem>);
924 }
925 let mut repr = ::std::ptr::null_mut::<::std::ffi::c_void>();
926 __unique_ptr_raw(&mut repr, raw);
927 repr
928 }
929 unsafe fn __unique_ptr_get(repr: *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<Self> {
930 extern "C" {
931 #[link_name = #link_unique_ptr_get]
932 fn __unique_ptr_get(this: *const *mut ::std::ffi::c_void) -> *const ::cxx::CxxVector<#elem>;
933 }
934 __unique_ptr_get(&repr)
935 }
936 unsafe fn __unique_ptr_release(mut repr: *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<Self> {
937 extern "C" {
938 #[link_name = #link_unique_ptr_release]
939 fn __unique_ptr_release(this: *mut *mut ::std::ffi::c_void) -> *mut ::cxx::CxxVector<#elem>;
940 }
941 __unique_ptr_release(&mut repr)
942 }
943 unsafe fn __unique_ptr_drop(mut repr: *mut ::std::ffi::c_void) {
944 extern "C" {
945 #[link_name = #link_unique_ptr_drop]
946 fn __unique_ptr_drop(this: *mut *mut ::std::ffi::c_void);
947 }
948 __unique_ptr_drop(&mut repr);
949 }
Myron Ahneba35cf2020-02-05 19:41:51 +0700950 }
951 }
952}
953
David Tolnay7db73692019-10-20 14:51:12 -0400954fn expand_return_type(ret: &Option<Type>) -> TokenStream {
955 match ret {
956 Some(ret) => quote!(-> #ret),
957 None => TokenStream::new(),
958 }
959}
960
David Tolnay75dca2e2020-03-25 20:17:52 -0700961fn indirect_return(sig: &Signature, types: &Types) -> bool {
962 sig.ret
David Tolnay1e548172020-03-16 13:37:09 -0700963 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700964 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay7db73692019-10-20 14:51:12 -0400965}
966
967fn expand_extern_type(ty: &Type) -> TokenStream {
968 match ty {
David Tolnaya52602b2020-03-06 10:24:34 -0800969 Type::Ident(ident) if ident == RustString => quote!(::cxx::private::RustString),
David Tolnay7db73692019-10-20 14:51:12 -0400970 Type::RustBox(ty) | Type::UniquePtr(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +0700971 let inner = expand_extern_type(&ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -0400972 quote!(*mut #inner)
973 }
David Tolnaye6d50212020-04-25 15:38:23 -0700974 Type::RustVec(ty) => {
975 let elem = expand_extern_type(&ty.inner);
976 quote!(::cxx::private::RustVec<#elem>)
977 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700978 Type::Ref(ty) => {
979 let mutability = ty.mutability;
980 match &ty.inner {
981 Type::Ident(ident) if ident == RustString => {
982 quote!(&#mutability ::cxx::private::RustString)
983 }
984 Type::RustVec(ty) => {
985 let inner = expand_extern_type(&ty.inner);
986 quote!(&#mutability ::cxx::private::RustVec<#inner>)
987 }
988 _ => quote!(#ty),
Myron Ahneba35cf2020-02-05 19:41:51 +0700989 }
David Tolnayf1c7f322020-08-27 00:46:01 -0700990 }
David Tolnay7db73692019-10-20 14:51:12 -0400991 Type::Str(_) => quote!(::cxx::private::RustStr),
Adrian Taylorf5dd5522020-04-13 16:50:14 -0700992 Type::SliceRefU8(_) => quote!(::cxx::private::RustSliceU8),
David Tolnay7db73692019-10-20 14:51:12 -0400993 _ => quote!(#ty),
994 }
995}
996
997fn expand_extern_return_type(ret: &Option<Type>, types: &Types) -> TokenStream {
998 let ret = match ret {
999 Some(ret) if !types.needs_indirect_abi(ret) => ret,
1000 _ => return TokenStream::new(),
1001 };
1002 let ty = expand_extern_type(ret);
1003 quote!(-> #ty)
1004}