blob: 5413b6827e27874e5049dab823dd931d7e0e7e24 [file] [log] [blame]
David Tolnay0c033e32020-11-01 15:15:48 -08001use crate::gen::block::Block;
David Tolnayf7b81fb2020-11-01 22:39:04 -08002use crate::gen::nested::NamespaceEntries;
David Tolnay8810a542020-10-31 21:39:22 -07003use crate::gen::out::OutFile;
David Tolnay942cffd2020-11-03 18:27:10 -08004use crate::gen::{builtin, include, Opt};
David Tolnay7db73692019-10-20 14:51:12 -04005use crate::syntax::atom::Atom::{self, *};
David Tolnay891061b2020-04-19 22:42:33 -07006use crate::syntax::symbol::Symbol;
Adrian Taylorc8713432020-10-21 18:20:55 -07007use crate::syntax::{
David Tolnayb960ed22020-11-27 14:34:30 -08008 derive, mangle, Api, Enum, ExternFn, ExternType, Pair, ResolvableName, Signature, Struct,
9 Trait, Type, Types, Var,
Adrian Taylorc8713432020-10-21 18:20:55 -070010};
David Tolnay7db73692019-10-20 14:51:12 -040011use proc_macro2::Ident;
David Tolnay5439fa12020-11-03 18:45:01 -080012use std::collections::{HashMap, HashSet};
David Tolnay7db73692019-10-20 14:51:12 -040013
David Tolnayac7188c2020-11-01 16:58:16 -080014pub(super) fn gen(apis: &[Api], types: &Types, opt: &Opt, header: bool) -> Vec<u8> {
David Tolnaye1476af2020-11-01 13:47:25 -080015 let mut out_file = OutFile::new(header, opt, types);
David Tolnay7db73692019-10-20 14:51:12 -040016 let out = &mut out_file;
17
David Tolnaydfb82d72020-11-02 00:10:10 -080018 pick_includes_and_builtins(out, apis);
David Tolnay4aae7c02020-10-28 12:35:42 -070019 out.include.extend(&opt.include);
David Tolnay7db73692019-10-20 14:51:12 -040020
David Tolnay7b0e5102020-11-01 23:22:12 -080021 write_forward_declarations(out, apis);
David Tolnaye1109d92020-11-01 20:51:56 -080022 write_data_structures(out, apis);
23 write_functions(out, apis);
David Tolnay169bb472020-11-01 21:04:24 -080024 write_generic_instantiations(out);
David Tolnay7db73692019-10-20 14:51:12 -040025
David Tolnay3374d8d2020-10-31 22:18:45 -070026 builtin::write(out);
David Tolnay2f3e90b2020-10-31 22:16:51 -070027 include::write(out);
Adrian Taylorc8713432020-10-21 18:20:55 -070028
David Tolnayac7188c2020-11-01 16:58:16 -080029 out_file.content()
Adrian Taylorc8713432020-10-21 18:20:55 -070030}
31
David Tolnay7b0e5102020-11-01 23:22:12 -080032fn write_forward_declarations(out: &mut OutFile, apis: &[Api]) {
33 let needs_forward_declaration = |api: &&Api| match api {
David Tolnay4bfca112020-11-01 23:59:11 -080034 Api::Struct(_) | Api::CxxType(_) | Api::RustType(_) => true,
David Tolnay17a934c2020-11-02 00:40:04 -080035 Api::Enum(enm) => !out.types.cxx.contains(&enm.name.rust),
David Tolnay7b0e5102020-11-01 23:22:12 -080036 _ => false,
37 };
Adrian Taylorc8713432020-10-21 18:20:55 -070038
David Tolnay7b0e5102020-11-01 23:22:12 -080039 let apis_by_namespace =
40 NamespaceEntries::new(apis.iter().filter(needs_forward_declaration).collect());
41
David Tolnayd920be52020-11-01 23:34:30 -080042 write(out, &apis_by_namespace, 0);
David Tolnay7b0e5102020-11-01 23:22:12 -080043
David Tolnayd920be52020-11-01 23:34:30 -080044 fn write(out: &mut OutFile, ns_entries: &NamespaceEntries, indent: usize) {
David Tolnay7b0e5102020-11-01 23:22:12 -080045 let apis = ns_entries.direct_content();
46
47 for api in apis {
David Tolnayd920be52020-11-01 23:34:30 -080048 write!(out, "{:1$}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080049 match api {
David Tolnay17a934c2020-11-02 00:40:04 -080050 Api::Struct(strct) => write_struct_decl(out, &strct.name.cxx),
David Tolnay0e3ee8e2020-11-01 23:46:52 -080051 Api::Enum(enm) => write_enum_decl(out, enm),
David Tolnay17a934c2020-11-02 00:40:04 -080052 Api::CxxType(ety) => write_struct_using(out, &ety.name),
53 Api::RustType(ety) => write_struct_decl(out, &ety.name.cxx),
David Tolnay7b0e5102020-11-01 23:22:12 -080054 _ => unreachable!(),
55 }
David Tolnay7db73692019-10-20 14:51:12 -040056 }
David Tolnay7db73692019-10-20 14:51:12 -040057
David Tolnay7b0e5102020-11-01 23:22:12 -080058 for (namespace, nested_ns_entries) in ns_entries.nested_content() {
David Tolnayd920be52020-11-01 23:34:30 -080059 writeln!(out, "{:2$}namespace {} {{", "", namespace, indent);
60 write(out, nested_ns_entries, indent + 2);
61 writeln!(out, "{:1$}}}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080062 }
Adrian Taylorf9213622020-10-31 22:25:42 -070063 }
64}
65
David Tolnaye1109d92020-11-01 20:51:56 -080066fn write_data_structures<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayf94bef12020-04-17 14:46:42 -070067 let mut methods_for_type = HashMap::new();
David Tolnay630af882020-10-31 22:03:47 -070068 for api in apis {
David Tolnay102c7ea2020-11-08 18:58:09 -080069 if let Api::CxxFunction(efn) | Api::RustFunction(efn) = api {
David Tolnayf94bef12020-04-17 14:46:42 -070070 if let Some(receiver) = &efn.sig.receiver {
71 methods_for_type
Adrian Taylorc8713432020-10-21 18:20:55 -070072 .entry(&receiver.ty.rust)
David Tolnayf94bef12020-04-17 14:46:42 -070073 .or_insert_with(Vec::new)
74 .push(efn);
75 }
76 }
77 }
Joel Galenson968738f2020-04-15 14:19:33 -070078
David Tolnay5439fa12020-11-03 18:45:01 -080079 let mut structs_written = HashSet::new();
80 let mut toposorted_structs = out.types.toposorted_structs.iter();
David Tolnay9622b462020-11-02 21:34:42 -080081 for api in apis {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -070082 match api {
David Tolnay5439fa12020-11-03 18:45:01 -080083 Api::Struct(strct) if !structs_written.contains(&strct.name.rust) => {
84 for next in &mut toposorted_structs {
85 if !out.types.cxx.contains(&strct.name.rust) {
86 out.next_section();
David Tolnay102c7ea2020-11-08 18:58:09 -080087 let methods = methods_for_type
88 .get(&strct.name.rust)
89 .map(Vec::as_slice)
90 .unwrap_or_default();
91 write_struct(out, next, methods);
David Tolnay5439fa12020-11-03 18:45:01 -080092 }
93 structs_written.insert(&next.name.rust);
94 if next.name.rust == strct.name.rust {
95 break;
96 }
97 }
98 }
Joel Galensonc03402a2020-04-23 17:31:09 -070099 Api::Enum(enm) => {
100 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800101 if out.types.cxx.contains(&enm.name.rust) {
Joel Galenson905eb2e2020-05-04 14:58:14 -0700102 check_enum(out, enm);
103 } else {
104 write_enum(out, enm);
105 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700106 }
David Tolnayc1fe0052020-04-17 15:15:06 -0700107 Api::RustType(ety) => {
David Tolnay17a934c2020-11-02 00:40:04 -0800108 if let Some(methods) = methods_for_type.get(&ety.name.rust) {
David Tolnay46a54e72020-04-17 14:48:21 -0700109 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700110 write_struct_with_methods(out, ety, methods);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700111 }
David Tolnayc1fe0052020-04-17 15:15:06 -0700112 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700113 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400114 }
115 }
116
David Tolnayfabca772020-10-03 21:25:41 -0700117 out.next_section();
118 for api in apis {
119 if let Api::TypeAlias(ety) = api {
David Tolnay17a934c2020-11-02 00:40:04 -0800120 if out.types.required_trivial.contains_key(&ety.name.rust) {
121 check_trivial_extern_type(out, &ety.name)
David Tolnayfabca772020-10-03 21:25:41 -0700122 }
123 }
124 }
David Tolnay1b0339c2020-11-01 20:37:50 -0800125}
126
David Tolnaye1109d92020-11-01 20:51:56 -0800127fn write_functions<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayce5a91f2020-10-31 22:42:08 -0700128 if !out.header {
David Tolnay7db73692019-10-20 14:51:12 -0400129 for api in apis {
David Tolnay04770742020-11-01 13:50:50 -0800130 match api {
David Tolnayb960ed22020-11-27 14:34:30 -0800131 Api::Struct(strct) => write_struct_operator_decls(out, strct),
David Tolnay04770742020-11-01 13:50:50 -0800132 Api::CxxFunction(efn) => write_cxx_function_shim(out, efn),
133 Api::RustFunction(efn) => write_rust_function_decl(out, efn),
134 _ => {}
135 }
David Tolnay7db73692019-10-20 14:51:12 -0400136 }
David Tolnay7da38202020-11-27 17:36:16 -0800137
138 write_std_specializations(out, apis);
David Tolnay7db73692019-10-20 14:51:12 -0400139 }
140
141 for api in apis {
David Tolnayb960ed22020-11-27 14:34:30 -0800142 match api {
143 Api::Struct(strct) => write_struct_operators(out, strct),
144 Api::RustFunction(efn) => {
145 out.next_section();
146 write_rust_function_shim(out, efn);
147 }
148 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400149 }
150 }
David Tolnay7db73692019-10-20 14:51:12 -0400151}
152
David Tolnay7da38202020-11-27 17:36:16 -0800153fn write_std_specializations(out: &mut OutFile, apis: &[Api]) {
154 out.set_namespace(Default::default());
155 out.begin_block(Block::Namespace("std"));
156
157 for api in apis {
158 if let Api::Struct(strct) = api {
159 if derive::contains(&strct.derives, Trait::Hash) {
160 out.next_section();
161 let qualified = strct.name.to_fully_qualified();
162 writeln!(out, "template <> struct hash<{}> {{", qualified);
163 writeln!(
164 out,
165 " size_t operator()(const {} &self) const noexcept {{",
166 qualified,
167 );
168 let link_name = mangle::operator(&strct.name, "hash");
169 write!(out, " return ::");
170 for name in &strct.name.namespace {
171 write!(out, "{}::", name);
172 }
173 writeln!(out, "{}(self);", link_name);
174 writeln!(out, " }}");
175 writeln!(out, "}};");
176 }
177 }
178 }
179
180 out.end_block(Block::Namespace("std"));
181}
182
David Tolnaydfb82d72020-11-02 00:10:10 -0800183fn pick_includes_and_builtins(out: &mut OutFile, apis: &[Api]) {
184 for api in apis {
185 if let Api::Include(include) = api {
186 out.include.insert(include);
187 }
188 }
189
David Tolnaya7c2ea12020-10-30 21:32:53 -0700190 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -0400191 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700192 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnay89e386d2020-10-03 19:02:19 -0700193 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
194 | Some(I64) => out.include.cstdint = true,
195 Some(Usize) => out.include.cstddef = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800196 Some(Isize) => out.builtin.rust_isize = true,
David Tolnay89e386d2020-10-03 19:02:19 -0700197 Some(CxxString) => out.include.string = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800198 Some(RustString) => out.builtin.rust_string = true,
David Tolnayb3873dc2020-11-25 19:47:49 -0800199 Some(Bool) | Some(Char) | Some(F32) | Some(F64) | None => {}
David Tolnay89e386d2020-10-03 19:02:19 -0700200 },
David Tolnaydcfa8e92020-11-02 09:50:06 -0800201 Type::RustBox(_) => out.builtin.rust_box = true,
202 Type::RustVec(_) => out.builtin.rust_vec = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700203 Type::UniquePtr(_) => out.include.memory = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800204 Type::Str(_) => out.builtin.rust_str = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700205 Type::CxxVector(_) => out.include.vector = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800206 Type::Fn(_) => out.builtin.rust_fn = true,
David Tolnay5515a9e2020-11-25 19:07:54 -0800207 Type::SliceRef(_) => out.builtin.rust_slice = true,
David Tolnaye8b1bb42020-11-24 20:37:37 -0800208 Type::Array(_) => out.include.array = true,
David Tolnay11bd7ff2020-11-01 19:44:58 -0800209 Type::Ref(_) | Type::Void(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -0400210 }
211 }
David Tolnayec66d112020-10-31 21:00:26 -0700212}
David Tolnayf51447e2020-03-06 14:14:27 -0800213
David Tolnay102c7ea2020-11-08 18:58:09 -0800214fn write_struct<'a>(out: &mut OutFile<'a>, strct: &'a Struct, methods: &[&ExternFn]) {
David Tolnayb960ed22020-11-27 14:34:30 -0800215 let operator_eq = derive::contains(&strct.derives, Trait::PartialEq);
David Tolnay84389352020-11-27 17:12:10 -0800216 let operator_ord = derive::contains(&strct.derives, Trait::PartialOrd);
David Tolnayb960ed22020-11-27 14:34:30 -0800217
David Tolnay17a934c2020-11-02 00:40:04 -0800218 out.set_namespace(&strct.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800219 let guard = format!("CXXBRIDGE1_STRUCT_{}", strct.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700220 writeln!(out, "#ifndef {}", guard);
221 writeln!(out, "#define {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400222 for line in strct.doc.to_string().lines() {
223 writeln!(out, "//{}", line);
224 }
David Tolnay17a934c2020-11-02 00:40:04 -0800225 writeln!(out, "struct {} final {{", strct.name.cxx);
David Tolnay84389352020-11-27 17:12:10 -0800226
David Tolnay7db73692019-10-20 14:51:12 -0400227 for field in &strct.fields {
228 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700229 write_type_space(out, &field.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400230 writeln!(out, "{};", field.ident);
231 }
David Tolnay84389352020-11-27 17:12:10 -0800232
233 if !methods.is_empty() || operator_eq || operator_ord {
David Tolnay102c7ea2020-11-08 18:58:09 -0800234 writeln!(out);
235 }
David Tolnay84389352020-11-27 17:12:10 -0800236
David Tolnay102c7ea2020-11-08 18:58:09 -0800237 for method in methods {
238 write!(out, " ");
239 let sig = &method.sig;
240 let local_name = method.name.cxx.to_string();
241 write_rust_function_shim_decl(out, &local_name, sig, false);
242 writeln!(out, ";");
243 }
David Tolnay84389352020-11-27 17:12:10 -0800244
David Tolnayb960ed22020-11-27 14:34:30 -0800245 if operator_eq {
246 writeln!(
247 out,
248 " bool operator==(const {} &) const noexcept;",
249 strct.name.cxx,
250 );
251 writeln!(
252 out,
253 " bool operator!=(const {} &) const noexcept;",
254 strct.name.cxx,
255 );
256 }
David Tolnay84389352020-11-27 17:12:10 -0800257
258 if operator_ord {
259 writeln!(
260 out,
261 " bool operator<(const {} &) const noexcept;",
262 strct.name.cxx,
263 );
264 writeln!(
265 out,
266 " bool operator<=(const {} &) const noexcept;",
267 strct.name.cxx,
268 );
269 writeln!(
270 out,
271 " bool operator>(const {} &) const noexcept;",
272 strct.name.cxx,
273 );
274 writeln!(
275 out,
276 " bool operator>=(const {} &) const noexcept;",
277 strct.name.cxx,
278 );
279 }
280
David Tolnay7db73692019-10-20 14:51:12 -0400281 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700282 writeln!(out, "#endif // {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400283}
284
285fn write_struct_decl(out: &mut OutFile, ident: &Ident) {
286 writeln!(out, "struct {};", ident);
287}
288
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800289fn write_enum_decl(out: &mut OutFile, enm: &Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800290 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800291 write_atom(out, enm.repr);
292 writeln!(out, ";");
293}
294
David Tolnay8faec772020-11-02 00:18:19 -0800295fn write_struct_using(out: &mut OutFile, ident: &Pair) {
296 writeln!(out, "using {} = {};", ident.cxx, ident.to_fully_qualified());
David Tolnay8861bee2020-01-20 18:39:24 -0800297}
298
David Tolnay0b9b9f82020-11-01 20:41:00 -0800299fn write_struct_with_methods<'a>(
300 out: &mut OutFile<'a>,
301 ety: &'a ExternType,
302 methods: &[&ExternFn],
303) {
David Tolnay17a934c2020-11-02 00:40:04 -0800304 out.set_namespace(&ety.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800305 let guard = format!("CXXBRIDGE1_STRUCT_{}", ety.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700306 writeln!(out, "#ifndef {}", guard);
307 writeln!(out, "#define {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700308 for line in ety.doc.to_string().lines() {
309 writeln!(out, "//{}", line);
310 }
David Tolnay365fc7c2020-11-25 16:08:13 -0800311 out.builtin.opaque = true;
David Tolnay7c06b862020-11-25 16:59:09 -0800312 writeln!(
313 out,
314 "struct {} final : public ::rust::Opaque {{",
315 ety.name.cxx,
316 );
Joel Galenson968738f2020-04-15 14:19:33 -0700317 for method in methods {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700318 write!(out, " ");
319 let sig = &method.sig;
David Tolnay17a934c2020-11-02 00:40:04 -0800320 let local_name = method.name.cxx.to_string();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700321 write_rust_function_shim_decl(out, &local_name, sig, false);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700322 writeln!(out, ";");
323 }
324 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700325 writeln!(out, "#endif // {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700326}
327
David Tolnay0b9b9f82020-11-01 20:41:00 -0800328fn write_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800329 out.set_namespace(&enm.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800330 let guard = format!("CXXBRIDGE1_ENUM_{}", enm.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700331 writeln!(out, "#ifndef {}", guard);
332 writeln!(out, "#define {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700333 for line in enm.doc.to_string().lines() {
334 writeln!(out, "//{}", line);
335 }
David Tolnay17a934c2020-11-02 00:40:04 -0800336 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700337 write_atom(out, enm.repr);
338 writeln!(out, " {{");
Joel Galensonc03402a2020-04-23 17:31:09 -0700339 for variant in &enm.variants {
Joel Galenson88547732020-05-05 08:23:42 -0700340 writeln!(out, " {} = {},", variant.ident, variant.discriminant);
Joel Galensonc03402a2020-04-23 17:31:09 -0700341 }
342 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700343 writeln!(out, "#endif // {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700344}
345
David Tolnay0b9b9f82020-11-01 20:41:00 -0800346fn check_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800347 out.set_namespace(&enm.name.namespace);
David Tolnay06711bc2020-11-19 19:25:14 -0800348 out.include.type_traits = true;
349 writeln!(
350 out,
351 "static_assert(::std::is_enum<{}>::value, \"expected enum\");",
352 enm.name.cxx,
353 );
David Tolnay17a934c2020-11-02 00:40:04 -0800354 write!(out, "static_assert(sizeof({}) == sizeof(", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700355 write_atom(out, enm.repr);
356 writeln!(out, "), \"incorrect size\");");
Joel Galenson0f654ff2020-05-04 20:04:21 -0700357 for variant in &enm.variants {
David Tolnayf6a89f22020-05-10 23:39:27 -0700358 write!(out, "static_assert(static_cast<");
359 write_atom(out, enm.repr);
Joel Galenson0f654ff2020-05-04 20:04:21 -0700360 writeln!(
361 out,
David Tolnayf6a89f22020-05-10 23:39:27 -0700362 ">({}::{}) == {}, \"disagrees with the value in #[cxx::bridge]\");",
David Tolnay17a934c2020-11-02 00:40:04 -0800363 enm.name.cxx, variant.ident, variant.discriminant,
Joel Galenson0f654ff2020-05-04 20:04:21 -0700364 );
Joel Galenson0f654ff2020-05-04 20:04:21 -0700365 }
Joel Galenson905eb2e2020-05-04 14:58:14 -0700366}
367
David Tolnay8faec772020-11-02 00:18:19 -0800368fn check_trivial_extern_type(out: &mut OutFile, id: &Pair) {
David Tolnay174bd952020-11-02 09:23:12 -0800369 // NOTE: The following static assertion is just nice-to-have and not
David Tolnayfd0034e2020-10-04 13:15:34 -0700370 // necessary for soundness. That's because triviality is always declared by
371 // the user in the form of an unsafe impl of cxx::ExternType:
372 //
373 // unsafe impl ExternType for MyType {
374 // type Id = cxx::type_id!("...");
375 // type Kind = cxx::kind::Trivial;
376 // }
377 //
378 // Since the user went on the record with their unsafe impl to unsafely
379 // claim they KNOW that the type is trivial, it's fine for that to be on
David Tolnay174bd952020-11-02 09:23:12 -0800380 // them if that were wrong. However, in practice correctly reasoning about
381 // the relocatability of C++ types is challenging, particularly if the type
382 // definition were to change over time, so for now we add this check.
David Tolnayfd0034e2020-10-04 13:15:34 -0700383 //
David Tolnay174bd952020-11-02 09:23:12 -0800384 // There may be legitimate reasons to opt out of this assertion for support
385 // of types that the programmer knows are soundly Rust-movable despite not
386 // being recognized as such by the C++ type system due to a move constructor
387 // or destructor. To opt out of the relocatability check, they need to do
388 // one of the following things in any header used by `include!` in their
389 // bridge.
390 //
391 // --- if they define the type:
392 // struct MyType {
393 // ...
394 // + using IsRelocatable = std::true_type;
395 // };
396 //
397 // --- otherwise:
398 // + template <>
399 // + struct rust::IsRelocatable<MyType> : std::true_type {};
400 //
David Tolnayfd0034e2020-10-04 13:15:34 -0700401
David Tolnay174bd952020-11-02 09:23:12 -0800402 let id = id.to_fully_qualified();
403 out.builtin.relocatable = true;
David Tolnay7426cc12020-10-03 19:04:04 -0700404 writeln!(out, "static_assert(");
David Tolnay174bd952020-11-02 09:23:12 -0800405 writeln!(out, " ::rust::IsRelocatable<{}>::value,", id);
David Tolnay7426cc12020-10-03 19:04:04 -0700406 writeln!(
407 out,
David Tolnay174bd952020-11-02 09:23:12 -0800408 " \"type {} marked as Trivial in Rust is not trivially move constructible and trivially destructible in C++\");",
David Tolnay7426cc12020-10-03 19:04:04 -0700409 id,
410 );
Adrian Taylor405e8742020-09-25 14:01:25 -0700411}
412
David Tolnayb960ed22020-11-27 14:34:30 -0800413fn write_struct_operator_decls<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
414 out.set_namespace(&strct.name.namespace);
415 out.begin_block(Block::ExternC);
416
417 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800418 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800419 writeln!(
420 out,
421 "bool {}(const {1} &, const {1} &) noexcept;",
422 link_name, strct.name.cxx,
423 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800424
425 if !derive::contains(&strct.derives, Trait::Eq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800426 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800427 writeln!(
428 out,
429 "bool {}(const {1} &, const {1} &) noexcept;",
430 link_name, strct.name.cxx,
431 );
432 }
David Tolnayb960ed22020-11-27 14:34:30 -0800433 }
434
David Tolnay84389352020-11-27 17:12:10 -0800435 if derive::contains(&strct.derives, Trait::PartialOrd) {
David Tolnaya05f9402020-11-27 17:44:05 -0800436 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800437 writeln!(
438 out,
439 "bool {}(const {1} &, const {1} &) noexcept;",
440 link_name, strct.name.cxx,
441 );
442
David Tolnaya05f9402020-11-27 17:44:05 -0800443 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800444 writeln!(
445 out,
446 "bool {}(const {1} &, const {1} &) noexcept;",
447 link_name, strct.name.cxx,
448 );
449
450 if !derive::contains(&strct.derives, Trait::Ord) {
David Tolnaya05f9402020-11-27 17:44:05 -0800451 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800452 writeln!(
453 out,
454 "bool {}(const {1} &, const {1} &) noexcept;",
455 link_name, strct.name.cxx,
456 );
457
David Tolnaya05f9402020-11-27 17:44:05 -0800458 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800459 writeln!(
460 out,
461 "bool {}(const {1} &, const {1} &) noexcept;",
462 link_name, strct.name.cxx,
463 );
464 }
465 }
466
David Tolnay7da38202020-11-27 17:36:16 -0800467 if derive::contains(&strct.derives, Trait::Hash) {
468 let link_name = mangle::operator(&strct.name, "hash");
469 writeln!(
470 out,
471 "size_t {}(const {} &) noexcept;",
472 link_name, strct.name.cxx,
473 );
474 }
475
David Tolnayb960ed22020-11-27 14:34:30 -0800476 out.end_block(Block::ExternC);
477}
478
479fn write_struct_operators<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
480 if out.header {
481 return;
482 }
483
484 out.set_namespace(&strct.name.namespace);
485
486 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnayb960ed22020-11-27 14:34:30 -0800487 out.next_section();
488 writeln!(
489 out,
490 "bool {0}::operator==(const {0} &rhs) const noexcept {{",
491 strct.name.cxx,
492 );
David Tolnaya05f9402020-11-27 17:44:05 -0800493 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800494 writeln!(out, " return {}(*this, rhs);", link_name);
495 writeln!(out, "}}");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800496
David Tolnayb960ed22020-11-27 14:34:30 -0800497 out.next_section();
498 writeln!(
499 out,
500 "bool {0}::operator!=(const {0} &rhs) const noexcept {{",
501 strct.name.cxx,
502 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800503 if derive::contains(&strct.derives, Trait::Eq) {
504 writeln!(out, " return !(*this == rhs);");
505 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800506 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800507 writeln!(out, " return {}(*this, rhs);", link_name);
508 }
David Tolnayb960ed22020-11-27 14:34:30 -0800509 writeln!(out, "}}");
510 }
David Tolnay84389352020-11-27 17:12:10 -0800511
512 if derive::contains(&strct.derives, Trait::PartialOrd) {
513 out.next_section();
514 writeln!(
515 out,
516 "bool {0}::operator<(const {0} &rhs) const noexcept {{",
517 strct.name.cxx,
518 );
David Tolnaya05f9402020-11-27 17:44:05 -0800519 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800520 writeln!(out, " return {}(*this, rhs);", link_name);
521 writeln!(out, "}}");
522
523 out.next_section();
524 writeln!(
525 out,
526 "bool {0}::operator<=(const {0} &rhs) const noexcept {{",
527 strct.name.cxx,
528 );
David Tolnaya05f9402020-11-27 17:44:05 -0800529 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800530 writeln!(out, " return {}(*this, rhs);", link_name);
531 writeln!(out, "}}");
532
533 out.next_section();
534 writeln!(
535 out,
536 "bool {0}::operator>(const {0} &rhs) const noexcept {{",
537 strct.name.cxx,
538 );
539 if derive::contains(&strct.derives, Trait::Ord) {
540 writeln!(out, " return !(*this <= rhs);");
541 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800542 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800543 writeln!(out, " return {}(*this, rhs);", link_name);
544 }
545 writeln!(out, "}}");
546
547 out.next_section();
548 writeln!(
549 out,
550 "bool {0}::operator>=(const {0} &rhs) const noexcept {{",
551 strct.name.cxx,
552 );
553 if derive::contains(&strct.derives, Trait::Ord) {
554 writeln!(out, " return !(*this < rhs);");
555 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800556 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800557 writeln!(out, " return {}(*this, rhs);", link_name);
558 }
559 writeln!(out, "}}");
560 }
David Tolnayb960ed22020-11-27 14:34:30 -0800561}
562
David Tolnay0b9b9f82020-11-01 20:41:00 -0800563fn write_cxx_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay04770742020-11-01 13:50:50 -0800564 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800565 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800566 out.begin_block(Block::ExternC);
David Tolnaye1476af2020-11-01 13:47:25 -0800567 if let Some(annotation) = &out.opt.cxx_impl_annotations {
David Tolnaycc1ae762020-10-31 15:53:50 -0700568 write!(out, "{} ", annotation);
Adrian Taylor21f0ff02020-07-21 16:21:48 -0700569 }
David Tolnayebef4a22020-03-17 15:33:47 -0700570 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700571 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700572 write!(out, "::rust::repr::PtrLen ");
David Tolnayebef4a22020-03-17 15:33:47 -0700573 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700574 write_extern_return_type_space(out, &efn.ret);
David Tolnayebef4a22020-03-17 15:33:47 -0700575 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700576 let mangled = mangle::extern_fn(efn, out.types);
David Tolnay3caa50a2020-04-19 21:25:34 -0700577 write!(out, "{}(", mangled);
David Tolnaye439c772020-04-20 00:23:55 -0700578 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800579 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700580 write!(out, "const ");
581 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700582 write!(
583 out,
584 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800585 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700586 );
Joel Galenson3d4f6122020-04-07 15:54:05 -0700587 }
David Tolnay7db73692019-10-20 14:51:12 -0400588 for (i, arg) in efn.args.iter().enumerate() {
Joel Galenson3d4f6122020-04-07 15:54:05 -0700589 if i > 0 || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400590 write!(out, ", ");
591 }
David Tolnaya46a2372020-03-06 10:03:48 -0800592 if arg.ty == RustString {
593 write!(out, "const ");
David Tolnay313b10e2020-04-25 16:30:51 -0700594 } else if let Type::RustVec(_) = arg.ty {
595 write!(out, "const ");
David Tolnaya46a2372020-03-06 10:03:48 -0800596 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700597 write_extern_arg(out, arg);
David Tolnay7db73692019-10-20 14:51:12 -0400598 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700599 let indirect_return = indirect_return(efn, out.types);
David Tolnay7db73692019-10-20 14:51:12 -0400600 if indirect_return {
myronahne3b78ea2020-05-23 01:08:13 +0700601 if !efn.args.is_empty() || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400602 write!(out, ", ");
603 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700604 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400605 write!(out, "*return$");
606 }
607 writeln!(out, ") noexcept {{");
608 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700609 write_return_type(out, &efn.ret);
Joel Galenson3d4f6122020-04-07 15:54:05 -0700610 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800611 None => write!(out, "(*{}$)(", efn.name.rust),
Adrian Taylorc8713432020-10-21 18:20:55 -0700612 Some(receiver) => write!(
613 out,
614 "({}::*{}$)(",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700615 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800616 efn.name.rust,
Adrian Taylorc8713432020-10-21 18:20:55 -0700617 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700618 }
David Tolnay7db73692019-10-20 14:51:12 -0400619 for (i, arg) in efn.args.iter().enumerate() {
620 if i > 0 {
621 write!(out, ", ");
622 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700623 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400624 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700625 write!(out, ")");
David Tolnay4e7123f2020-04-19 21:11:37 -0700626 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800627 if !receiver.mutable {
David Tolnay4e7123f2020-04-19 21:11:37 -0700628 write!(out, " const");
629 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700630 }
631 write!(out, " = ");
632 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800633 None => write!(out, "{}", efn.name.to_fully_qualified()),
Adrian Taylorc8713432020-10-21 18:20:55 -0700634 Some(receiver) => write!(
635 out,
636 "&{}::{}",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700637 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800638 efn.name.cxx,
Adrian Taylorc8713432020-10-21 18:20:55 -0700639 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700640 }
641 writeln!(out, ";");
David Tolnay7db73692019-10-20 14:51:12 -0400642 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700643 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700644 out.builtin.ptr_len = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700645 out.builtin.trycatch = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700646 writeln!(out, "::rust::repr::PtrLen throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700647 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700648 writeln!(out, " [&] {{");
649 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700650 }
David Tolnay7db73692019-10-20 14:51:12 -0400651 if indirect_return {
David Tolnay0ecd05a2020-07-29 16:32:03 -0700652 out.include.new = true;
David Tolnay7db73692019-10-20 14:51:12 -0400653 write!(out, "new (return$) ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700654 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400655 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700656 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400657 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700658 }
659 match &efn.ret {
660 Some(Type::Ref(_)) => write!(out, "&"),
David Tolnay74d6d512020-10-31 22:22:03 -0700661 Some(Type::Str(_)) if !indirect_return => {
662 out.builtin.rust_str_repr = true;
663 write!(out, "::rust::impl<::rust::Str>::repr(");
664 }
David Tolnay73b72642020-11-25 17:44:05 -0800665 Some(ty @ Type::SliceRef(_)) if !indirect_return => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700666 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800667 write!(out, "::rust::impl<");
668 write_type(out, ty);
669 write!(out, ">::repr(");
David Tolnayeb952ba2020-04-14 15:02:24 -0700670 }
David Tolnay99642622020-03-25 13:07:35 -0700671 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400672 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700673 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800674 None => write!(out, "{}$(", efn.name.rust),
675 Some(_) => write!(out, "(self.*{}$)(", efn.name.rust),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700676 }
David Tolnay7db73692019-10-20 14:51:12 -0400677 for (i, arg) in efn.args.iter().enumerate() {
678 if i > 0 {
679 write!(out, ", ");
680 }
681 if let Type::RustBox(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700682 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400683 write!(out, "::from_raw({})", arg.ident);
684 } else if let Type::UniquePtr(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700685 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400686 write!(out, "({})", arg.ident);
David Tolnay0356d332020-10-31 19:46:41 -0700687 } else if let Type::Str(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700688 out.builtin.rust_str_new_unchecked = true;
David Tolnay0356d332020-10-31 19:46:41 -0700689 write!(
690 out,
691 "::rust::impl<::rust::Str>::new_unchecked({})",
692 arg.ident,
693 );
David Tolnaya46a2372020-03-06 10:03:48 -0800694 } else if arg.ty == RustString {
David Tolnay74d6d512020-10-31 22:22:03 -0700695 out.builtin.unsafe_bitcopy = true;
David Tolnaycc3767f2020-03-06 10:41:51 -0800696 write!(
697 out,
698 "::rust::String(::rust::unsafe_bitcopy, *{})",
699 arg.ident,
700 );
David Tolnay313b10e2020-04-25 16:30:51 -0700701 } else if let Type::RustVec(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700702 out.builtin.unsafe_bitcopy = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700703 write_type(out, &arg.ty);
David Tolnay313b10e2020-04-25 16:30:51 -0700704 write!(out, "(::rust::unsafe_bitcopy, *{})", arg.ident);
David Tolnay73b72642020-11-25 17:44:05 -0800705 } else if let Type::SliceRef(slice) = &arg.ty {
David Tolnayc5629f02020-11-23 18:32:46 -0800706 write_type(out, &arg.ty);
707 write!(out, "(static_cast<");
708 if slice.mutability.is_none() {
709 write!(out, "const ");
710 }
David Tolnay5515a9e2020-11-25 19:07:54 -0800711 write_type_space(out, &slice.inner);
712 write!(out, "*>({0}.ptr), {0}.len)", arg.ident);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700713 } else if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700714 out.include.utility = true;
David Tolnay7e219b82020-03-01 13:14:51 -0800715 write!(out, "::std::move(*{})", arg.ident);
David Tolnay7db73692019-10-20 14:51:12 -0400716 } else {
717 write!(out, "{}", arg.ident);
718 }
719 }
720 write!(out, ")");
721 match &efn.ret {
722 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
723 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800724 Some(Type::Str(_)) | Some(Type::SliceRef(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400725 _ => {}
726 }
727 if indirect_return {
728 write!(out, ")");
729 }
730 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700731 if efn.throws {
732 out.include.cstring = true;
David Tolnay1f010c62020-11-01 20:27:46 -0800733 out.builtin.exception = true;
David Tolnay5d121442020-03-17 22:14:40 -0700734 writeln!(out, " throw$.ptr = nullptr;");
735 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700736 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700737 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700738 writeln!(
739 out,
David Tolnayc5629f02020-11-23 18:32:46 -0800740 " throw$.ptr = const_cast<char *>(::cxxbridge1$exception(catch$, throw$.len));",
David Tolnayebef4a22020-03-17 15:33:47 -0700741 );
David Tolnay5d121442020-03-17 22:14:40 -0700742 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700743 writeln!(out, " return throw$;");
744 }
David Tolnay7db73692019-10-20 14:51:12 -0400745 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700746 for arg in &efn.args {
747 if let Type::Fn(f) = &arg.ty {
748 let var = &arg.ident;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700749 write_function_pointer_trampoline(out, efn, var, f);
David Tolnay75dca2e2020-03-25 20:17:52 -0700750 }
751 }
David Tolnayca563ee2020-11-01 20:12:27 -0800752 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700753}
754
755fn write_function_pointer_trampoline(
756 out: &mut OutFile,
757 efn: &ExternFn,
758 var: &Ident,
759 f: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700760) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700761 let r_trampoline = mangle::r_trampoline(efn, var, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700762 let indirect_call = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700763 write_rust_function_decl_impl(out, &r_trampoline, f, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700764
765 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700766 let c_trampoline = mangle::c_trampoline(efn, var, out.types).to_string();
767 write_rust_function_shim_impl(out, &c_trampoline, f, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400768}
769
David Tolnay0b9b9f82020-11-01 20:41:00 -0800770fn write_rust_function_decl<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800771 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800772 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700773 let link_name = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700774 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700775 write_rust_function_decl_impl(out, &link_name, efn, indirect_call);
David Tolnayca563ee2020-11-01 20:12:27 -0800776 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700777}
778
779fn write_rust_function_decl_impl(
780 out: &mut OutFile,
David Tolnay891061b2020-04-19 22:42:33 -0700781 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700782 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700783 indirect_call: bool,
784) {
David Tolnay04770742020-11-01 13:50:50 -0800785 out.next_section();
David Tolnay75dca2e2020-03-25 20:17:52 -0700786 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700787 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700788 write!(out, "::rust::repr::PtrLen ");
David Tolnay1e548172020-03-16 13:37:09 -0700789 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700790 write_extern_return_type_space(out, &sig.ret);
David Tolnay1e548172020-03-16 13:37:09 -0700791 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700792 write!(out, "{}(", link_name);
793 let mut needs_comma = false;
David Tolnaye439c772020-04-20 00:23:55 -0700794 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800795 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700796 write!(out, "const ");
797 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700798 write!(
799 out,
800 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800801 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700802 );
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700803 needs_comma = true;
804 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700805 for arg in &sig.args {
806 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400807 write!(out, ", ");
808 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700809 write_extern_arg(out, arg);
David Tolnay75dca2e2020-03-25 20:17:52 -0700810 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400811 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700812 if indirect_return(sig, out.types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700813 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400814 write!(out, ", ");
815 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700816 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400817 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700818 needs_comma = true;
819 }
820 if indirect_call {
821 if needs_comma {
822 write!(out, ", ");
823 }
824 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400825 }
826 writeln!(out, ") noexcept;");
827}
828
David Tolnay0b9b9f82020-11-01 20:41:00 -0800829fn write_rust_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800830 out.set_namespace(&efn.name.namespace);
David Tolnay7db73692019-10-20 14:51:12 -0400831 for line in efn.doc.to_string().lines() {
832 writeln!(out, "//{}", line);
833 }
David Tolnaya73853b2020-04-20 01:19:56 -0700834 let local_name = match &efn.sig.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800835 None => efn.name.cxx.to_string(),
836 Some(receiver) => format!("{}::{}", out.types.resolve(&receiver.ty).cxx, efn.name.cxx),
David Tolnaya73853b2020-04-20 01:19:56 -0700837 };
David Tolnaya7c2ea12020-10-30 21:32:53 -0700838 let invoke = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700839 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700840 write_rust_function_shim_impl(out, &local_name, efn, &invoke, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700841}
842
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700843fn write_rust_function_shim_decl(
David Tolnay75dca2e2020-03-25 20:17:52 -0700844 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700845 local_name: &str,
David Tolnay75dca2e2020-03-25 20:17:52 -0700846 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700847 indirect_call: bool,
848) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700849 write_return_type(out, &sig.ret);
David Tolnay75dca2e2020-03-25 20:17:52 -0700850 write!(out, "{}(", local_name);
851 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400852 if i > 0 {
853 write!(out, ", ");
854 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700855 write_type_space(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400856 write!(out, "{}", arg.ident);
857 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700858 if indirect_call {
859 if !sig.args.is_empty() {
860 write!(out, ", ");
861 }
862 write!(out, "void *extern$");
863 }
David Tolnay1e548172020-03-16 13:37:09 -0700864 write!(out, ")");
David Tolnay86710612020-04-20 00:30:32 -0700865 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800866 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700867 write!(out, " const");
868 }
869 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700870 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700871 write!(out, " noexcept");
872 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700873}
874
875fn write_rust_function_shim_impl(
876 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700877 local_name: &str,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700878 sig: &Signature,
David Tolnay891061b2020-04-19 22:42:33 -0700879 invoke: &Symbol,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700880 indirect_call: bool,
881) {
882 if out.header && sig.receiver.is_some() {
883 // We've already defined this inside the struct.
884 return;
885 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700886 write_rust_function_shim_decl(out, local_name, sig, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400887 if out.header {
888 writeln!(out, ";");
David Tolnay439cde22020-04-20 00:46:25 -0700889 return;
David Tolnay7db73692019-10-20 14:51:12 -0400890 }
David Tolnay439cde22020-04-20 00:46:25 -0700891 writeln!(out, " {{");
892 for arg in &sig.args {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700893 if arg.ty != RustString && out.types.needs_indirect_abi(&arg.ty) {
David Tolnay439cde22020-04-20 00:46:25 -0700894 out.include.utility = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700895 out.builtin.manually_drop = true;
David Tolnay439cde22020-04-20 00:46:25 -0700896 write!(out, " ::rust::ManuallyDrop<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700897 write_type(out, &arg.ty);
David Tolnay439cde22020-04-20 00:46:25 -0700898 writeln!(out, "> {}$(::std::move({0}));", arg.ident);
899 }
900 }
901 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700902 let indirect_return = indirect_return(sig, out.types);
David Tolnay439cde22020-04-20 00:46:25 -0700903 if indirect_return {
David Tolnay74d6d512020-10-31 22:22:03 -0700904 out.builtin.maybe_uninit = true;
David Tolnay439cde22020-04-20 00:46:25 -0700905 write!(out, "::rust::MaybeUninit<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700906 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay439cde22020-04-20 00:46:25 -0700907 writeln!(out, "> return$;");
908 write!(out, " ");
909 } else if let Some(ret) = &sig.ret {
910 write!(out, "return ");
911 match ret {
912 Type::RustBox(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700913 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700914 write!(out, "::from_raw(");
915 }
916 Type::UniquePtr(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700917 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700918 write!(out, "(");
919 }
920 Type::Ref(_) => write!(out, "*"),
David Tolnay74d6d512020-10-31 22:22:03 -0700921 Type::Str(_) => {
922 out.builtin.rust_str_new_unchecked = true;
923 write!(out, "::rust::impl<::rust::Str>::new_unchecked(");
924 }
David Tolnay73b72642020-11-25 17:44:05 -0800925 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700926 out.builtin.rust_slice_new = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800927 write!(out, "::rust::impl<");
928 write_type(out, ret);
929 write!(out, ">::slice(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700930 }
David Tolnay439cde22020-04-20 00:46:25 -0700931 _ => {}
932 }
933 }
934 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700935 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700936 write!(out, "::rust::repr::PtrLen error$ = ");
David Tolnay439cde22020-04-20 00:46:25 -0700937 }
938 write!(out, "{}(", invoke);
David Tolnay9d840362020-11-10 08:50:40 -0800939 let mut needs_comma = false;
David Tolnay439cde22020-04-20 00:46:25 -0700940 if sig.receiver.is_some() {
941 write!(out, "*this");
David Tolnay9d840362020-11-10 08:50:40 -0800942 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700943 }
David Tolnay9d840362020-11-10 08:50:40 -0800944 for arg in &sig.args {
945 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700946 write!(out, ", ");
947 }
948 match &arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700949 Type::Str(_) => {
950 out.builtin.rust_str_repr = true;
951 write!(out, "::rust::impl<::rust::Str>::repr(");
952 }
David Tolnay73b72642020-11-25 17:44:05 -0800953 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700954 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800955 write!(out, "::rust::impl<");
956 write_type(out, &arg.ty);
957 write!(out, ">::repr(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700958 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700959 ty if out.types.needs_indirect_abi(ty) => write!(out, "&"),
David Tolnay439cde22020-04-20 00:46:25 -0700960 _ => {}
961 }
962 write!(out, "{}", arg.ident);
963 match &arg.ty {
964 Type::RustBox(_) => write!(out, ".into_raw()"),
965 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800966 Type::Str(_) | Type::SliceRef(_) => write!(out, ")"),
David Tolnaya7c2ea12020-10-30 21:32:53 -0700967 ty if ty != RustString && out.types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay439cde22020-04-20 00:46:25 -0700968 _ => {}
969 }
David Tolnay9d840362020-11-10 08:50:40 -0800970 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700971 }
972 if indirect_return {
David Tolnay9d840362020-11-10 08:50:40 -0800973 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700974 write!(out, ", ");
975 }
976 write!(out, "&return$.value");
David Tolnay9d840362020-11-10 08:50:40 -0800977 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700978 }
979 if indirect_call {
David Tolnay9d840362020-11-10 08:50:40 -0800980 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700981 write!(out, ", ");
982 }
983 write!(out, "extern$");
984 }
985 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -0400986 if !indirect_return {
987 if let Some(ret) = &sig.ret {
David Tolnay73b72642020-11-25 17:44:05 -0800988 if let Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::SliceRef(_) = ret {
David Tolnay22602b42020-09-21 18:04:05 -0400989 write!(out, ")");
990 }
David Tolnay439cde22020-04-20 00:46:25 -0700991 }
992 }
993 writeln!(out, ";");
994 if sig.throws {
David Tolnay74d6d512020-10-31 22:22:03 -0700995 out.builtin.rust_error = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700996 writeln!(out, " if (error$.ptr) {{");
David Tolnay84ddf9e2020-10-31 15:36:48 -0700997 writeln!(out, " throw ::rust::impl<::rust::Error>::error(error$);");
David Tolnay439cde22020-04-20 00:46:25 -0700998 writeln!(out, " }}");
999 }
1000 if indirect_return {
1001 out.include.utility = true;
1002 writeln!(out, " return ::std::move(return$.value);");
1003 }
1004 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001005}
1006
David Tolnaya7c2ea12020-10-30 21:32:53 -07001007fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001008 match ty {
1009 None => write!(out, "void "),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001010 Some(ty) => write_type_space(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001011 }
1012}
1013
David Tolnay75dca2e2020-03-25 20:17:52 -07001014fn indirect_return(sig: &Signature, types: &Types) -> bool {
1015 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -07001016 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -07001017 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -07001018}
1019
David Tolnaya7c2ea12020-10-30 21:32:53 -07001020fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
David Tolnay99642622020-03-25 13:07:35 -07001021 match ty {
1022 Type::RustBox(ty) | Type::UniquePtr(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001023 write_type_space(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001024 write!(out, "*");
1025 }
1026 Type::Ref(ty) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001027 if !ty.mutable {
David Tolnay99642622020-03-25 13:07:35 -07001028 write!(out, "const ");
1029 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001030 write_type(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001031 write!(out, " *");
1032 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001033 _ => write_type(out, ty),
David Tolnay99642622020-03-25 13:07:35 -07001034 }
1035}
1036
David Tolnaya7c2ea12020-10-30 21:32:53 -07001037fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
1038 write_indirect_return_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001039 match ty {
1040 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
David Tolnay73b72642020-11-25 17:44:05 -08001041 Type::Str(_) | Type::SliceRef(_) => write!(out, " "),
David Tolnay99642622020-03-25 13:07:35 -07001042 _ => write_space_after_type(out, ty),
1043 }
1044}
1045
David Tolnaya7c2ea12020-10-30 21:32:53 -07001046fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001047 match ty {
1048 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001049 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001050 write!(out, "*");
1051 }
David Tolnay4a441222020-01-25 16:24:27 -08001052 Some(Type::Ref(ty)) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001053 if !ty.mutable {
David Tolnay4a441222020-01-25 16:24:27 -08001054 write!(out, "const ");
1055 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001056 write_type(out, &ty.inner);
David Tolnay4a441222020-01-25 16:24:27 -08001057 write!(out, " *");
1058 }
David Tolnay73b72642020-11-25 17:44:05 -08001059 Some(Type::Str(_)) | Some(Type::SliceRef(_)) => {
David Tolnay919085c2020-10-31 22:32:22 -07001060 out.builtin.ptr_len = true;
1061 write!(out, "::rust::repr::PtrLen ");
1062 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001063 Some(ty) if out.types.needs_indirect_abi(ty) => write!(out, "void "),
1064 _ => write_return_type(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001065 }
1066}
1067
David Tolnaya7c2ea12020-10-30 21:32:53 -07001068fn write_extern_arg(out: &mut OutFile, arg: &Var) {
David Tolnay7db73692019-10-20 14:51:12 -04001069 match &arg.ty {
David Tolnay4377a9e2020-04-24 15:20:26 -07001070 Type::RustBox(ty) | Type::UniquePtr(ty) | Type::CxxVector(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001071 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001072 write!(out, "*");
1073 }
David Tolnay73b72642020-11-25 17:44:05 -08001074 Type::Str(_) | Type::SliceRef(_) => {
David Tolnay919085c2020-10-31 22:32:22 -07001075 out.builtin.ptr_len = true;
1076 write!(out, "::rust::repr::PtrLen ");
1077 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001078 _ => write_type_space(out, &arg.ty),
David Tolnay7db73692019-10-20 14:51:12 -04001079 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001080 if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay7db73692019-10-20 14:51:12 -04001081 write!(out, "*");
1082 }
1083 write!(out, "{}", arg.ident);
1084}
1085
David Tolnaya7c2ea12020-10-30 21:32:53 -07001086fn write_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001087 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -07001088 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnayf6a89f22020-05-10 23:39:27 -07001089 Some(atom) => write_atom(out, atom),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001090 None => write!(out, "{}", out.types.resolve(ident).to_fully_qualified()),
David Tolnay7db73692019-10-20 14:51:12 -04001091 },
1092 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -08001093 write!(out, "::rust::Box<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001094 write_type(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001095 write!(out, ">");
1096 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001097 Type::RustVec(ty) => {
1098 write!(out, "::rust::Vec<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001099 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001100 write!(out, ">");
1101 }
David Tolnay7db73692019-10-20 14:51:12 -04001102 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -08001103 write!(out, "::std::unique_ptr<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001104 write_type(out, &ptr.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001105 write!(out, ">");
1106 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001107 Type::CxxVector(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +07001108 write!(out, "::std::vector<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001109 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001110 write!(out, ">");
1111 }
David Tolnay7db73692019-10-20 14:51:12 -04001112 Type::Ref(r) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001113 if !r.mutable {
David Tolnay7db73692019-10-20 14:51:12 -04001114 write!(out, "const ");
1115 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001116 write_type(out, &r.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001117 write!(out, " &");
1118 }
1119 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -08001120 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -04001121 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001122 Type::SliceRef(slice) => {
David Tolnayc5629f02020-11-23 18:32:46 -08001123 write!(out, "::rust::Slice<");
David Tolnay5515a9e2020-11-25 19:07:54 -08001124 if slice.mutability.is_none() {
David Tolnayc5629f02020-11-23 18:32:46 -08001125 write!(out, "const ");
1126 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001127 write_type(out, &slice.inner);
1128 write!(out, ">");
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001129 }
David Tolnay75dca2e2020-03-25 20:17:52 -07001130 Type::Fn(f) => {
David Tolnayf031c322020-11-29 19:41:33 -08001131 write!(out, "::rust::Fn<");
David Tolnay75dca2e2020-03-25 20:17:52 -07001132 match &f.ret {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001133 Some(ret) => write_type(out, ret),
David Tolnay75dca2e2020-03-25 20:17:52 -07001134 None => write!(out, "void"),
1135 }
1136 write!(out, "(");
1137 for (i, arg) in f.args.iter().enumerate() {
1138 if i > 0 {
1139 write!(out, ", ");
1140 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001141 write_type(out, &arg.ty);
David Tolnay75dca2e2020-03-25 20:17:52 -07001142 }
1143 write!(out, ")>");
1144 }
Xiangpeng Hao78762352020-11-12 10:24:18 +08001145 Type::Array(a) => {
1146 write!(out, "::std::array<");
1147 write_type(out, &a.inner);
1148 write!(out, ", {}>", &a.len);
1149 }
David Tolnay2fb14e92020-03-15 23:11:38 -07001150 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001151 }
1152}
1153
David Tolnayf6a89f22020-05-10 23:39:27 -07001154fn write_atom(out: &mut OutFile, atom: Atom) {
1155 match atom {
1156 Bool => write!(out, "bool"),
David Tolnayb3873dc2020-11-25 19:47:49 -08001157 Char => write!(out, "char"),
David Tolnayf6a89f22020-05-10 23:39:27 -07001158 U8 => write!(out, "uint8_t"),
1159 U16 => write!(out, "uint16_t"),
1160 U32 => write!(out, "uint32_t"),
1161 U64 => write!(out, "uint64_t"),
1162 Usize => write!(out, "size_t"),
1163 I8 => write!(out, "int8_t"),
1164 I16 => write!(out, "int16_t"),
1165 I32 => write!(out, "int32_t"),
1166 I64 => write!(out, "int64_t"),
1167 Isize => write!(out, "::rust::isize"),
1168 F32 => write!(out, "float"),
1169 F64 => write!(out, "double"),
1170 CxxString => write!(out, "::std::string"),
1171 RustString => write!(out, "::rust::String"),
1172 }
1173}
1174
David Tolnaya7c2ea12020-10-30 21:32:53 -07001175fn write_type_space(out: &mut OutFile, ty: &Type) {
1176 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001177 write_space_after_type(out, ty);
1178}
1179
1180fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001181 match ty {
David Tolnayeb952ba2020-04-14 15:02:24 -07001182 Type::Ident(_)
1183 | Type::RustBox(_)
1184 | Type::UniquePtr(_)
1185 | Type::Str(_)
David Tolnay4377a9e2020-04-24 15:20:26 -07001186 | Type::CxxVector(_)
Myron Ahneba35cf2020-02-05 19:41:51 +07001187 | Type::RustVec(_)
David Tolnay73b72642020-11-25 17:44:05 -08001188 | Type::SliceRef(_)
Xiangpeng Hao78762352020-11-12 10:24:18 +08001189 | Type::Fn(_)
1190 | Type::Array(_) => write!(out, " "),
David Tolnay7db73692019-10-20 14:51:12 -04001191 Type::Ref(_) => {}
David Tolnaye0dca7b2020-11-25 17:18:57 -08001192 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001193 }
1194}
1195
David Tolnay1bdb4712020-11-25 07:27:54 -08001196#[derive(Copy, Clone)]
1197enum UniquePtr<'a> {
1198 Ident(&'a ResolvableName),
1199 CxxVector(&'a ResolvableName),
1200}
1201
1202trait ToTypename {
1203 fn to_typename(&self, types: &Types) -> String;
1204}
1205
1206impl ToTypename for ResolvableName {
1207 fn to_typename(&self, types: &Types) -> String {
1208 types.resolve(self).to_fully_qualified()
David Tolnay2eca4a02020-04-24 19:50:51 -07001209 }
1210}
1211
David Tolnay1bdb4712020-11-25 07:27:54 -08001212impl<'a> ToTypename for UniquePtr<'a> {
1213 fn to_typename(&self, types: &Types) -> String {
1214 match self {
1215 UniquePtr::Ident(ident) => ident.to_typename(types),
1216 UniquePtr::CxxVector(element) => {
1217 format!("::std::vector<{}>", element.to_typename(types))
1218 }
1219 }
1220 }
1221}
1222
1223trait ToMangled {
1224 fn to_mangled(&self, types: &Types) -> Symbol;
1225}
1226
1227impl ToMangled for ResolvableName {
1228 fn to_mangled(&self, types: &Types) -> Symbol {
1229 self.to_symbol(types)
1230 }
1231}
1232
1233impl<'a> ToMangled for UniquePtr<'a> {
1234 fn to_mangled(&self, types: &Types) -> Symbol {
1235 match self {
1236 UniquePtr::Ident(ident) => ident.to_mangled(types),
1237 UniquePtr::CxxVector(element) => element.to_mangled(types).prefix_with("std$vector$"),
1238 }
David Tolnaybae50ef2020-04-25 12:38:41 -07001239 }
1240}
1241
David Tolnaya7c2ea12020-10-30 21:32:53 -07001242fn write_generic_instantiations(out: &mut OutFile) {
David Tolnay169bb472020-11-01 21:04:24 -08001243 if out.header {
1244 return;
1245 }
1246
1247 out.next_section();
David Tolnay078c90f2020-11-01 13:31:08 -08001248 out.set_namespace(Default::default());
David Tolnay0c033e32020-11-01 15:15:48 -08001249 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -07001250 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -04001251 if let Type::RustBox(ty) = ty {
1252 if let Type::Ident(inner) = &ty.inner {
1253 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001254 write_rust_box_extern(out, &out.types.resolve(&inner));
David Tolnay7db73692019-10-20 14:51:12 -04001255 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001256 } else if let Type::RustVec(ty) = ty {
David Tolnay6787be62020-04-25 11:01:02 -07001257 if let Type::Ident(inner) = &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001258 if Atom::from(&inner.rust).is_none() {
David Tolnay6787be62020-04-25 11:01:02 -07001259 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001260 write_rust_vec_extern(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001261 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001262 }
David Tolnay7db73692019-10-20 14:51:12 -04001263 } else if let Type::UniquePtr(ptr) = ty {
1264 if let Type::Ident(inner) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001265 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001266 && (!out.types.aliases.contains_key(&inner.rust)
1267 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001268 {
David Tolnay7db73692019-10-20 14:51:12 -04001269 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001270 write_unique_ptr(out, inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001271 }
1272 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001273 } else if let Type::CxxVector(ptr) = ty {
Myron Ahneba35cf2020-02-05 19:41:51 +07001274 if let Type::Ident(inner) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001275 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001276 && (!out.types.aliases.contains_key(&inner.rust)
1277 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001278 {
Myron Ahneba35cf2020-02-05 19:41:51 +07001279 out.next_section();
David Tolnay1bdb4712020-11-25 07:27:54 -08001280 write_cxx_vector(out, inner);
David Tolnay7db73692019-10-20 14:51:12 -04001281 }
1282 }
1283 }
1284 }
David Tolnay0c033e32020-11-01 15:15:48 -08001285 out.end_block(Block::ExternC);
David Tolnay7db73692019-10-20 14:51:12 -04001286
David Tolnay0c033e32020-11-01 15:15:48 -08001287 out.begin_block(Block::Namespace("rust"));
David Tolnay0f0162f2020-11-16 23:43:37 -08001288 out.begin_block(Block::InlineNamespace("cxxbridge1"));
David Tolnaya7c2ea12020-10-30 21:32:53 -07001289 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -04001290 if let Type::RustBox(ty) = ty {
1291 if let Type::Ident(inner) = &ty.inner {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001292 write_rust_box_impl(out, &out.types.resolve(&inner));
David Tolnay7db73692019-10-20 14:51:12 -04001293 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001294 } else if let Type::RustVec(ty) = ty {
David Tolnay6787be62020-04-25 11:01:02 -07001295 if let Type::Ident(inner) = &ty.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001296 if Atom::from(&inner.rust).is_none() {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001297 write_rust_vec_impl(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001298 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001299 }
David Tolnay7db73692019-10-20 14:51:12 -04001300 }
1301 }
David Tolnay0f0162f2020-11-16 23:43:37 -08001302 out.end_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay0c033e32020-11-01 15:15:48 -08001303 out.end_block(Block::Namespace("rust"));
David Tolnay7db73692019-10-20 14:51:12 -04001304}
1305
David Tolnay8faec772020-11-02 00:18:19 -08001306fn write_rust_box_extern(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001307 let inner = ident.to_fully_qualified();
1308 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001309
David Tolnay0f0162f2020-11-16 23:43:37 -08001310 writeln!(out, "#ifndef CXXBRIDGE1_RUST_BOX_{}", instance);
1311 writeln!(out, "#define CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001312 writeln!(
1313 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001314 "void cxxbridge1$box${}$uninit(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001315 instance, inner,
1316 );
1317 writeln!(
1318 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001319 "void cxxbridge1$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001320 instance, inner,
1321 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001322 writeln!(out, "#endif // CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001323}
1324
David Tolnaya7c2ea12020-10-30 21:32:53 -07001325fn write_rust_vec_extern(out: &mut OutFile, element: &ResolvableName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001326 let inner = element.to_typename(out.types);
1327 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001328
David Tolnay0f0162f2020-11-16 23:43:37 -08001329 writeln!(out, "#ifndef CXXBRIDGE1_RUST_VEC_{}", instance);
1330 writeln!(out, "#define CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001331 writeln!(
1332 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001333 "void cxxbridge1$rust_vec${}$new(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnayf97c2d52020-04-25 16:37:48 -07001334 instance, inner,
1335 );
1336 writeln!(
1337 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001338 "void cxxbridge1$rust_vec${}$drop(::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001339 instance, inner,
1340 );
1341 writeln!(
1342 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001343 "size_t cxxbridge1$rust_vec${}$len(const ::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001344 instance, inner,
1345 );
David Tolnay219c0792020-04-24 20:31:37 -07001346 writeln!(
1347 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001348 "const {} *cxxbridge1$rust_vec${}$data(const ::rust::Vec<{0}> *ptr) noexcept;",
David Tolnay219c0792020-04-24 20:31:37 -07001349 inner, instance,
1350 );
David Tolnay503d0192020-04-24 22:18:56 -07001351 writeln!(
1352 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001353 "void cxxbridge1$rust_vec${}$reserve_total(::rust::Vec<{}> *ptr, size_t cap) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001354 instance, inner,
1355 );
1356 writeln!(
1357 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001358 "void cxxbridge1$rust_vec${}$set_len(::rust::Vec<{}> *ptr, size_t len) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001359 instance, inner,
1360 );
1361 writeln!(
1362 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001363 "size_t cxxbridge1$rust_vec${}$stride() noexcept;",
David Tolnay503d0192020-04-24 22:18:56 -07001364 instance,
1365 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001366 writeln!(out, "#endif // CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001367}
1368
David Tolnay8faec772020-11-02 00:18:19 -08001369fn write_rust_box_impl(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001370 let inner = ident.to_fully_qualified();
1371 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001372
1373 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -08001374 writeln!(out, "void Box<{}>::uninit() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001375 writeln!(out, " cxxbridge1$box${}$uninit(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001376 writeln!(out, "}}");
1377
1378 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -08001379 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001380 writeln!(out, " cxxbridge1$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001381 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001382}
1383
David Tolnaya7c2ea12020-10-30 21:32:53 -07001384fn write_rust_vec_impl(out: &mut OutFile, element: &ResolvableName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001385 let inner = element.to_typename(out.types);
1386 let instance = element.to_mangled(out.types);
David Tolnay4791f1c2020-03-17 21:53:16 -07001387
Myron Ahneba35cf2020-02-05 19:41:51 +07001388 writeln!(out, "template <>");
David Tolnayf97c2d52020-04-25 16:37:48 -07001389 writeln!(out, "Vec<{}>::Vec() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001390 writeln!(out, " cxxbridge1$rust_vec${}$new(this);", instance);
David Tolnayf97c2d52020-04-25 16:37:48 -07001391 writeln!(out, "}}");
1392
1393 writeln!(out, "template <>");
Myron Ahneba35cf2020-02-05 19:41:51 +07001394 writeln!(out, "void Vec<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001395 writeln!(out, " return cxxbridge1$rust_vec${}$drop(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001396 writeln!(out, "}}");
1397
1398 writeln!(out, "template <>");
1399 writeln!(out, "size_t Vec<{}>::size() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001400 writeln!(out, " return cxxbridge1$rust_vec${}$len(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001401 writeln!(out, "}}");
David Tolnay219c0792020-04-24 20:31:37 -07001402
1403 writeln!(out, "template <>");
1404 writeln!(out, "const {} *Vec<{0}>::data() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001405 writeln!(out, " return cxxbridge1$rust_vec${}$data(this);", instance);
David Tolnay219c0792020-04-24 20:31:37 -07001406 writeln!(out, "}}");
David Tolnay503d0192020-04-24 22:18:56 -07001407
1408 writeln!(out, "template <>");
David Tolnayfb6b73c2020-11-10 14:32:16 -08001409 writeln!(
1410 out,
1411 "void Vec<{}>::reserve_total(size_t cap) noexcept {{",
1412 inner,
1413 );
1414 writeln!(
1415 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001416 " return cxxbridge1$rust_vec${}$reserve_total(this, cap);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001417 instance,
1418 );
1419 writeln!(out, "}}");
1420
1421 writeln!(out, "template <>");
1422 writeln!(out, "void Vec<{}>::set_len(size_t len) noexcept {{", inner);
1423 writeln!(
1424 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001425 " return cxxbridge1$rust_vec${}$set_len(this, len);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001426 instance,
1427 );
1428 writeln!(out, "}}");
1429
1430 writeln!(out, "template <>");
David Tolnay503d0192020-04-24 22:18:56 -07001431 writeln!(out, "size_t Vec<{}>::stride() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001432 writeln!(out, " return cxxbridge1$rust_vec${}$stride();", instance);
David Tolnay503d0192020-04-24 22:18:56 -07001433 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001434}
1435
David Tolnaya7c2ea12020-10-30 21:32:53 -07001436fn write_unique_ptr(out: &mut OutFile, ident: &ResolvableName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001437 let ty = UniquePtr::Ident(ident);
1438 let instance = ty.to_mangled(out.types);
David Tolnay63da4d32020-04-25 09:41:12 -07001439
David Tolnay0f0162f2020-11-16 23:43:37 -08001440 writeln!(out, "#ifndef CXXBRIDGE1_UNIQUE_PTR_{}", instance);
1441 writeln!(out, "#define CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001442
David Tolnay1bdb4712020-11-25 07:27:54 -08001443 write_unique_ptr_common(out, ty);
David Tolnay63da4d32020-04-25 09:41:12 -07001444
David Tolnay0f0162f2020-11-16 23:43:37 -08001445 writeln!(out, "#endif // CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001446}
1447
1448// Shared by UniquePtr<T> and UniquePtr<CxxVector<T>>.
David Tolnay1bdb4712020-11-25 07:27:54 -08001449fn write_unique_ptr_common(out: &mut OutFile, ty: UniquePtr) {
David Tolnay0ecd05a2020-07-29 16:32:03 -07001450 out.include.new = true;
Myron Ahneba35cf2020-02-05 19:41:51 +07001451 out.include.utility = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001452 let inner = ty.to_typename(out.types);
1453 let instance = ty.to_mangled(out.types);
David Tolnay7db73692019-10-20 14:51:12 -04001454
David Tolnay63da4d32020-04-25 09:41:12 -07001455 let can_construct_from_value = match ty {
David Tolnayca0f9da2020-10-16 13:16:17 -07001456 // Some aliases are to opaque types; some are to trivial types. We can't
1457 // know at code generation time, so we generate both C++ and Rust side
1458 // bindings for a "new" method anyway. But the Rust code can't be called
1459 // for Opaque types because the 'new' method is not implemented.
David Tolnay1bdb4712020-11-25 07:27:54 -08001460 UniquePtr::Ident(ident) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001461 out.types.structs.contains_key(&ident.rust)
David Tolnay15609ab2020-11-25 07:14:12 -08001462 || out.types.enums.contains_key(&ident.rust)
David Tolnaya7c2ea12020-10-30 21:32:53 -07001463 || out.types.aliases.contains_key(&ident.rust)
David Tolnayca0f9da2020-10-16 13:16:17 -07001464 }
David Tolnay1bdb4712020-11-25 07:27:54 -08001465 UniquePtr::CxxVector(_) => false,
David Tolnay63da4d32020-04-25 09:41:12 -07001466 };
1467
David Tolnay53462762020-11-25 07:08:10 -08001468 let conditional_delete = match ty {
1469 UniquePtr::Ident(ident) => {
1470 !out.types.structs.contains_key(&ident.rust)
1471 && !out.types.enums.contains_key(&ident.rust)
1472 }
1473 UniquePtr::CxxVector(_) => false,
1474 };
1475
1476 if conditional_delete {
1477 out.builtin.is_complete = true;
1478 let definition = match ty {
1479 UniquePtr::Ident(ty) => &out.types.resolve(ty).cxx,
1480 UniquePtr::CxxVector(_) => unreachable!(),
1481 };
1482 writeln!(
1483 out,
1484 "static_assert(::rust::is_complete<{}>::value, \"definition of {} is required\");",
1485 inner, definition,
1486 );
1487 }
David Tolnay7db73692019-10-20 14:51:12 -04001488 writeln!(
1489 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001490 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001491 inner,
1492 );
1493 writeln!(
1494 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001495 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001496 inner,
1497 );
1498 writeln!(
1499 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001500 "void cxxbridge1$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001501 instance, inner,
1502 );
David Tolnay7e219b82020-03-01 13:14:51 -08001503 writeln!(out, " new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001504 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001505 if can_construct_from_value {
1506 writeln!(
David Tolnay53838912020-04-09 20:56:44 -07001507 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001508 "void cxxbridge1$unique_ptr${}$new(::std::unique_ptr<{}> *ptr, {} *value) noexcept {{",
David Tolnay53838912020-04-09 20:56:44 -07001509 instance, inner, inner,
1510 );
David Tolnay63da4d32020-04-25 09:41:12 -07001511 writeln!(
1512 out,
1513 " new (ptr) ::std::unique_ptr<{}>(new {}(::std::move(*value)));",
1514 inner, inner,
1515 );
1516 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -07001517 }
David Tolnay7db73692019-10-20 14:51:12 -04001518 writeln!(
1519 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001520 "void cxxbridge1$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001521 instance, inner, inner,
1522 );
David Tolnay7e219b82020-03-01 13:14:51 -08001523 writeln!(out, " new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001524 writeln!(out, "}}");
1525 writeln!(
1526 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001527 "const {} *cxxbridge1$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001528 inner, instance, inner,
1529 );
1530 writeln!(out, " return ptr.get();");
1531 writeln!(out, "}}");
1532 writeln!(
1533 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001534 "{} *cxxbridge1$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001535 inner, instance, inner,
1536 );
1537 writeln!(out, " return ptr.release();");
1538 writeln!(out, "}}");
1539 writeln!(
1540 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001541 "void cxxbridge1$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001542 instance, inner,
1543 );
David Tolnay53462762020-11-25 07:08:10 -08001544 if conditional_delete {
1545 out.builtin.deleter_if = true;
1546 writeln!(
1547 out,
1548 " ::rust::deleter_if<::rust::is_complete<{}>::value>{{}}(ptr);",
1549 inner,
1550 );
1551 } else {
1552 writeln!(out, " ptr->~unique_ptr();");
1553 }
David Tolnay7db73692019-10-20 14:51:12 -04001554 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001555}
Myron Ahneba35cf2020-02-05 19:41:51 +07001556
David Tolnay1bdb4712020-11-25 07:27:54 -08001557fn write_cxx_vector(out: &mut OutFile, element: &ResolvableName) {
1558 let inner = element.to_typename(out.types);
1559 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001560
David Tolnay0f0162f2020-11-16 23:43:37 -08001561 writeln!(out, "#ifndef CXXBRIDGE1_VECTOR_{}", instance);
1562 writeln!(out, "#define CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001563 writeln!(
1564 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001565 "size_t cxxbridge1$std$vector${}$size(const ::std::vector<{}> &s) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001566 instance, inner,
1567 );
1568 writeln!(out, " return s.size();");
1569 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001570 writeln!(
1571 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001572 "const {} *cxxbridge1$std$vector${}$get_unchecked(const ::std::vector<{}> &s, size_t pos) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001573 inner, instance, inner,
1574 );
David Tolnayb3fcf7b2020-04-30 22:58:28 -07001575 writeln!(out, " return &s[pos];");
Myron Ahneba35cf2020-02-05 19:41:51 +07001576 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001577
David Tolnay1bdb4712020-11-25 07:27:54 -08001578 write_unique_ptr_common(out, UniquePtr::CxxVector(element));
David Tolnay63da4d32020-04-25 09:41:12 -07001579
David Tolnay0f0162f2020-11-16 23:43:37 -08001580 writeln!(out, "#endif // CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001581}