blob: 7da60ac39fd5bb6186f2ccef5fd028061452831d [file] [log] [blame]
David Tolnayb725d5a2020-12-20 20:27:10 -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;
David Tolnay5b1d8632020-12-20 22:05:11 -08007use crate::syntax::trivial::{self, TrivialReason};
Adrian Taylorc8713432020-10-21 18:20:55 -07008use crate::syntax::{
David Tolnay75ea17c2020-12-06 21:08:34 -08009 derive, mangle, Api, Enum, ExternFn, ExternType, Pair, RustName, Signature, Struct, Trait,
David Tolnay5b1d8632020-12-20 22:05:11 -080010 Type, TypeAlias, Types, Var,
Adrian Taylorc8713432020-10-21 18:20:55 -070011};
David Tolnay7db73692019-10-20 14:51:12 -040012use proc_macro2::Ident;
David Tolnay5439fa12020-11-03 18:45:01 -080013use std::collections::{HashMap, HashSet};
David Tolnay7db73692019-10-20 14:51:12 -040014
David Tolnayac7188c2020-11-01 16:58:16 -080015pub(super) fn gen(apis: &[Api], types: &Types, opt: &Opt, header: bool) -> Vec<u8> {
David Tolnaye1476af2020-11-01 13:47:25 -080016 let mut out_file = OutFile::new(header, opt, types);
David Tolnay7db73692019-10-20 14:51:12 -040017 let out = &mut out_file;
18
David Tolnaydfb82d72020-11-02 00:10:10 -080019 pick_includes_and_builtins(out, apis);
David Tolnay4aae7c02020-10-28 12:35:42 -070020 out.include.extend(&opt.include);
David Tolnay7db73692019-10-20 14:51:12 -040021
David Tolnay7b0e5102020-11-01 23:22:12 -080022 write_forward_declarations(out, apis);
David Tolnaye1109d92020-11-01 20:51:56 -080023 write_data_structures(out, apis);
24 write_functions(out, apis);
David Tolnay169bb472020-11-01 21:04:24 -080025 write_generic_instantiations(out);
David Tolnay7db73692019-10-20 14:51:12 -040026
David Tolnay3374d8d2020-10-31 22:18:45 -070027 builtin::write(out);
David Tolnay2f3e90b2020-10-31 22:16:51 -070028 include::write(out);
Adrian Taylorc8713432020-10-21 18:20:55 -070029
David Tolnayac7188c2020-11-01 16:58:16 -080030 out_file.content()
Adrian Taylorc8713432020-10-21 18:20:55 -070031}
32
David Tolnay7b0e5102020-11-01 23:22:12 -080033fn write_forward_declarations(out: &mut OutFile, apis: &[Api]) {
34 let needs_forward_declaration = |api: &&Api| match api {
David Tolnay4bfca112020-11-01 23:59:11 -080035 Api::Struct(_) | Api::CxxType(_) | Api::RustType(_) => true,
David Tolnay17a934c2020-11-02 00:40:04 -080036 Api::Enum(enm) => !out.types.cxx.contains(&enm.name.rust),
David Tolnay7b0e5102020-11-01 23:22:12 -080037 _ => false,
38 };
Adrian Taylorc8713432020-10-21 18:20:55 -070039
David Tolnay7b0e5102020-11-01 23:22:12 -080040 let apis_by_namespace =
41 NamespaceEntries::new(apis.iter().filter(needs_forward_declaration).collect());
42
David Tolnayd920be52020-11-01 23:34:30 -080043 write(out, &apis_by_namespace, 0);
David Tolnay7b0e5102020-11-01 23:22:12 -080044
David Tolnayd920be52020-11-01 23:34:30 -080045 fn write(out: &mut OutFile, ns_entries: &NamespaceEntries, indent: usize) {
David Tolnay7b0e5102020-11-01 23:22:12 -080046 let apis = ns_entries.direct_content();
47
48 for api in apis {
David Tolnayd920be52020-11-01 23:34:30 -080049 write!(out, "{:1$}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080050 match api {
David Tolnay17a934c2020-11-02 00:40:04 -080051 Api::Struct(strct) => write_struct_decl(out, &strct.name.cxx),
David Tolnay0e3ee8e2020-11-01 23:46:52 -080052 Api::Enum(enm) => write_enum_decl(out, enm),
David Tolnay17a934c2020-11-02 00:40:04 -080053 Api::CxxType(ety) => write_struct_using(out, &ety.name),
54 Api::RustType(ety) => write_struct_decl(out, &ety.name.cxx),
David Tolnay7b0e5102020-11-01 23:22:12 -080055 _ => unreachable!(),
56 }
David Tolnay7db73692019-10-20 14:51:12 -040057 }
David Tolnay7db73692019-10-20 14:51:12 -040058
David Tolnay7b0e5102020-11-01 23:22:12 -080059 for (namespace, nested_ns_entries) in ns_entries.nested_content() {
David Tolnayd920be52020-11-01 23:34:30 -080060 writeln!(out, "{:2$}namespace {} {{", "", namespace, indent);
61 write(out, nested_ns_entries, indent + 2);
62 writeln!(out, "{:1$}}}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080063 }
Adrian Taylorf9213622020-10-31 22:25:42 -070064 }
65}
66
David Tolnaye1109d92020-11-01 20:51:56 -080067fn write_data_structures<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayf94bef12020-04-17 14:46:42 -070068 let mut methods_for_type = HashMap::new();
David Tolnay630af882020-10-31 22:03:47 -070069 for api in apis {
David Tolnay102c7ea2020-11-08 18:58:09 -080070 if let Api::CxxFunction(efn) | Api::RustFunction(efn) = api {
David Tolnayf94bef12020-04-17 14:46:42 -070071 if let Some(receiver) = &efn.sig.receiver {
72 methods_for_type
Adrian Taylorc8713432020-10-21 18:20:55 -070073 .entry(&receiver.ty.rust)
David Tolnayf94bef12020-04-17 14:46:42 -070074 .or_insert_with(Vec::new)
75 .push(efn);
76 }
77 }
78 }
Joel Galenson968738f2020-04-15 14:19:33 -070079
David Tolnay5439fa12020-11-03 18:45:01 -080080 let mut structs_written = HashSet::new();
81 let mut toposorted_structs = out.types.toposorted_structs.iter();
David Tolnay9622b462020-11-02 21:34:42 -080082 for api in apis {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -070083 match api {
David Tolnay5439fa12020-11-03 18:45:01 -080084 Api::Struct(strct) if !structs_written.contains(&strct.name.rust) => {
85 for next in &mut toposorted_structs {
86 if !out.types.cxx.contains(&strct.name.rust) {
87 out.next_section();
David Tolnay102c7ea2020-11-08 18:58:09 -080088 let methods = methods_for_type
89 .get(&strct.name.rust)
90 .map(Vec::as_slice)
91 .unwrap_or_default();
92 write_struct(out, next, methods);
David Tolnay5439fa12020-11-03 18:45:01 -080093 }
94 structs_written.insert(&next.name.rust);
95 if next.name.rust == strct.name.rust {
96 break;
97 }
98 }
99 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700100 Api::Enum(enm) => {
101 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800102 if out.types.cxx.contains(&enm.name.rust) {
Joel Galenson905eb2e2020-05-04 14:58:14 -0700103 check_enum(out, enm);
104 } else {
105 write_enum(out, enm);
106 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700107 }
David Tolnayc1fe0052020-04-17 15:15:06 -0700108 Api::RustType(ety) => {
David Tolnay17a934c2020-11-02 00:40:04 -0800109 if let Some(methods) = methods_for_type.get(&ety.name.rust) {
David Tolnay46a54e72020-04-17 14:48:21 -0700110 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700111 write_struct_with_methods(out, ety, methods);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700112 }
David Tolnayc1fe0052020-04-17 15:15:06 -0700113 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700114 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400115 }
116 }
117
David Tolnayfabca772020-10-03 21:25:41 -0700118 out.next_section();
119 for api in apis {
120 if let Api::TypeAlias(ety) = api {
Adrian Taylorf831a5a2020-12-07 16:49:19 -0800121 if let Some(reasons) = out.types.required_trivial.get(&ety.name.rust) {
David Tolnay5b1d8632020-12-20 22:05:11 -0800122 check_trivial_extern_type(out, ety, reasons)
David Tolnayfabca772020-10-03 21:25:41 -0700123 }
124 }
125 }
David Tolnay1b0339c2020-11-01 20:37:50 -0800126}
127
David Tolnaye1109d92020-11-01 20:51:56 -0800128fn write_functions<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayce5a91f2020-10-31 22:42:08 -0700129 if !out.header {
David Tolnay7db73692019-10-20 14:51:12 -0400130 for api in apis {
David Tolnay04770742020-11-01 13:50:50 -0800131 match api {
David Tolnayb960ed22020-11-27 14:34:30 -0800132 Api::Struct(strct) => write_struct_operator_decls(out, strct),
David Tolnay04770742020-11-01 13:50:50 -0800133 Api::CxxFunction(efn) => write_cxx_function_shim(out, efn),
134 Api::RustFunction(efn) => write_rust_function_decl(out, efn),
135 _ => {}
136 }
David Tolnay7db73692019-10-20 14:51:12 -0400137 }
David Tolnay7da38202020-11-27 17:36:16 -0800138
139 write_std_specializations(out, apis);
David Tolnay7db73692019-10-20 14:51:12 -0400140 }
141
142 for api in apis {
David Tolnayb960ed22020-11-27 14:34:30 -0800143 match api {
144 Api::Struct(strct) => write_struct_operators(out, strct),
145 Api::RustFunction(efn) => {
146 out.next_section();
147 write_rust_function_shim(out, efn);
148 }
149 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400150 }
151 }
David Tolnay7db73692019-10-20 14:51:12 -0400152}
153
David Tolnay7da38202020-11-27 17:36:16 -0800154fn write_std_specializations(out: &mut OutFile, apis: &[Api]) {
155 out.set_namespace(Default::default());
156 out.begin_block(Block::Namespace("std"));
157
158 for api in apis {
159 if let Api::Struct(strct) = api {
160 if derive::contains(&strct.derives, Trait::Hash) {
161 out.next_section();
David Tolnay12320e12020-12-09 23:09:36 -0800162 out.include.cstddef = true;
David Tolnay08a03db2020-12-09 23:04:36 -0800163 out.include.functional = true;
David Tolnay7da38202020-11-27 17:36:16 -0800164 let qualified = strct.name.to_fully_qualified();
165 writeln!(out, "template <> struct hash<{}> {{", qualified);
166 writeln!(
167 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800168 " ::std::size_t operator()(const {} &self) const noexcept {{",
David Tolnay7da38202020-11-27 17:36:16 -0800169 qualified,
170 );
171 let link_name = mangle::operator(&strct.name, "hash");
172 write!(out, " return ::");
173 for name in &strct.name.namespace {
174 write!(out, "{}::", name);
175 }
176 writeln!(out, "{}(self);", link_name);
177 writeln!(out, " }}");
178 writeln!(out, "}};");
179 }
180 }
181 }
182
183 out.end_block(Block::Namespace("std"));
184}
185
David Tolnaydfb82d72020-11-02 00:10:10 -0800186fn pick_includes_and_builtins(out: &mut OutFile, apis: &[Api]) {
187 for api in apis {
188 if let Api::Include(include) = api {
189 out.include.insert(include);
190 }
191 }
192
David Tolnaya7c2ea12020-10-30 21:32:53 -0700193 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -0400194 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700195 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnay89e386d2020-10-03 19:02:19 -0700196 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
197 | Some(I64) => out.include.cstdint = true,
198 Some(Usize) => out.include.cstddef = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800199 Some(Isize) => out.builtin.rust_isize = true,
David Tolnay89e386d2020-10-03 19:02:19 -0700200 Some(CxxString) => out.include.string = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800201 Some(RustString) => out.builtin.rust_string = true,
David Tolnayb3873dc2020-11-25 19:47:49 -0800202 Some(Bool) | Some(Char) | Some(F32) | Some(F64) | None => {}
David Tolnay89e386d2020-10-03 19:02:19 -0700203 },
David Tolnaydcfa8e92020-11-02 09:50:06 -0800204 Type::RustBox(_) => out.builtin.rust_box = true,
205 Type::RustVec(_) => out.builtin.rust_vec = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700206 Type::UniquePtr(_) => out.include.memory = true,
David Tolnayb3b24a12020-12-01 15:27:43 -0800207 Type::SharedPtr(_) => out.include.memory = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800208 Type::Str(_) => out.builtin.rust_str = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700209 Type::CxxVector(_) => out.include.vector = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800210 Type::Fn(_) => out.builtin.rust_fn = true,
David Tolnay5515a9e2020-11-25 19:07:54 -0800211 Type::SliceRef(_) => out.builtin.rust_slice = true,
David Tolnaye8b1bb42020-11-24 20:37:37 -0800212 Type::Array(_) => out.include.array = true,
David Tolnay11bd7ff2020-11-01 19:44:58 -0800213 Type::Ref(_) | Type::Void(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -0400214 }
215 }
David Tolnayec66d112020-10-31 21:00:26 -0700216}
David Tolnayf51447e2020-03-06 14:14:27 -0800217
David Tolnay102c7ea2020-11-08 18:58:09 -0800218fn write_struct<'a>(out: &mut OutFile<'a>, strct: &'a Struct, methods: &[&ExternFn]) {
David Tolnayb960ed22020-11-27 14:34:30 -0800219 let operator_eq = derive::contains(&strct.derives, Trait::PartialEq);
David Tolnay84389352020-11-27 17:12:10 -0800220 let operator_ord = derive::contains(&strct.derives, Trait::PartialOrd);
David Tolnayb960ed22020-11-27 14:34:30 -0800221
David Tolnay17a934c2020-11-02 00:40:04 -0800222 out.set_namespace(&strct.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800223 let guard = format!("CXXBRIDGE1_STRUCT_{}", strct.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700224 writeln!(out, "#ifndef {}", guard);
225 writeln!(out, "#define {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400226 for line in strct.doc.to_string().lines() {
227 writeln!(out, "//{}", line);
228 }
David Tolnay17a934c2020-11-02 00:40:04 -0800229 writeln!(out, "struct {} final {{", strct.name.cxx);
David Tolnay84389352020-11-27 17:12:10 -0800230
David Tolnay7db73692019-10-20 14:51:12 -0400231 for field in &strct.fields {
232 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700233 write_type_space(out, &field.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400234 writeln!(out, "{};", field.ident);
235 }
David Tolnay84389352020-11-27 17:12:10 -0800236
David Tolnay5554ebd2020-12-10 11:34:41 -0800237 writeln!(out);
David Tolnay84389352020-11-27 17:12:10 -0800238
David Tolnay102c7ea2020-11-08 18:58:09 -0800239 for method in methods {
240 write!(out, " ");
241 let sig = &method.sig;
242 let local_name = method.name.cxx.to_string();
243 write_rust_function_shim_decl(out, &local_name, sig, false);
244 writeln!(out, ";");
245 }
David Tolnay84389352020-11-27 17:12:10 -0800246
David Tolnayb960ed22020-11-27 14:34:30 -0800247 if operator_eq {
248 writeln!(
249 out,
250 " bool operator==(const {} &) const noexcept;",
251 strct.name.cxx,
252 );
253 writeln!(
254 out,
255 " bool operator!=(const {} &) const noexcept;",
256 strct.name.cxx,
257 );
258 }
David Tolnay84389352020-11-27 17:12:10 -0800259
260 if operator_ord {
261 writeln!(
262 out,
263 " bool operator<(const {} &) const noexcept;",
264 strct.name.cxx,
265 );
266 writeln!(
267 out,
268 " bool operator<=(const {} &) const noexcept;",
269 strct.name.cxx,
270 );
271 writeln!(
272 out,
273 " bool operator>(const {} &) const noexcept;",
274 strct.name.cxx,
275 );
276 writeln!(
277 out,
278 " bool operator>=(const {} &) const noexcept;",
279 strct.name.cxx,
280 );
281 }
282
David Tolnay5554ebd2020-12-10 11:34:41 -0800283 out.include.type_traits = true;
284 writeln!(out, " using IsRelocatable = ::std::true_type;");
285
David Tolnay7db73692019-10-20 14:51:12 -0400286 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700287 writeln!(out, "#endif // {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400288}
289
290fn write_struct_decl(out: &mut OutFile, ident: &Ident) {
291 writeln!(out, "struct {};", ident);
292}
293
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800294fn write_enum_decl(out: &mut OutFile, enm: &Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800295 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800296 write_atom(out, enm.repr);
297 writeln!(out, ";");
298}
299
David Tolnay8faec772020-11-02 00:18:19 -0800300fn write_struct_using(out: &mut OutFile, ident: &Pair) {
301 writeln!(out, "using {} = {};", ident.cxx, ident.to_fully_qualified());
David Tolnay8861bee2020-01-20 18:39:24 -0800302}
303
David Tolnay0b9b9f82020-11-01 20:41:00 -0800304fn write_struct_with_methods<'a>(
305 out: &mut OutFile<'a>,
306 ety: &'a ExternType,
307 methods: &[&ExternFn],
308) {
David Tolnay17a934c2020-11-02 00:40:04 -0800309 out.set_namespace(&ety.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800310 let guard = format!("CXXBRIDGE1_STRUCT_{}", ety.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700311 writeln!(out, "#ifndef {}", guard);
312 writeln!(out, "#define {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700313 for line in ety.doc.to_string().lines() {
314 writeln!(out, "//{}", line);
315 }
David Tolnay365fc7c2020-11-25 16:08:13 -0800316 out.builtin.opaque = true;
David Tolnay7c06b862020-11-25 16:59:09 -0800317 writeln!(
318 out,
319 "struct {} final : public ::rust::Opaque {{",
320 ety.name.cxx,
321 );
Joel Galenson968738f2020-04-15 14:19:33 -0700322 for method in methods {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700323 write!(out, " ");
324 let sig = &method.sig;
David Tolnay17a934c2020-11-02 00:40:04 -0800325 let local_name = method.name.cxx.to_string();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700326 write_rust_function_shim_decl(out, &local_name, sig, false);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700327 writeln!(out, ";");
328 }
329 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700330 writeln!(out, "#endif // {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700331}
332
David Tolnay0b9b9f82020-11-01 20:41:00 -0800333fn write_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800334 out.set_namespace(&enm.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800335 let guard = format!("CXXBRIDGE1_ENUM_{}", enm.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700336 writeln!(out, "#ifndef {}", guard);
337 writeln!(out, "#define {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700338 for line in enm.doc.to_string().lines() {
339 writeln!(out, "//{}", line);
340 }
David Tolnay17a934c2020-11-02 00:40:04 -0800341 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700342 write_atom(out, enm.repr);
343 writeln!(out, " {{");
Joel Galensonc03402a2020-04-23 17:31:09 -0700344 for variant in &enm.variants {
David Tolnaye6f62142020-12-21 16:00:41 -0800345 writeln!(out, " {} = {},", variant.name.cxx, variant.discriminant);
Joel Galensonc03402a2020-04-23 17:31:09 -0700346 }
347 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700348 writeln!(out, "#endif // {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700349}
350
David Tolnay0b9b9f82020-11-01 20:41:00 -0800351fn check_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800352 out.set_namespace(&enm.name.namespace);
David Tolnay06711bc2020-11-19 19:25:14 -0800353 out.include.type_traits = true;
354 writeln!(
355 out,
356 "static_assert(::std::is_enum<{}>::value, \"expected enum\");",
357 enm.name.cxx,
358 );
David Tolnay17a934c2020-11-02 00:40:04 -0800359 write!(out, "static_assert(sizeof({}) == sizeof(", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700360 write_atom(out, enm.repr);
361 writeln!(out, "), \"incorrect size\");");
Joel Galenson0f654ff2020-05-04 20:04:21 -0700362 for variant in &enm.variants {
David Tolnayf6a89f22020-05-10 23:39:27 -0700363 write!(out, "static_assert(static_cast<");
364 write_atom(out, enm.repr);
Joel Galenson0f654ff2020-05-04 20:04:21 -0700365 writeln!(
366 out,
David Tolnayf6a89f22020-05-10 23:39:27 -0700367 ">({}::{}) == {}, \"disagrees with the value in #[cxx::bridge]\");",
David Tolnaye6f62142020-12-21 16:00:41 -0800368 enm.name.cxx, variant.name.cxx, variant.discriminant,
Joel Galenson0f654ff2020-05-04 20:04:21 -0700369 );
Joel Galenson0f654ff2020-05-04 20:04:21 -0700370 }
Joel Galenson905eb2e2020-05-04 14:58:14 -0700371}
372
David Tolnay5b1d8632020-12-20 22:05:11 -0800373fn check_trivial_extern_type(out: &mut OutFile, alias: &TypeAlias, reasons: &[TrivialReason]) {
David Tolnay174bd952020-11-02 09:23:12 -0800374 // NOTE: The following static assertion is just nice-to-have and not
David Tolnayfd0034e2020-10-04 13:15:34 -0700375 // necessary for soundness. That's because triviality is always declared by
376 // the user in the form of an unsafe impl of cxx::ExternType:
377 //
378 // unsafe impl ExternType for MyType {
379 // type Id = cxx::type_id!("...");
380 // type Kind = cxx::kind::Trivial;
381 // }
382 //
383 // Since the user went on the record with their unsafe impl to unsafely
384 // claim they KNOW that the type is trivial, it's fine for that to be on
David Tolnay174bd952020-11-02 09:23:12 -0800385 // them if that were wrong. However, in practice correctly reasoning about
386 // the relocatability of C++ types is challenging, particularly if the type
387 // definition were to change over time, so for now we add this check.
David Tolnayfd0034e2020-10-04 13:15:34 -0700388 //
David Tolnay174bd952020-11-02 09:23:12 -0800389 // There may be legitimate reasons to opt out of this assertion for support
390 // of types that the programmer knows are soundly Rust-movable despite not
391 // being recognized as such by the C++ type system due to a move constructor
392 // or destructor. To opt out of the relocatability check, they need to do
393 // one of the following things in any header used by `include!` in their
394 // bridge.
395 //
396 // --- if they define the type:
397 // struct MyType {
398 // ...
399 // + using IsRelocatable = std::true_type;
400 // };
401 //
402 // --- otherwise:
403 // + template <>
404 // + struct rust::IsRelocatable<MyType> : std::true_type {};
405 //
David Tolnayfd0034e2020-10-04 13:15:34 -0700406
David Tolnay5b1d8632020-12-20 22:05:11 -0800407 let id = alias.name.to_fully_qualified();
David Tolnay174bd952020-11-02 09:23:12 -0800408 out.builtin.relocatable = true;
David Tolnay5b1d8632020-12-20 22:05:11 -0800409 writeln!(out, "static_assert(");
410 writeln!(out, " ::rust::IsRelocatable<{}>::value,", id);
411 writeln!(
412 out,
413 " \"type {} should be trivially move constructible and trivially destructible in C++ to be used as {} in Rust\");",
414 id.trim_start_matches("::"),
415 trivial::as_what(&alias.name, reasons),
416 );
Adrian Taylor405e8742020-09-25 14:01:25 -0700417}
418
David Tolnayb960ed22020-11-27 14:34:30 -0800419fn write_struct_operator_decls<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
420 out.set_namespace(&strct.name.namespace);
421 out.begin_block(Block::ExternC);
422
423 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800424 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800425 writeln!(
426 out,
427 "bool {}(const {1} &, const {1} &) noexcept;",
428 link_name, strct.name.cxx,
429 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800430
431 if !derive::contains(&strct.derives, Trait::Eq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800432 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800433 writeln!(
434 out,
435 "bool {}(const {1} &, const {1} &) noexcept;",
436 link_name, strct.name.cxx,
437 );
438 }
David Tolnayb960ed22020-11-27 14:34:30 -0800439 }
440
David Tolnay84389352020-11-27 17:12:10 -0800441 if derive::contains(&strct.derives, Trait::PartialOrd) {
David Tolnaya05f9402020-11-27 17:44:05 -0800442 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800443 writeln!(
444 out,
445 "bool {}(const {1} &, const {1} &) noexcept;",
446 link_name, strct.name.cxx,
447 );
448
David Tolnaya05f9402020-11-27 17:44:05 -0800449 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800450 writeln!(
451 out,
452 "bool {}(const {1} &, const {1} &) noexcept;",
453 link_name, strct.name.cxx,
454 );
455
456 if !derive::contains(&strct.derives, Trait::Ord) {
David Tolnaya05f9402020-11-27 17:44:05 -0800457 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800458 writeln!(
459 out,
460 "bool {}(const {1} &, const {1} &) noexcept;",
461 link_name, strct.name.cxx,
462 );
463
David Tolnaya05f9402020-11-27 17:44:05 -0800464 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800465 writeln!(
466 out,
467 "bool {}(const {1} &, const {1} &) noexcept;",
468 link_name, strct.name.cxx,
469 );
470 }
471 }
472
David Tolnay7da38202020-11-27 17:36:16 -0800473 if derive::contains(&strct.derives, Trait::Hash) {
David Tolnay12320e12020-12-09 23:09:36 -0800474 out.include.cstddef = true;
David Tolnay7da38202020-11-27 17:36:16 -0800475 let link_name = mangle::operator(&strct.name, "hash");
476 writeln!(
477 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800478 "::std::size_t {}(const {} &) noexcept;",
David Tolnay7da38202020-11-27 17:36:16 -0800479 link_name, strct.name.cxx,
480 );
481 }
482
David Tolnayb960ed22020-11-27 14:34:30 -0800483 out.end_block(Block::ExternC);
484}
485
486fn write_struct_operators<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
487 if out.header {
488 return;
489 }
490
491 out.set_namespace(&strct.name.namespace);
492
493 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnayb960ed22020-11-27 14:34:30 -0800494 out.next_section();
495 writeln!(
496 out,
497 "bool {0}::operator==(const {0} &rhs) const noexcept {{",
498 strct.name.cxx,
499 );
David Tolnaya05f9402020-11-27 17:44:05 -0800500 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800501 writeln!(out, " return {}(*this, rhs);", link_name);
502 writeln!(out, "}}");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800503
David Tolnayb960ed22020-11-27 14:34:30 -0800504 out.next_section();
505 writeln!(
506 out,
507 "bool {0}::operator!=(const {0} &rhs) const noexcept {{",
508 strct.name.cxx,
509 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800510 if derive::contains(&strct.derives, Trait::Eq) {
511 writeln!(out, " return !(*this == rhs);");
512 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800513 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800514 writeln!(out, " return {}(*this, rhs);", link_name);
515 }
David Tolnayb960ed22020-11-27 14:34:30 -0800516 writeln!(out, "}}");
517 }
David Tolnay84389352020-11-27 17:12:10 -0800518
519 if derive::contains(&strct.derives, Trait::PartialOrd) {
520 out.next_section();
521 writeln!(
522 out,
523 "bool {0}::operator<(const {0} &rhs) const noexcept {{",
524 strct.name.cxx,
525 );
David Tolnaya05f9402020-11-27 17:44:05 -0800526 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800527 writeln!(out, " return {}(*this, rhs);", link_name);
528 writeln!(out, "}}");
529
530 out.next_section();
531 writeln!(
532 out,
533 "bool {0}::operator<=(const {0} &rhs) const noexcept {{",
534 strct.name.cxx,
535 );
David Tolnaya05f9402020-11-27 17:44:05 -0800536 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800537 writeln!(out, " return {}(*this, rhs);", link_name);
538 writeln!(out, "}}");
539
540 out.next_section();
541 writeln!(
542 out,
543 "bool {0}::operator>(const {0} &rhs) const noexcept {{",
544 strct.name.cxx,
545 );
546 if derive::contains(&strct.derives, Trait::Ord) {
547 writeln!(out, " return !(*this <= rhs);");
548 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800549 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800550 writeln!(out, " return {}(*this, rhs);", link_name);
551 }
552 writeln!(out, "}}");
553
554 out.next_section();
555 writeln!(
556 out,
557 "bool {0}::operator>=(const {0} &rhs) const noexcept {{",
558 strct.name.cxx,
559 );
560 if derive::contains(&strct.derives, Trait::Ord) {
561 writeln!(out, " return !(*this < rhs);");
562 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800563 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800564 writeln!(out, " return {}(*this, rhs);", link_name);
565 }
566 writeln!(out, "}}");
567 }
David Tolnayb960ed22020-11-27 14:34:30 -0800568}
569
David Tolnay0b9b9f82020-11-01 20:41:00 -0800570fn write_cxx_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay04770742020-11-01 13:50:50 -0800571 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800572 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800573 out.begin_block(Block::ExternC);
David Tolnaye1476af2020-11-01 13:47:25 -0800574 if let Some(annotation) = &out.opt.cxx_impl_annotations {
David Tolnaycc1ae762020-10-31 15:53:50 -0700575 write!(out, "{} ", annotation);
Adrian Taylor21f0ff02020-07-21 16:21:48 -0700576 }
David Tolnayebef4a22020-03-17 15:33:47 -0700577 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700578 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700579 write!(out, "::rust::repr::PtrLen ");
David Tolnayebef4a22020-03-17 15:33:47 -0700580 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700581 write_extern_return_type_space(out, &efn.ret);
David Tolnayebef4a22020-03-17 15:33:47 -0700582 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700583 let mangled = mangle::extern_fn(efn, out.types);
David Tolnay3caa50a2020-04-19 21:25:34 -0700584 write!(out, "{}(", mangled);
David Tolnaye439c772020-04-20 00:23:55 -0700585 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800586 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700587 write!(out, "const ");
588 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700589 write!(
590 out,
591 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800592 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700593 );
Joel Galenson3d4f6122020-04-07 15:54:05 -0700594 }
David Tolnay7db73692019-10-20 14:51:12 -0400595 for (i, arg) in efn.args.iter().enumerate() {
Joel Galenson3d4f6122020-04-07 15:54:05 -0700596 if i > 0 || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400597 write!(out, ", ");
598 }
David Tolnaya46a2372020-03-06 10:03:48 -0800599 if arg.ty == RustString {
600 write!(out, "const ");
David Tolnay313b10e2020-04-25 16:30:51 -0700601 } else if let Type::RustVec(_) = arg.ty {
602 write!(out, "const ");
David Tolnaya46a2372020-03-06 10:03:48 -0800603 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700604 write_extern_arg(out, arg);
David Tolnay7db73692019-10-20 14:51:12 -0400605 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700606 let indirect_return = indirect_return(efn, out.types);
David Tolnay7db73692019-10-20 14:51:12 -0400607 if indirect_return {
myronahne3b78ea2020-05-23 01:08:13 +0700608 if !efn.args.is_empty() || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400609 write!(out, ", ");
610 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700611 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400612 write!(out, "*return$");
613 }
614 writeln!(out, ") noexcept {{");
615 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700616 write_return_type(out, &efn.ret);
Joel Galenson3d4f6122020-04-07 15:54:05 -0700617 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800618 None => write!(out, "(*{}$)(", efn.name.rust),
Adrian Taylorc8713432020-10-21 18:20:55 -0700619 Some(receiver) => write!(
620 out,
621 "({}::*{}$)(",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700622 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800623 efn.name.rust,
Adrian Taylorc8713432020-10-21 18:20:55 -0700624 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700625 }
David Tolnay7db73692019-10-20 14:51:12 -0400626 for (i, arg) in efn.args.iter().enumerate() {
627 if i > 0 {
628 write!(out, ", ");
629 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700630 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400631 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700632 write!(out, ")");
David Tolnay4e7123f2020-04-19 21:11:37 -0700633 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800634 if !receiver.mutable {
David Tolnay4e7123f2020-04-19 21:11:37 -0700635 write!(out, " const");
636 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700637 }
638 write!(out, " = ");
639 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800640 None => write!(out, "{}", efn.name.to_fully_qualified()),
Adrian Taylorc8713432020-10-21 18:20:55 -0700641 Some(receiver) => write!(
642 out,
643 "&{}::{}",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700644 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800645 efn.name.cxx,
Adrian Taylorc8713432020-10-21 18:20:55 -0700646 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700647 }
648 writeln!(out, ";");
David Tolnay7db73692019-10-20 14:51:12 -0400649 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700650 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700651 out.builtin.ptr_len = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700652 out.builtin.trycatch = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700653 writeln!(out, "::rust::repr::PtrLen throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700654 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700655 writeln!(out, " [&] {{");
656 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700657 }
David Tolnay7db73692019-10-20 14:51:12 -0400658 if indirect_return {
David Tolnay0ecd05a2020-07-29 16:32:03 -0700659 out.include.new = true;
David Tolnay7db73692019-10-20 14:51:12 -0400660 write!(out, "new (return$) ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700661 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400662 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700663 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400664 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700665 }
666 match &efn.ret {
667 Some(Type::Ref(_)) => write!(out, "&"),
David Tolnay74d6d512020-10-31 22:22:03 -0700668 Some(Type::Str(_)) if !indirect_return => {
669 out.builtin.rust_str_repr = true;
670 write!(out, "::rust::impl<::rust::Str>::repr(");
671 }
David Tolnay73b72642020-11-25 17:44:05 -0800672 Some(ty @ Type::SliceRef(_)) if !indirect_return => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700673 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800674 write!(out, "::rust::impl<");
675 write_type(out, ty);
676 write!(out, ">::repr(");
David Tolnayeb952ba2020-04-14 15:02:24 -0700677 }
David Tolnay99642622020-03-25 13:07:35 -0700678 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400679 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700680 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800681 None => write!(out, "{}$(", efn.name.rust),
682 Some(_) => write!(out, "(self.*{}$)(", efn.name.rust),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700683 }
David Tolnay7db73692019-10-20 14:51:12 -0400684 for (i, arg) in efn.args.iter().enumerate() {
685 if i > 0 {
686 write!(out, ", ");
687 }
688 if let Type::RustBox(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700689 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400690 write!(out, "::from_raw({})", arg.ident);
691 } else if let Type::UniquePtr(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700692 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400693 write!(out, "({})", arg.ident);
David Tolnay0356d332020-10-31 19:46:41 -0700694 } else if let Type::Str(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700695 out.builtin.rust_str_new_unchecked = true;
David Tolnay0356d332020-10-31 19:46:41 -0700696 write!(
697 out,
698 "::rust::impl<::rust::Str>::new_unchecked({})",
699 arg.ident,
700 );
David Tolnaya46a2372020-03-06 10:03:48 -0800701 } else if arg.ty == RustString {
David Tolnay74d6d512020-10-31 22:22:03 -0700702 out.builtin.unsafe_bitcopy = true;
David Tolnaycc3767f2020-03-06 10:41:51 -0800703 write!(
704 out,
705 "::rust::String(::rust::unsafe_bitcopy, *{})",
706 arg.ident,
707 );
David Tolnay313b10e2020-04-25 16:30:51 -0700708 } else if let Type::RustVec(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700709 out.builtin.unsafe_bitcopy = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700710 write_type(out, &arg.ty);
David Tolnay313b10e2020-04-25 16:30:51 -0700711 write!(out, "(::rust::unsafe_bitcopy, *{})", arg.ident);
David Tolnay73b72642020-11-25 17:44:05 -0800712 } else if let Type::SliceRef(slice) = &arg.ty {
David Tolnayc5629f02020-11-23 18:32:46 -0800713 write_type(out, &arg.ty);
714 write!(out, "(static_cast<");
715 if slice.mutability.is_none() {
716 write!(out, "const ");
717 }
David Tolnay5515a9e2020-11-25 19:07:54 -0800718 write_type_space(out, &slice.inner);
719 write!(out, "*>({0}.ptr), {0}.len)", arg.ident);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700720 } else if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700721 out.include.utility = true;
David Tolnay7e219b82020-03-01 13:14:51 -0800722 write!(out, "::std::move(*{})", arg.ident);
David Tolnay7db73692019-10-20 14:51:12 -0400723 } else {
724 write!(out, "{}", arg.ident);
725 }
726 }
727 write!(out, ")");
728 match &efn.ret {
729 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
730 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800731 Some(Type::Str(_)) | Some(Type::SliceRef(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400732 _ => {}
733 }
734 if indirect_return {
735 write!(out, ")");
736 }
737 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700738 if efn.throws {
739 out.include.cstring = true;
David Tolnay1f010c62020-11-01 20:27:46 -0800740 out.builtin.exception = true;
David Tolnay5d121442020-03-17 22:14:40 -0700741 writeln!(out, " throw$.ptr = nullptr;");
742 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700743 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700744 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700745 writeln!(
746 out,
David Tolnayc5629f02020-11-23 18:32:46 -0800747 " throw$.ptr = const_cast<char *>(::cxxbridge1$exception(catch$, throw$.len));",
David Tolnayebef4a22020-03-17 15:33:47 -0700748 );
David Tolnay5d121442020-03-17 22:14:40 -0700749 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700750 writeln!(out, " return throw$;");
751 }
David Tolnay7db73692019-10-20 14:51:12 -0400752 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700753 for arg in &efn.args {
754 if let Type::Fn(f) = &arg.ty {
755 let var = &arg.ident;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700756 write_function_pointer_trampoline(out, efn, var, f);
David Tolnay75dca2e2020-03-25 20:17:52 -0700757 }
758 }
David Tolnayca563ee2020-11-01 20:12:27 -0800759 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700760}
761
762fn write_function_pointer_trampoline(
763 out: &mut OutFile,
764 efn: &ExternFn,
765 var: &Ident,
766 f: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700767) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700768 let r_trampoline = mangle::r_trampoline(efn, var, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700769 let indirect_call = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700770 write_rust_function_decl_impl(out, &r_trampoline, f, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700771
772 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700773 let c_trampoline = mangle::c_trampoline(efn, var, out.types).to_string();
774 write_rust_function_shim_impl(out, &c_trampoline, f, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400775}
776
David Tolnay0b9b9f82020-11-01 20:41:00 -0800777fn write_rust_function_decl<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800778 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800779 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700780 let link_name = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700781 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700782 write_rust_function_decl_impl(out, &link_name, efn, indirect_call);
David Tolnayca563ee2020-11-01 20:12:27 -0800783 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700784}
785
786fn write_rust_function_decl_impl(
787 out: &mut OutFile,
David Tolnay891061b2020-04-19 22:42:33 -0700788 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700789 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700790 indirect_call: bool,
791) {
David Tolnay04770742020-11-01 13:50:50 -0800792 out.next_section();
David Tolnay75dca2e2020-03-25 20:17:52 -0700793 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700794 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700795 write!(out, "::rust::repr::PtrLen ");
David Tolnay1e548172020-03-16 13:37:09 -0700796 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700797 write_extern_return_type_space(out, &sig.ret);
David Tolnay1e548172020-03-16 13:37:09 -0700798 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700799 write!(out, "{}(", link_name);
800 let mut needs_comma = false;
David Tolnaye439c772020-04-20 00:23:55 -0700801 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800802 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700803 write!(out, "const ");
804 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700805 write!(
806 out,
807 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800808 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700809 );
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700810 needs_comma = true;
811 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700812 for arg in &sig.args {
813 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400814 write!(out, ", ");
815 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700816 write_extern_arg(out, arg);
David Tolnay75dca2e2020-03-25 20:17:52 -0700817 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400818 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700819 if indirect_return(sig, out.types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700820 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400821 write!(out, ", ");
822 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700823 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400824 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700825 needs_comma = true;
826 }
827 if indirect_call {
828 if needs_comma {
829 write!(out, ", ");
830 }
831 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400832 }
833 writeln!(out, ") noexcept;");
834}
835
David Tolnay0b9b9f82020-11-01 20:41:00 -0800836fn write_rust_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800837 out.set_namespace(&efn.name.namespace);
David Tolnay7db73692019-10-20 14:51:12 -0400838 for line in efn.doc.to_string().lines() {
839 writeln!(out, "//{}", line);
840 }
David Tolnaya73853b2020-04-20 01:19:56 -0700841 let local_name = match &efn.sig.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800842 None => efn.name.cxx.to_string(),
843 Some(receiver) => format!("{}::{}", out.types.resolve(&receiver.ty).cxx, efn.name.cxx),
David Tolnaya73853b2020-04-20 01:19:56 -0700844 };
David Tolnaya7c2ea12020-10-30 21:32:53 -0700845 let invoke = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700846 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700847 write_rust_function_shim_impl(out, &local_name, efn, &invoke, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700848}
849
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700850fn write_rust_function_shim_decl(
David Tolnay75dca2e2020-03-25 20:17:52 -0700851 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700852 local_name: &str,
David Tolnay75dca2e2020-03-25 20:17:52 -0700853 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700854 indirect_call: bool,
855) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700856 write_return_type(out, &sig.ret);
David Tolnay75dca2e2020-03-25 20:17:52 -0700857 write!(out, "{}(", local_name);
858 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400859 if i > 0 {
860 write!(out, ", ");
861 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700862 write_type_space(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400863 write!(out, "{}", arg.ident);
864 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700865 if indirect_call {
866 if !sig.args.is_empty() {
867 write!(out, ", ");
868 }
869 write!(out, "void *extern$");
870 }
David Tolnay1e548172020-03-16 13:37:09 -0700871 write!(out, ")");
David Tolnay86710612020-04-20 00:30:32 -0700872 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800873 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700874 write!(out, " const");
875 }
876 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700877 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700878 write!(out, " noexcept");
879 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700880}
881
882fn write_rust_function_shim_impl(
883 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700884 local_name: &str,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700885 sig: &Signature,
David Tolnay891061b2020-04-19 22:42:33 -0700886 invoke: &Symbol,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700887 indirect_call: bool,
888) {
889 if out.header && sig.receiver.is_some() {
890 // We've already defined this inside the struct.
891 return;
892 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700893 write_rust_function_shim_decl(out, local_name, sig, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400894 if out.header {
895 writeln!(out, ";");
David Tolnay439cde22020-04-20 00:46:25 -0700896 return;
David Tolnay7db73692019-10-20 14:51:12 -0400897 }
David Tolnay439cde22020-04-20 00:46:25 -0700898 writeln!(out, " {{");
899 for arg in &sig.args {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700900 if arg.ty != RustString && out.types.needs_indirect_abi(&arg.ty) {
David Tolnay439cde22020-04-20 00:46:25 -0700901 out.include.utility = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700902 out.builtin.manually_drop = true;
David Tolnay439cde22020-04-20 00:46:25 -0700903 write!(out, " ::rust::ManuallyDrop<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700904 write_type(out, &arg.ty);
David Tolnay439cde22020-04-20 00:46:25 -0700905 writeln!(out, "> {}$(::std::move({0}));", arg.ident);
906 }
907 }
908 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700909 let indirect_return = indirect_return(sig, out.types);
David Tolnay439cde22020-04-20 00:46:25 -0700910 if indirect_return {
David Tolnay74d6d512020-10-31 22:22:03 -0700911 out.builtin.maybe_uninit = true;
David Tolnay439cde22020-04-20 00:46:25 -0700912 write!(out, "::rust::MaybeUninit<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700913 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay439cde22020-04-20 00:46:25 -0700914 writeln!(out, "> return$;");
915 write!(out, " ");
916 } else if let Some(ret) = &sig.ret {
917 write!(out, "return ");
918 match ret {
919 Type::RustBox(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700920 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700921 write!(out, "::from_raw(");
922 }
923 Type::UniquePtr(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700924 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700925 write!(out, "(");
926 }
927 Type::Ref(_) => write!(out, "*"),
David Tolnay74d6d512020-10-31 22:22:03 -0700928 Type::Str(_) => {
929 out.builtin.rust_str_new_unchecked = true;
930 write!(out, "::rust::impl<::rust::Str>::new_unchecked(");
931 }
David Tolnay73b72642020-11-25 17:44:05 -0800932 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700933 out.builtin.rust_slice_new = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800934 write!(out, "::rust::impl<");
935 write_type(out, ret);
936 write!(out, ">::slice(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700937 }
David Tolnay439cde22020-04-20 00:46:25 -0700938 _ => {}
939 }
940 }
941 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700942 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700943 write!(out, "::rust::repr::PtrLen error$ = ");
David Tolnay439cde22020-04-20 00:46:25 -0700944 }
945 write!(out, "{}(", invoke);
David Tolnay9d840362020-11-10 08:50:40 -0800946 let mut needs_comma = false;
David Tolnay439cde22020-04-20 00:46:25 -0700947 if sig.receiver.is_some() {
948 write!(out, "*this");
David Tolnay9d840362020-11-10 08:50:40 -0800949 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700950 }
David Tolnay9d840362020-11-10 08:50:40 -0800951 for arg in &sig.args {
952 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700953 write!(out, ", ");
954 }
955 match &arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700956 Type::Str(_) => {
957 out.builtin.rust_str_repr = true;
958 write!(out, "::rust::impl<::rust::Str>::repr(");
959 }
David Tolnay73b72642020-11-25 17:44:05 -0800960 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700961 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800962 write!(out, "::rust::impl<");
963 write_type(out, &arg.ty);
964 write!(out, ">::repr(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700965 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700966 ty if out.types.needs_indirect_abi(ty) => write!(out, "&"),
David Tolnay439cde22020-04-20 00:46:25 -0700967 _ => {}
968 }
969 write!(out, "{}", arg.ident);
970 match &arg.ty {
971 Type::RustBox(_) => write!(out, ".into_raw()"),
972 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800973 Type::Str(_) | Type::SliceRef(_) => write!(out, ")"),
David Tolnaya7c2ea12020-10-30 21:32:53 -0700974 ty if ty != RustString && out.types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay439cde22020-04-20 00:46:25 -0700975 _ => {}
976 }
David Tolnay9d840362020-11-10 08:50:40 -0800977 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700978 }
979 if indirect_return {
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, "&return$.value");
David Tolnay9d840362020-11-10 08:50:40 -0800984 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700985 }
986 if indirect_call {
David Tolnay9d840362020-11-10 08:50:40 -0800987 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700988 write!(out, ", ");
989 }
990 write!(out, "extern$");
991 }
992 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -0400993 if !indirect_return {
994 if let Some(ret) = &sig.ret {
David Tolnay73b72642020-11-25 17:44:05 -0800995 if let Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::SliceRef(_) = ret {
David Tolnay22602b42020-09-21 18:04:05 -0400996 write!(out, ")");
997 }
David Tolnay439cde22020-04-20 00:46:25 -0700998 }
999 }
1000 writeln!(out, ";");
1001 if sig.throws {
David Tolnay74d6d512020-10-31 22:22:03 -07001002 out.builtin.rust_error = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001003 writeln!(out, " if (error$.ptr) {{");
David Tolnay84ddf9e2020-10-31 15:36:48 -07001004 writeln!(out, " throw ::rust::impl<::rust::Error>::error(error$);");
David Tolnay439cde22020-04-20 00:46:25 -07001005 writeln!(out, " }}");
1006 }
1007 if indirect_return {
1008 out.include.utility = true;
1009 writeln!(out, " return ::std::move(return$.value);");
1010 }
1011 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001012}
1013
David Tolnaya7c2ea12020-10-30 21:32:53 -07001014fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001015 match ty {
1016 None => write!(out, "void "),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001017 Some(ty) => write_type_space(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001018 }
1019}
1020
David Tolnay75dca2e2020-03-25 20:17:52 -07001021fn indirect_return(sig: &Signature, types: &Types) -> bool {
1022 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -07001023 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -07001024 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -07001025}
1026
David Tolnaya7c2ea12020-10-30 21:32:53 -07001027fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
David Tolnay99642622020-03-25 13:07:35 -07001028 match ty {
1029 Type::RustBox(ty) | Type::UniquePtr(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001030 write_type_space(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001031 write!(out, "*");
1032 }
1033 Type::Ref(ty) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001034 if !ty.mutable {
David Tolnay99642622020-03-25 13:07:35 -07001035 write!(out, "const ");
1036 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001037 write_type(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001038 write!(out, " *");
1039 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001040 _ => write_type(out, ty),
David Tolnay99642622020-03-25 13:07:35 -07001041 }
1042}
1043
David Tolnaya7c2ea12020-10-30 21:32:53 -07001044fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
1045 write_indirect_return_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001046 match ty {
1047 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
David Tolnay73b72642020-11-25 17:44:05 -08001048 Type::Str(_) | Type::SliceRef(_) => write!(out, " "),
David Tolnay99642622020-03-25 13:07:35 -07001049 _ => write_space_after_type(out, ty),
1050 }
1051}
1052
David Tolnaya7c2ea12020-10-30 21:32:53 -07001053fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001054 match ty {
1055 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001056 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001057 write!(out, "*");
1058 }
David Tolnay4a441222020-01-25 16:24:27 -08001059 Some(Type::Ref(ty)) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001060 if !ty.mutable {
David Tolnay4a441222020-01-25 16:24:27 -08001061 write!(out, "const ");
1062 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001063 write_type(out, &ty.inner);
David Tolnay4a441222020-01-25 16:24:27 -08001064 write!(out, " *");
1065 }
David Tolnay73b72642020-11-25 17:44:05 -08001066 Some(Type::Str(_)) | Some(Type::SliceRef(_)) => {
David Tolnay919085c2020-10-31 22:32:22 -07001067 out.builtin.ptr_len = true;
1068 write!(out, "::rust::repr::PtrLen ");
1069 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001070 Some(ty) if out.types.needs_indirect_abi(ty) => write!(out, "void "),
1071 _ => write_return_type(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001072 }
1073}
1074
David Tolnaya7c2ea12020-10-30 21:32:53 -07001075fn write_extern_arg(out: &mut OutFile, arg: &Var) {
David Tolnay7db73692019-10-20 14:51:12 -04001076 match &arg.ty {
David Tolnay4377a9e2020-04-24 15:20:26 -07001077 Type::RustBox(ty) | Type::UniquePtr(ty) | Type::CxxVector(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001078 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001079 write!(out, "*");
1080 }
David Tolnay73b72642020-11-25 17:44:05 -08001081 Type::Str(_) | Type::SliceRef(_) => {
David Tolnay919085c2020-10-31 22:32:22 -07001082 out.builtin.ptr_len = true;
1083 write!(out, "::rust::repr::PtrLen ");
1084 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001085 _ => write_type_space(out, &arg.ty),
David Tolnay7db73692019-10-20 14:51:12 -04001086 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001087 if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay7db73692019-10-20 14:51:12 -04001088 write!(out, "*");
1089 }
1090 write!(out, "{}", arg.ident);
1091}
1092
David Tolnaya7c2ea12020-10-30 21:32:53 -07001093fn write_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001094 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -07001095 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnayf6a89f22020-05-10 23:39:27 -07001096 Some(atom) => write_atom(out, atom),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001097 None => write!(out, "{}", out.types.resolve(ident).to_fully_qualified()),
David Tolnay7db73692019-10-20 14:51:12 -04001098 },
1099 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -08001100 write!(out, "::rust::Box<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001101 write_type(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001102 write!(out, ">");
1103 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001104 Type::RustVec(ty) => {
1105 write!(out, "::rust::Vec<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001106 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001107 write!(out, ">");
1108 }
David Tolnay7db73692019-10-20 14:51:12 -04001109 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -08001110 write!(out, "::std::unique_ptr<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001111 write_type(out, &ptr.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001112 write!(out, ">");
1113 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001114 Type::SharedPtr(ptr) => {
1115 write!(out, "::std::shared_ptr<");
1116 write_type(out, &ptr.inner);
1117 write!(out, ">");
1118 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001119 Type::CxxVector(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +07001120 write!(out, "::std::vector<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001121 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001122 write!(out, ">");
1123 }
David Tolnay7db73692019-10-20 14:51:12 -04001124 Type::Ref(r) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001125 if !r.mutable {
David Tolnay7db73692019-10-20 14:51:12 -04001126 write!(out, "const ");
1127 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001128 write_type(out, &r.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001129 write!(out, " &");
1130 }
1131 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -08001132 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -04001133 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001134 Type::SliceRef(slice) => {
David Tolnayc5629f02020-11-23 18:32:46 -08001135 write!(out, "::rust::Slice<");
David Tolnay5515a9e2020-11-25 19:07:54 -08001136 if slice.mutability.is_none() {
David Tolnayc5629f02020-11-23 18:32:46 -08001137 write!(out, "const ");
1138 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001139 write_type(out, &slice.inner);
1140 write!(out, ">");
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001141 }
David Tolnay75dca2e2020-03-25 20:17:52 -07001142 Type::Fn(f) => {
David Tolnayf031c322020-11-29 19:41:33 -08001143 write!(out, "::rust::Fn<");
David Tolnay75dca2e2020-03-25 20:17:52 -07001144 match &f.ret {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001145 Some(ret) => write_type(out, ret),
David Tolnay75dca2e2020-03-25 20:17:52 -07001146 None => write!(out, "void"),
1147 }
1148 write!(out, "(");
1149 for (i, arg) in f.args.iter().enumerate() {
1150 if i > 0 {
1151 write!(out, ", ");
1152 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001153 write_type(out, &arg.ty);
David Tolnay75dca2e2020-03-25 20:17:52 -07001154 }
1155 write!(out, ")>");
1156 }
Xiangpeng Hao78762352020-11-12 10:24:18 +08001157 Type::Array(a) => {
1158 write!(out, "::std::array<");
1159 write_type(out, &a.inner);
1160 write!(out, ", {}>", &a.len);
1161 }
David Tolnay2fb14e92020-03-15 23:11:38 -07001162 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001163 }
1164}
1165
David Tolnayf6a89f22020-05-10 23:39:27 -07001166fn write_atom(out: &mut OutFile, atom: Atom) {
1167 match atom {
1168 Bool => write!(out, "bool"),
David Tolnayb3873dc2020-11-25 19:47:49 -08001169 Char => write!(out, "char"),
David Tolnaybe3cbf72020-12-12 22:12:07 -08001170 U8 => write!(out, "::std::uint8_t"),
1171 U16 => write!(out, "::std::uint16_t"),
1172 U32 => write!(out, "::std::uint32_t"),
1173 U64 => write!(out, "::std::uint64_t"),
1174 Usize => write!(out, "::std::size_t"),
1175 I8 => write!(out, "::std::int8_t"),
1176 I16 => write!(out, "::std::int16_t"),
1177 I32 => write!(out, "::std::int32_t"),
1178 I64 => write!(out, "::std::int64_t"),
David Tolnayf6a89f22020-05-10 23:39:27 -07001179 Isize => write!(out, "::rust::isize"),
1180 F32 => write!(out, "float"),
1181 F64 => write!(out, "double"),
1182 CxxString => write!(out, "::std::string"),
1183 RustString => write!(out, "::rust::String"),
1184 }
1185}
1186
David Tolnaya7c2ea12020-10-30 21:32:53 -07001187fn write_type_space(out: &mut OutFile, ty: &Type) {
1188 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001189 write_space_after_type(out, ty);
1190}
1191
1192fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001193 match ty {
David Tolnayeb952ba2020-04-14 15:02:24 -07001194 Type::Ident(_)
1195 | Type::RustBox(_)
1196 | Type::UniquePtr(_)
David Tolnayb3b24a12020-12-01 15:27:43 -08001197 | Type::SharedPtr(_)
David Tolnayeb952ba2020-04-14 15:02:24 -07001198 | Type::Str(_)
David Tolnay4377a9e2020-04-24 15:20:26 -07001199 | Type::CxxVector(_)
Myron Ahneba35cf2020-02-05 19:41:51 +07001200 | Type::RustVec(_)
David Tolnay73b72642020-11-25 17:44:05 -08001201 | Type::SliceRef(_)
Xiangpeng Hao78762352020-11-12 10:24:18 +08001202 | Type::Fn(_)
1203 | Type::Array(_) => write!(out, " "),
David Tolnay7db73692019-10-20 14:51:12 -04001204 Type::Ref(_) => {}
David Tolnaye0dca7b2020-11-25 17:18:57 -08001205 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001206 }
1207}
1208
David Tolnay1bdb4712020-11-25 07:27:54 -08001209#[derive(Copy, Clone)]
1210enum UniquePtr<'a> {
David Tolnay75ea17c2020-12-06 21:08:34 -08001211 Ident(&'a RustName),
1212 CxxVector(&'a RustName),
David Tolnay1bdb4712020-11-25 07:27:54 -08001213}
1214
1215trait ToTypename {
1216 fn to_typename(&self, types: &Types) -> String;
1217}
1218
David Tolnay75ea17c2020-12-06 21:08:34 -08001219impl ToTypename for RustName {
David Tolnay1bdb4712020-11-25 07:27:54 -08001220 fn to_typename(&self, types: &Types) -> String {
1221 types.resolve(self).to_fully_qualified()
David Tolnay2eca4a02020-04-24 19:50:51 -07001222 }
1223}
1224
David Tolnay1bdb4712020-11-25 07:27:54 -08001225impl<'a> ToTypename for UniquePtr<'a> {
1226 fn to_typename(&self, types: &Types) -> String {
1227 match self {
1228 UniquePtr::Ident(ident) => ident.to_typename(types),
1229 UniquePtr::CxxVector(element) => {
1230 format!("::std::vector<{}>", element.to_typename(types))
1231 }
1232 }
1233 }
1234}
1235
1236trait ToMangled {
1237 fn to_mangled(&self, types: &Types) -> Symbol;
1238}
1239
David Tolnay75ea17c2020-12-06 21:08:34 -08001240impl ToMangled for RustName {
David Tolnay1bdb4712020-11-25 07:27:54 -08001241 fn to_mangled(&self, types: &Types) -> Symbol {
1242 self.to_symbol(types)
1243 }
1244}
1245
1246impl<'a> ToMangled for UniquePtr<'a> {
1247 fn to_mangled(&self, types: &Types) -> Symbol {
1248 match self {
1249 UniquePtr::Ident(ident) => ident.to_mangled(types),
1250 UniquePtr::CxxVector(element) => element.to_mangled(types).prefix_with("std$vector$"),
1251 }
David Tolnaybae50ef2020-04-25 12:38:41 -07001252 }
1253}
1254
David Tolnaya7c2ea12020-10-30 21:32:53 -07001255fn write_generic_instantiations(out: &mut OutFile) {
David Tolnay169bb472020-11-01 21:04:24 -08001256 if out.header {
1257 return;
1258 }
1259
1260 out.next_section();
David Tolnay078c90f2020-11-01 13:31:08 -08001261 out.set_namespace(Default::default());
David Tolnay0c033e32020-11-01 15:15:48 -08001262 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -07001263 for ty in out.types {
David Tolnayf33bc242020-12-04 18:27:02 -08001264 if let Type::RustBox(ptr) = ty {
1265 if let Type::Ident(inner) = &ptr.inner {
1266 if Atom::from(&inner.rust).is_none()
1267 && (!out.types.aliases.contains_key(&inner.rust)
1268 || out.types.explicit_impls.contains(ty))
1269 {
1270 out.next_section();
1271 write_rust_box_extern(out, &out.types.resolve(&inner));
1272 }
David Tolnay7db73692019-10-20 14:51:12 -04001273 }
David Tolnayf33bc242020-12-04 18:27:02 -08001274 } else if let Type::RustVec(vec) = ty {
1275 if let Type::Ident(inner) = &vec.inner {
1276 if Atom::from(&inner.rust).is_none()
1277 && (!out.types.aliases.contains_key(&inner.rust)
1278 || out.types.explicit_impls.contains(ty))
1279 {
David Tolnay6787be62020-04-25 11:01:02 -07001280 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001281 write_rust_vec_extern(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001282 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001283 }
David Tolnay7db73692019-10-20 14:51:12 -04001284 } else if let Type::UniquePtr(ptr) = ty {
1285 if let Type::Ident(inner) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001286 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001287 && (!out.types.aliases.contains_key(&inner.rust)
1288 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001289 {
David Tolnay7db73692019-10-20 14:51:12 -04001290 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001291 write_unique_ptr(out, inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001292 }
1293 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001294 } else if let Type::SharedPtr(ptr) = ty {
1295 if let Type::Ident(inner) = &ptr.inner {
1296 if Atom::from(&inner.rust).is_none()
David Tolnayb7453812020-12-01 21:46:55 -08001297 && (!out.types.aliases.contains_key(&inner.rust)
David Tolnayb3b24a12020-12-01 15:27:43 -08001298 || out.types.explicit_impls.contains(ty))
1299 {
1300 out.next_section();
1301 write_shared_ptr(out, inner);
1302 }
1303 }
David Tolnayf33bc242020-12-04 18:27:02 -08001304 } else if let Type::CxxVector(vector) = ty {
1305 if let Type::Ident(inner) = &vector.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001306 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001307 && (!out.types.aliases.contains_key(&inner.rust)
1308 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001309 {
Myron Ahneba35cf2020-02-05 19:41:51 +07001310 out.next_section();
David Tolnay1bdb4712020-11-25 07:27:54 -08001311 write_cxx_vector(out, inner);
David Tolnay7db73692019-10-20 14:51:12 -04001312 }
1313 }
1314 }
1315 }
David Tolnay0c033e32020-11-01 15:15:48 -08001316 out.end_block(Block::ExternC);
David Tolnay7db73692019-10-20 14:51:12 -04001317
David Tolnay0c033e32020-11-01 15:15:48 -08001318 out.begin_block(Block::Namespace("rust"));
David Tolnay0f0162f2020-11-16 23:43:37 -08001319 out.begin_block(Block::InlineNamespace("cxxbridge1"));
David Tolnaya7c2ea12020-10-30 21:32:53 -07001320 for ty in out.types {
David Tolnayf33bc242020-12-04 18:27:02 -08001321 if let Type::RustBox(ptr) = ty {
1322 if let Type::Ident(inner) = &ptr.inner {
1323 if Atom::from(&inner.rust).is_none()
1324 && (!out.types.aliases.contains_key(&inner.rust)
1325 || out.types.explicit_impls.contains(ty))
1326 {
1327 write_rust_box_impl(out, &out.types.resolve(&inner));
1328 }
David Tolnay7db73692019-10-20 14:51:12 -04001329 }
David Tolnayf33bc242020-12-04 18:27:02 -08001330 } else if let Type::RustVec(vec) = ty {
1331 if let Type::Ident(inner) = &vec.inner {
1332 if Atom::from(&inner.rust).is_none()
1333 && (!out.types.aliases.contains_key(&inner.rust)
1334 || out.types.explicit_impls.contains(ty))
1335 {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001336 write_rust_vec_impl(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001337 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001338 }
David Tolnay7db73692019-10-20 14:51:12 -04001339 }
1340 }
David Tolnay0f0162f2020-11-16 23:43:37 -08001341 out.end_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay0c033e32020-11-01 15:15:48 -08001342 out.end_block(Block::Namespace("rust"));
David Tolnay7db73692019-10-20 14:51:12 -04001343}
1344
David Tolnay8faec772020-11-02 00:18:19 -08001345fn write_rust_box_extern(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001346 let inner = ident.to_fully_qualified();
1347 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001348
David Tolnay0f0162f2020-11-16 23:43:37 -08001349 writeln!(out, "#ifndef CXXBRIDGE1_RUST_BOX_{}", instance);
1350 writeln!(out, "#define CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001351 writeln!(
1352 out,
David Tolnayc4b34222020-12-12 13:06:26 -08001353 "{} *cxxbridge1$box${}$alloc() noexcept;",
1354 inner, instance,
David Tolnay7db73692019-10-20 14:51:12 -04001355 );
1356 writeln!(
1357 out,
David Tolnay25b3c1d2020-12-12 15:50:10 -08001358 "void cxxbridge1$box${}$dealloc({} *) noexcept;",
1359 instance, inner,
1360 );
1361 writeln!(
1362 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001363 "void cxxbridge1$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001364 instance, inner,
1365 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001366 writeln!(out, "#endif // CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001367}
1368
David Tolnay75ea17c2020-12-06 21:08:34 -08001369fn write_rust_vec_extern(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001370 let inner = element.to_typename(out.types);
1371 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001372
David Tolnay12320e12020-12-09 23:09:36 -08001373 out.include.cstddef = true;
1374
David Tolnay0f0162f2020-11-16 23:43:37 -08001375 writeln!(out, "#ifndef CXXBRIDGE1_RUST_VEC_{}", instance);
1376 writeln!(out, "#define CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001377 writeln!(
1378 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001379 "void cxxbridge1$rust_vec${}$new(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnayf97c2d52020-04-25 16:37:48 -07001380 instance, inner,
1381 );
1382 writeln!(
1383 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001384 "void cxxbridge1$rust_vec${}$drop(::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001385 instance, inner,
1386 );
1387 writeln!(
1388 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001389 "::std::size_t cxxbridge1$rust_vec${}$len(const ::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001390 instance, inner,
1391 );
David Tolnay219c0792020-04-24 20:31:37 -07001392 writeln!(
1393 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001394 "::std::size_t cxxbridge1$rust_vec${}$capacity(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnaydc62d712020-12-11 13:51:53 -08001395 instance, inner,
1396 );
1397 writeln!(
1398 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001399 "const {} *cxxbridge1$rust_vec${}$data(const ::rust::Vec<{0}> *ptr) noexcept;",
David Tolnay219c0792020-04-24 20:31:37 -07001400 inner, instance,
1401 );
David Tolnay503d0192020-04-24 22:18:56 -07001402 writeln!(
1403 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001404 "void cxxbridge1$rust_vec${}$reserve_total(::rust::Vec<{}> *ptr, ::std::size_t cap) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001405 instance, inner,
1406 );
1407 writeln!(
1408 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001409 "void cxxbridge1$rust_vec${}$set_len(::rust::Vec<{}> *ptr, ::std::size_t len) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001410 instance, inner,
1411 );
1412 writeln!(
1413 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001414 "::std::size_t cxxbridge1$rust_vec${}$stride() noexcept;",
David Tolnay503d0192020-04-24 22:18:56 -07001415 instance,
1416 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001417 writeln!(out, "#endif // CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001418}
1419
David Tolnay8faec772020-11-02 00:18:19 -08001420fn write_rust_box_impl(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001421 let inner = ident.to_fully_qualified();
1422 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001423
1424 writeln!(out, "template <>");
David Tolnaye5703162020-12-12 16:26:35 -08001425 writeln!(
1426 out,
1427 "{} *Box<{}>::allocation::alloc() noexcept {{",
1428 inner, inner,
1429 );
David Tolnayc4b34222020-12-12 13:06:26 -08001430 writeln!(out, " return cxxbridge1$box${}$alloc();", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001431 writeln!(out, "}}");
1432
1433 writeln!(out, "template <>");
David Tolnay25b3c1d2020-12-12 15:50:10 -08001434 writeln!(
1435 out,
David Tolnaye5703162020-12-12 16:26:35 -08001436 "void Box<{}>::allocation::dealloc({} *ptr) noexcept {{",
David Tolnay25b3c1d2020-12-12 15:50:10 -08001437 inner, inner,
1438 );
1439 writeln!(out, " cxxbridge1$box${}$dealloc(ptr);", instance);
1440 writeln!(out, "}}");
1441
1442 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -08001443 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001444 writeln!(out, " cxxbridge1$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001445 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001446}
1447
David Tolnay75ea17c2020-12-06 21:08:34 -08001448fn write_rust_vec_impl(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001449 let inner = element.to_typename(out.types);
1450 let instance = element.to_mangled(out.types);
David Tolnay4791f1c2020-03-17 21:53:16 -07001451
David Tolnay12320e12020-12-09 23:09:36 -08001452 out.include.cstddef = true;
1453
Myron Ahneba35cf2020-02-05 19:41:51 +07001454 writeln!(out, "template <>");
David Tolnayf97c2d52020-04-25 16:37:48 -07001455 writeln!(out, "Vec<{}>::Vec() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001456 writeln!(out, " cxxbridge1$rust_vec${}$new(this);", instance);
David Tolnayf97c2d52020-04-25 16:37:48 -07001457 writeln!(out, "}}");
1458
1459 writeln!(out, "template <>");
Myron Ahneba35cf2020-02-05 19:41:51 +07001460 writeln!(out, "void Vec<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001461 writeln!(out, " return cxxbridge1$rust_vec${}$drop(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001462 writeln!(out, "}}");
1463
1464 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001465 writeln!(
1466 out,
1467 "::std::size_t Vec<{}>::size() const noexcept {{",
1468 inner,
1469 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001470 writeln!(out, " return cxxbridge1$rust_vec${}$len(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001471 writeln!(out, "}}");
David Tolnay219c0792020-04-24 20:31:37 -07001472
1473 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001474 writeln!(
1475 out,
1476 "::std::size_t Vec<{}>::capacity() const noexcept {{",
1477 inner,
1478 );
David Tolnay381d9532020-12-11 13:59:45 -08001479 writeln!(
1480 out,
1481 " return cxxbridge1$rust_vec${}$capacity(this);",
1482 instance,
1483 );
David Tolnaydc62d712020-12-11 13:51:53 -08001484 writeln!(out, "}}");
1485
1486 writeln!(out, "template <>");
David Tolnay219c0792020-04-24 20:31:37 -07001487 writeln!(out, "const {} *Vec<{0}>::data() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001488 writeln!(out, " return cxxbridge1$rust_vec${}$data(this);", instance);
David Tolnay219c0792020-04-24 20:31:37 -07001489 writeln!(out, "}}");
David Tolnay503d0192020-04-24 22:18:56 -07001490
1491 writeln!(out, "template <>");
David Tolnayfb6b73c2020-11-10 14:32:16 -08001492 writeln!(
1493 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001494 "void Vec<{}>::reserve_total(::std::size_t cap) noexcept {{",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001495 inner,
1496 );
1497 writeln!(
1498 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001499 " return cxxbridge1$rust_vec${}$reserve_total(this, cap);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001500 instance,
1501 );
1502 writeln!(out, "}}");
1503
1504 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001505 writeln!(
1506 out,
1507 "void Vec<{}>::set_len(::std::size_t len) noexcept {{",
1508 inner,
1509 );
David Tolnayfb6b73c2020-11-10 14:32:16 -08001510 writeln!(
1511 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001512 " return cxxbridge1$rust_vec${}$set_len(this, len);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001513 instance,
1514 );
1515 writeln!(out, "}}");
1516
1517 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001518 writeln!(out, "::std::size_t Vec<{}>::stride() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001519 writeln!(out, " return cxxbridge1$rust_vec${}$stride();", instance);
David Tolnay503d0192020-04-24 22:18:56 -07001520 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001521}
1522
David Tolnay75ea17c2020-12-06 21:08:34 -08001523fn write_unique_ptr(out: &mut OutFile, ident: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001524 let ty = UniquePtr::Ident(ident);
1525 let instance = ty.to_mangled(out.types);
David Tolnay63da4d32020-04-25 09:41:12 -07001526
David Tolnay0f0162f2020-11-16 23:43:37 -08001527 writeln!(out, "#ifndef CXXBRIDGE1_UNIQUE_PTR_{}", instance);
1528 writeln!(out, "#define CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001529
David Tolnay1bdb4712020-11-25 07:27:54 -08001530 write_unique_ptr_common(out, ty);
David Tolnay63da4d32020-04-25 09:41:12 -07001531
David Tolnay0f0162f2020-11-16 23:43:37 -08001532 writeln!(out, "#endif // CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001533}
1534
1535// Shared by UniquePtr<T> and UniquePtr<CxxVector<T>>.
David Tolnay1bdb4712020-11-25 07:27:54 -08001536fn write_unique_ptr_common(out: &mut OutFile, ty: UniquePtr) {
David Tolnay0ecd05a2020-07-29 16:32:03 -07001537 out.include.new = true;
Myron Ahneba35cf2020-02-05 19:41:51 +07001538 out.include.utility = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001539 let inner = ty.to_typename(out.types);
1540 let instance = ty.to_mangled(out.types);
David Tolnay7db73692019-10-20 14:51:12 -04001541
David Tolnay63da4d32020-04-25 09:41:12 -07001542 let can_construct_from_value = match ty {
David Tolnayca0f9da2020-10-16 13:16:17 -07001543 // Some aliases are to opaque types; some are to trivial types. We can't
1544 // know at code generation time, so we generate both C++ and Rust side
1545 // bindings for a "new" method anyway. But the Rust code can't be called
1546 // for Opaque types because the 'new' method is not implemented.
David Tolnay1bdb4712020-11-25 07:27:54 -08001547 UniquePtr::Ident(ident) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001548 out.types.structs.contains_key(&ident.rust)
David Tolnay15609ab2020-11-25 07:14:12 -08001549 || out.types.enums.contains_key(&ident.rust)
David Tolnaya7c2ea12020-10-30 21:32:53 -07001550 || out.types.aliases.contains_key(&ident.rust)
David Tolnayca0f9da2020-10-16 13:16:17 -07001551 }
David Tolnay1bdb4712020-11-25 07:27:54 -08001552 UniquePtr::CxxVector(_) => false,
David Tolnay63da4d32020-04-25 09:41:12 -07001553 };
1554
David Tolnay53462762020-11-25 07:08:10 -08001555 let conditional_delete = match ty {
1556 UniquePtr::Ident(ident) => {
1557 !out.types.structs.contains_key(&ident.rust)
1558 && !out.types.enums.contains_key(&ident.rust)
1559 }
1560 UniquePtr::CxxVector(_) => false,
1561 };
1562
1563 if conditional_delete {
1564 out.builtin.is_complete = true;
1565 let definition = match ty {
1566 UniquePtr::Ident(ty) => &out.types.resolve(ty).cxx,
1567 UniquePtr::CxxVector(_) => unreachable!(),
1568 };
1569 writeln!(
1570 out,
1571 "static_assert(::rust::is_complete<{}>::value, \"definition of {} is required\");",
1572 inner, definition,
1573 );
1574 }
David Tolnay7db73692019-10-20 14:51:12 -04001575 writeln!(
1576 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001577 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001578 inner,
1579 );
1580 writeln!(
1581 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001582 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001583 inner,
1584 );
1585 writeln!(
1586 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001587 "void cxxbridge1$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001588 instance, inner,
1589 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001590 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001591 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001592 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001593 out.builtin.maybe_uninit = true;
David Tolnay63da4d32020-04-25 09:41:12 -07001594 writeln!(
David Tolnay53838912020-04-09 20:56:44 -07001595 out,
David Tolnay0b881402020-12-01 21:05:08 -08001596 "{} *cxxbridge1$unique_ptr${}$uninit(::std::unique_ptr<{}> *ptr) noexcept {{",
1597 inner, instance, inner,
David Tolnay53838912020-04-09 20:56:44 -07001598 );
David Tolnay63da4d32020-04-25 09:41:12 -07001599 writeln!(
1600 out,
David Tolnay0b881402020-12-01 21:05:08 -08001601 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1602 inner, inner, inner,
David Tolnay63da4d32020-04-25 09:41:12 -07001603 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001604 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001605 writeln!(out, " return uninit;");
David Tolnay63da4d32020-04-25 09:41:12 -07001606 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -07001607 }
David Tolnay7db73692019-10-20 14:51:12 -04001608 writeln!(
1609 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001610 "void cxxbridge1$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001611 instance, inner, inner,
1612 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001613 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001614 writeln!(out, "}}");
1615 writeln!(
1616 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001617 "const {} *cxxbridge1$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001618 inner, instance, inner,
1619 );
1620 writeln!(out, " return ptr.get();");
1621 writeln!(out, "}}");
1622 writeln!(
1623 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001624 "{} *cxxbridge1$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001625 inner, instance, inner,
1626 );
1627 writeln!(out, " return ptr.release();");
1628 writeln!(out, "}}");
1629 writeln!(
1630 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001631 "void cxxbridge1$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001632 instance, inner,
1633 );
David Tolnay53462762020-11-25 07:08:10 -08001634 if conditional_delete {
1635 out.builtin.deleter_if = true;
1636 writeln!(
1637 out,
1638 " ::rust::deleter_if<::rust::is_complete<{}>::value>{{}}(ptr);",
1639 inner,
1640 );
1641 } else {
1642 writeln!(out, " ptr->~unique_ptr();");
1643 }
David Tolnay7db73692019-10-20 14:51:12 -04001644 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001645}
Myron Ahneba35cf2020-02-05 19:41:51 +07001646
David Tolnay75ea17c2020-12-06 21:08:34 -08001647fn write_shared_ptr(out: &mut OutFile, ident: &RustName) {
David Tolnayb3b24a12020-12-01 15:27:43 -08001648 let resolved = out.types.resolve(ident);
1649 let inner = resolved.to_fully_qualified();
1650 let instance = ident.to_symbol(out.types);
1651
1652 writeln!(out, "#ifndef CXXBRIDGE1_SHARED_PTR_{}", instance);
1653 writeln!(out, "#define CXXBRIDGE1_SHARED_PTR_{}", instance);
1654 out.include.new = true;
1655 out.include.utility = true;
1656
1657 // Some aliases are to opaque types; some are to trivial types. We can't
1658 // know at code generation time, so we generate both C++ and Rust side
1659 // bindings for a "new" method anyway. But the Rust code can't be called for
1660 // Opaque types because the 'new' method is not implemented.
1661 let can_construct_from_value = out.types.structs.contains_key(&ident.rust)
1662 || out.types.enums.contains_key(&ident.rust)
1663 || out.types.aliases.contains_key(&ident.rust);
1664
David Tolnayb3b24a12020-12-01 15:27:43 -08001665 writeln!(
1666 out,
1667 "static_assert(sizeof(::std::shared_ptr<{}>) == 2 * sizeof(void *), \"\");",
1668 inner,
1669 );
1670 writeln!(
1671 out,
1672 "static_assert(alignof(::std::shared_ptr<{}>) == alignof(void *), \"\");",
1673 inner,
1674 );
1675 writeln!(
1676 out,
1677 "void cxxbridge1$shared_ptr${}$null(::std::shared_ptr<{}> *ptr) noexcept {{",
1678 instance, inner,
1679 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001680 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>();", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001681 writeln!(out, "}}");
1682 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001683 out.builtin.maybe_uninit = true;
David Tolnayb3b24a12020-12-01 15:27:43 -08001684 writeln!(
1685 out,
David Tolnay0b881402020-12-01 21:05:08 -08001686 "{} *cxxbridge1$shared_ptr${}$uninit(::std::shared_ptr<{}> *ptr) noexcept {{",
1687 inner, instance, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001688 );
1689 writeln!(
1690 out,
David Tolnay0b881402020-12-01 21:05:08 -08001691 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1692 inner, inner, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001693 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001694 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001695 writeln!(out, " return uninit;");
David Tolnayb3b24a12020-12-01 15:27:43 -08001696 writeln!(out, "}}");
1697 }
1698 writeln!(
1699 out,
1700 "void cxxbridge1$shared_ptr${}$clone(const ::std::shared_ptr<{}>& self, ::std::shared_ptr<{}> *ptr) noexcept {{",
1701 instance, inner, inner,
1702 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001703 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(self);", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001704 writeln!(out, "}}");
1705 writeln!(
1706 out,
1707 "const {} *cxxbridge1$shared_ptr${}$get(const ::std::shared_ptr<{}>& self) noexcept {{",
1708 inner, instance, inner,
1709 );
1710 writeln!(out, " return self.get();");
1711 writeln!(out, "}}");
1712 writeln!(
1713 out,
1714 "void cxxbridge1$shared_ptr${}$drop(::std::shared_ptr<{}> *self) noexcept {{",
1715 instance, inner,
1716 );
1717 writeln!(out, " self->~shared_ptr();");
1718 writeln!(out, "}}");
1719
1720 writeln!(out, "#endif // CXXBRIDGE1_SHARED_PTR_{}", instance);
1721}
1722
David Tolnay75ea17c2020-12-06 21:08:34 -08001723fn write_cxx_vector(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001724 let inner = element.to_typename(out.types);
1725 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001726
David Tolnay12320e12020-12-09 23:09:36 -08001727 out.include.cstddef = true;
1728
David Tolnay0f0162f2020-11-16 23:43:37 -08001729 writeln!(out, "#ifndef CXXBRIDGE1_VECTOR_{}", instance);
1730 writeln!(out, "#define CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001731 writeln!(
1732 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001733 "::std::size_t cxxbridge1$std$vector${}$size(const ::std::vector<{}> &s) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001734 instance, inner,
1735 );
1736 writeln!(out, " return s.size();");
1737 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001738 writeln!(
1739 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001740 "const {} *cxxbridge1$std$vector${}$get_unchecked(const ::std::vector<{}> &s, ::std::size_t pos) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001741 inner, instance, inner,
1742 );
David Tolnayb3fcf7b2020-04-30 22:58:28 -07001743 writeln!(out, " return &s[pos];");
Myron Ahneba35cf2020-02-05 19:41:51 +07001744 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001745
David Tolnay70820672020-12-07 11:15:14 -08001746 out.include.memory = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001747 write_unique_ptr_common(out, UniquePtr::CxxVector(element));
David Tolnay63da4d32020-04-25 09:41:12 -07001748
David Tolnay0f0162f2020-11-16 23:43:37 -08001749 writeln!(out, "#endif // CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001750}