blob: 4b95613d862476a5bc69415d9721313161818bc2 [file] [log] [blame]
David Tolnayf7b81fb2020-11-01 22:39:04 -08001use crate::gen::nested::NamespaceEntries;
David Tolnay8810a542020-10-31 21:39:22 -07002use crate::gen::out::OutFile;
David Tolnay942cffd2020-11-03 18:27:10 -08003use crate::gen::{builtin, include, Opt};
David Tolnay7db73692019-10-20 14:51:12 -04004use crate::syntax::atom::Atom::{self, *};
David Tolnay891061b2020-04-19 22:42:33 -07005use crate::syntax::symbol::Symbol;
Adrian Taylorc8713432020-10-21 18:20:55 -07006use crate::syntax::{
David Tolnay75ea17c2020-12-06 21:08:34 -08007 derive, mangle, Api, Enum, ExternFn, ExternType, Pair, RustName, Signature, Struct, Trait,
8 Type, Types, Var,
Adrian Taylorc8713432020-10-21 18:20:55 -07009};
Adrian Taylor7b443f72020-12-07 15:47:19 -080010use crate::{gen::block::Block, syntax::types::TrivialReason};
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 {
Adrian Taylorf831a5a2020-12-07 16:49:19 -0800120 if let Some(reasons) = out.types.required_trivial.get(&ety.name.rust) {
121 check_trivial_extern_type(out, &ety.name, reasons)
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();
David Tolnay12320e12020-12-09 23:09:36 -0800161 out.include.cstddef = true;
David Tolnay08a03db2020-12-09 23:04:36 -0800162 out.include.functional = true;
David Tolnay7da38202020-11-27 17:36:16 -0800163 let qualified = strct.name.to_fully_qualified();
164 writeln!(out, "template <> struct hash<{}> {{", qualified);
165 writeln!(
166 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800167 " ::std::size_t operator()(const {} &self) const noexcept {{",
David Tolnay7da38202020-11-27 17:36:16 -0800168 qualified,
169 );
170 let link_name = mangle::operator(&strct.name, "hash");
171 write!(out, " return ::");
172 for name in &strct.name.namespace {
173 write!(out, "{}::", name);
174 }
175 writeln!(out, "{}(self);", link_name);
176 writeln!(out, " }}");
177 writeln!(out, "}};");
178 }
179 }
180 }
181
182 out.end_block(Block::Namespace("std"));
183}
184
David Tolnaydfb82d72020-11-02 00:10:10 -0800185fn pick_includes_and_builtins(out: &mut OutFile, apis: &[Api]) {
186 for api in apis {
187 if let Api::Include(include) = api {
188 out.include.insert(include);
189 }
190 }
191
David Tolnaya7c2ea12020-10-30 21:32:53 -0700192 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -0400193 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700194 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnay89e386d2020-10-03 19:02:19 -0700195 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
196 | Some(I64) => out.include.cstdint = true,
197 Some(Usize) => out.include.cstddef = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800198 Some(Isize) => out.builtin.rust_isize = true,
David Tolnay89e386d2020-10-03 19:02:19 -0700199 Some(CxxString) => out.include.string = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800200 Some(RustString) => out.builtin.rust_string = true,
David Tolnayb3873dc2020-11-25 19:47:49 -0800201 Some(Bool) | Some(Char) | Some(F32) | Some(F64) | None => {}
David Tolnay89e386d2020-10-03 19:02:19 -0700202 },
David Tolnaydcfa8e92020-11-02 09:50:06 -0800203 Type::RustBox(_) => out.builtin.rust_box = true,
204 Type::RustVec(_) => out.builtin.rust_vec = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700205 Type::UniquePtr(_) => out.include.memory = true,
David Tolnayb3b24a12020-12-01 15:27:43 -0800206 Type::SharedPtr(_) => out.include.memory = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800207 Type::Str(_) => out.builtin.rust_str = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700208 Type::CxxVector(_) => out.include.vector = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800209 Type::Fn(_) => out.builtin.rust_fn = true,
David Tolnay5515a9e2020-11-25 19:07:54 -0800210 Type::SliceRef(_) => out.builtin.rust_slice = true,
David Tolnaye8b1bb42020-11-24 20:37:37 -0800211 Type::Array(_) => out.include.array = true,
David Tolnay11bd7ff2020-11-01 19:44:58 -0800212 Type::Ref(_) | Type::Void(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -0400213 }
214 }
David Tolnayec66d112020-10-31 21:00:26 -0700215}
David Tolnayf51447e2020-03-06 14:14:27 -0800216
David Tolnay102c7ea2020-11-08 18:58:09 -0800217fn write_struct<'a>(out: &mut OutFile<'a>, strct: &'a Struct, methods: &[&ExternFn]) {
David Tolnayb960ed22020-11-27 14:34:30 -0800218 let operator_eq = derive::contains(&strct.derives, Trait::PartialEq);
David Tolnay84389352020-11-27 17:12:10 -0800219 let operator_ord = derive::contains(&strct.derives, Trait::PartialOrd);
David Tolnayb960ed22020-11-27 14:34:30 -0800220
David Tolnay17a934c2020-11-02 00:40:04 -0800221 out.set_namespace(&strct.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800222 let guard = format!("CXXBRIDGE1_STRUCT_{}", strct.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700223 writeln!(out, "#ifndef {}", guard);
224 writeln!(out, "#define {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400225 for line in strct.doc.to_string().lines() {
226 writeln!(out, "//{}", line);
227 }
David Tolnay17a934c2020-11-02 00:40:04 -0800228 writeln!(out, "struct {} final {{", strct.name.cxx);
David Tolnay84389352020-11-27 17:12:10 -0800229
David Tolnay7db73692019-10-20 14:51:12 -0400230 for field in &strct.fields {
231 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700232 write_type_space(out, &field.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400233 writeln!(out, "{};", field.ident);
234 }
David Tolnay84389352020-11-27 17:12:10 -0800235
David Tolnay5554ebd2020-12-10 11:34:41 -0800236 writeln!(out);
David Tolnay84389352020-11-27 17:12:10 -0800237
David Tolnay102c7ea2020-11-08 18:58:09 -0800238 for method in methods {
239 write!(out, " ");
240 let sig = &method.sig;
241 let local_name = method.name.cxx.to_string();
242 write_rust_function_shim_decl(out, &local_name, sig, false);
243 writeln!(out, ";");
244 }
David Tolnay84389352020-11-27 17:12:10 -0800245
David Tolnayb960ed22020-11-27 14:34:30 -0800246 if operator_eq {
247 writeln!(
248 out,
249 " bool operator==(const {} &) const noexcept;",
250 strct.name.cxx,
251 );
252 writeln!(
253 out,
254 " bool operator!=(const {} &) const noexcept;",
255 strct.name.cxx,
256 );
257 }
David Tolnay84389352020-11-27 17:12:10 -0800258
259 if operator_ord {
260 writeln!(
261 out,
262 " bool operator<(const {} &) const noexcept;",
263 strct.name.cxx,
264 );
265 writeln!(
266 out,
267 " bool operator<=(const {} &) const noexcept;",
268 strct.name.cxx,
269 );
270 writeln!(
271 out,
272 " bool operator>(const {} &) const noexcept;",
273 strct.name.cxx,
274 );
275 writeln!(
276 out,
277 " bool operator>=(const {} &) const noexcept;",
278 strct.name.cxx,
279 );
280 }
281
David Tolnay5554ebd2020-12-10 11:34:41 -0800282 out.include.type_traits = true;
283 writeln!(out, " using IsRelocatable = ::std::true_type;");
284
David Tolnay7db73692019-10-20 14:51:12 -0400285 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700286 writeln!(out, "#endif // {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400287}
288
289fn write_struct_decl(out: &mut OutFile, ident: &Ident) {
290 writeln!(out, "struct {};", ident);
291}
292
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800293fn write_enum_decl(out: &mut OutFile, enm: &Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800294 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800295 write_atom(out, enm.repr);
296 writeln!(out, ";");
297}
298
David Tolnay8faec772020-11-02 00:18:19 -0800299fn write_struct_using(out: &mut OutFile, ident: &Pair) {
300 writeln!(out, "using {} = {};", ident.cxx, ident.to_fully_qualified());
David Tolnay8861bee2020-01-20 18:39:24 -0800301}
302
David Tolnay0b9b9f82020-11-01 20:41:00 -0800303fn write_struct_with_methods<'a>(
304 out: &mut OutFile<'a>,
305 ety: &'a ExternType,
306 methods: &[&ExternFn],
307) {
David Tolnay17a934c2020-11-02 00:40:04 -0800308 out.set_namespace(&ety.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800309 let guard = format!("CXXBRIDGE1_STRUCT_{}", ety.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700310 writeln!(out, "#ifndef {}", guard);
311 writeln!(out, "#define {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700312 for line in ety.doc.to_string().lines() {
313 writeln!(out, "//{}", line);
314 }
David Tolnay365fc7c2020-11-25 16:08:13 -0800315 out.builtin.opaque = true;
David Tolnay7c06b862020-11-25 16:59:09 -0800316 writeln!(
317 out,
318 "struct {} final : public ::rust::Opaque {{",
319 ety.name.cxx,
320 );
Joel Galenson968738f2020-04-15 14:19:33 -0700321 for method in methods {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700322 write!(out, " ");
323 let sig = &method.sig;
David Tolnay17a934c2020-11-02 00:40:04 -0800324 let local_name = method.name.cxx.to_string();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700325 write_rust_function_shim_decl(out, &local_name, sig, false);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700326 writeln!(out, ";");
327 }
328 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700329 writeln!(out, "#endif // {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700330}
331
David Tolnay0b9b9f82020-11-01 20:41:00 -0800332fn write_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800333 out.set_namespace(&enm.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800334 let guard = format!("CXXBRIDGE1_ENUM_{}", enm.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700335 writeln!(out, "#ifndef {}", guard);
336 writeln!(out, "#define {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700337 for line in enm.doc.to_string().lines() {
338 writeln!(out, "//{}", line);
339 }
David Tolnay17a934c2020-11-02 00:40:04 -0800340 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700341 write_atom(out, enm.repr);
342 writeln!(out, " {{");
Joel Galensonc03402a2020-04-23 17:31:09 -0700343 for variant in &enm.variants {
Joel Galenson88547732020-05-05 08:23:42 -0700344 writeln!(out, " {} = {},", variant.ident, variant.discriminant);
Joel Galensonc03402a2020-04-23 17:31:09 -0700345 }
346 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700347 writeln!(out, "#endif // {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700348}
349
David Tolnay0b9b9f82020-11-01 20:41:00 -0800350fn check_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800351 out.set_namespace(&enm.name.namespace);
David Tolnay06711bc2020-11-19 19:25:14 -0800352 out.include.type_traits = true;
353 writeln!(
354 out,
355 "static_assert(::std::is_enum<{}>::value, \"expected enum\");",
356 enm.name.cxx,
357 );
David Tolnay17a934c2020-11-02 00:40:04 -0800358 write!(out, "static_assert(sizeof({}) == sizeof(", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700359 write_atom(out, enm.repr);
360 writeln!(out, "), \"incorrect size\");");
Joel Galenson0f654ff2020-05-04 20:04:21 -0700361 for variant in &enm.variants {
David Tolnayf6a89f22020-05-10 23:39:27 -0700362 write!(out, "static_assert(static_cast<");
363 write_atom(out, enm.repr);
Joel Galenson0f654ff2020-05-04 20:04:21 -0700364 writeln!(
365 out,
David Tolnayf6a89f22020-05-10 23:39:27 -0700366 ">({}::{}) == {}, \"disagrees with the value in #[cxx::bridge]\");",
David Tolnay17a934c2020-11-02 00:40:04 -0800367 enm.name.cxx, variant.ident, variant.discriminant,
Joel Galenson0f654ff2020-05-04 20:04:21 -0700368 );
Joel Galenson0f654ff2020-05-04 20:04:21 -0700369 }
Joel Galenson905eb2e2020-05-04 14:58:14 -0700370}
371
Adrian Taylorf831a5a2020-12-07 16:49:19 -0800372fn check_trivial_extern_type(out: &mut OutFile, id: &Pair, reasons: &[TrivialReason]) {
David Tolnay174bd952020-11-02 09:23:12 -0800373 // NOTE: The following static assertion is just nice-to-have and not
David Tolnayfd0034e2020-10-04 13:15:34 -0700374 // necessary for soundness. That's because triviality is always declared by
375 // the user in the form of an unsafe impl of cxx::ExternType:
376 //
377 // unsafe impl ExternType for MyType {
378 // type Id = cxx::type_id!("...");
379 // type Kind = cxx::kind::Trivial;
380 // }
381 //
382 // Since the user went on the record with their unsafe impl to unsafely
383 // claim they KNOW that the type is trivial, it's fine for that to be on
David Tolnay174bd952020-11-02 09:23:12 -0800384 // them if that were wrong. However, in practice correctly reasoning about
385 // the relocatability of C++ types is challenging, particularly if the type
386 // definition were to change over time, so for now we add this check.
David Tolnayfd0034e2020-10-04 13:15:34 -0700387 //
David Tolnay174bd952020-11-02 09:23:12 -0800388 // There may be legitimate reasons to opt out of this assertion for support
389 // of types that the programmer knows are soundly Rust-movable despite not
390 // being recognized as such by the C++ type system due to a move constructor
391 // or destructor. To opt out of the relocatability check, they need to do
392 // one of the following things in any header used by `include!` in their
393 // bridge.
394 //
395 // --- if they define the type:
396 // struct MyType {
397 // ...
398 // + using IsRelocatable = std::true_type;
399 // };
400 //
401 // --- otherwise:
402 // + template <>
403 // + struct rust::IsRelocatable<MyType> : std::true_type {};
404 //
David Tolnayfd0034e2020-10-04 13:15:34 -0700405
David Tolnay174bd952020-11-02 09:23:12 -0800406 let id = id.to_fully_qualified();
407 out.builtin.relocatable = true;
Adrian Taylorf831a5a2020-12-07 16:49:19 -0800408 for reason in reasons {
409 writeln!(out, "static_assert(");
410 writeln!(out, " ::rust::IsRelocatable<{}>::value,", id);
411 writeln!(
412 out,
413 " \"type {} is not move constructible and trivially destructible in C++ yet is used as a trivial type in Rust ({})\");",
414 id,
415 reason
416 );
417 }
Adrian Taylor405e8742020-09-25 14:01:25 -0700418}
419
David Tolnayb960ed22020-11-27 14:34:30 -0800420fn write_struct_operator_decls<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
421 out.set_namespace(&strct.name.namespace);
422 out.begin_block(Block::ExternC);
423
424 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800425 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800426 writeln!(
427 out,
428 "bool {}(const {1} &, const {1} &) noexcept;",
429 link_name, strct.name.cxx,
430 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800431
432 if !derive::contains(&strct.derives, Trait::Eq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800433 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800434 writeln!(
435 out,
436 "bool {}(const {1} &, const {1} &) noexcept;",
437 link_name, strct.name.cxx,
438 );
439 }
David Tolnayb960ed22020-11-27 14:34:30 -0800440 }
441
David Tolnay84389352020-11-27 17:12:10 -0800442 if derive::contains(&strct.derives, Trait::PartialOrd) {
David Tolnaya05f9402020-11-27 17:44:05 -0800443 let link_name = mangle::operator(&strct.name, "lt");
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
David Tolnaya05f9402020-11-27 17:44:05 -0800450 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800451 writeln!(
452 out,
453 "bool {}(const {1} &, const {1} &) noexcept;",
454 link_name, strct.name.cxx,
455 );
456
457 if !derive::contains(&strct.derives, Trait::Ord) {
David Tolnaya05f9402020-11-27 17:44:05 -0800458 let link_name = mangle::operator(&strct.name, "gt");
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
David Tolnaya05f9402020-11-27 17:44:05 -0800465 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800466 writeln!(
467 out,
468 "bool {}(const {1} &, const {1} &) noexcept;",
469 link_name, strct.name.cxx,
470 );
471 }
472 }
473
David Tolnay7da38202020-11-27 17:36:16 -0800474 if derive::contains(&strct.derives, Trait::Hash) {
David Tolnay12320e12020-12-09 23:09:36 -0800475 out.include.cstddef = true;
David Tolnay7da38202020-11-27 17:36:16 -0800476 let link_name = mangle::operator(&strct.name, "hash");
477 writeln!(
478 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800479 "::std::size_t {}(const {} &) noexcept;",
David Tolnay7da38202020-11-27 17:36:16 -0800480 link_name, strct.name.cxx,
481 );
482 }
483
David Tolnayb960ed22020-11-27 14:34:30 -0800484 out.end_block(Block::ExternC);
485}
486
487fn write_struct_operators<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
488 if out.header {
489 return;
490 }
491
492 out.set_namespace(&strct.name.namespace);
493
494 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnayb960ed22020-11-27 14:34:30 -0800495 out.next_section();
496 writeln!(
497 out,
498 "bool {0}::operator==(const {0} &rhs) const noexcept {{",
499 strct.name.cxx,
500 );
David Tolnaya05f9402020-11-27 17:44:05 -0800501 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800502 writeln!(out, " return {}(*this, rhs);", link_name);
503 writeln!(out, "}}");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800504
David Tolnayb960ed22020-11-27 14:34:30 -0800505 out.next_section();
506 writeln!(
507 out,
508 "bool {0}::operator!=(const {0} &rhs) const noexcept {{",
509 strct.name.cxx,
510 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800511 if derive::contains(&strct.derives, Trait::Eq) {
512 writeln!(out, " return !(*this == rhs);");
513 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800514 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800515 writeln!(out, " return {}(*this, rhs);", link_name);
516 }
David Tolnayb960ed22020-11-27 14:34:30 -0800517 writeln!(out, "}}");
518 }
David Tolnay84389352020-11-27 17:12:10 -0800519
520 if derive::contains(&strct.derives, Trait::PartialOrd) {
521 out.next_section();
522 writeln!(
523 out,
524 "bool {0}::operator<(const {0} &rhs) const noexcept {{",
525 strct.name.cxx,
526 );
David Tolnaya05f9402020-11-27 17:44:05 -0800527 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800528 writeln!(out, " return {}(*this, rhs);", link_name);
529 writeln!(out, "}}");
530
531 out.next_section();
532 writeln!(
533 out,
534 "bool {0}::operator<=(const {0} &rhs) const noexcept {{",
535 strct.name.cxx,
536 );
David Tolnaya05f9402020-11-27 17:44:05 -0800537 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800538 writeln!(out, " return {}(*this, rhs);", link_name);
539 writeln!(out, "}}");
540
541 out.next_section();
542 writeln!(
543 out,
544 "bool {0}::operator>(const {0} &rhs) const noexcept {{",
545 strct.name.cxx,
546 );
547 if derive::contains(&strct.derives, Trait::Ord) {
548 writeln!(out, " return !(*this <= rhs);");
549 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800550 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800551 writeln!(out, " return {}(*this, rhs);", link_name);
552 }
553 writeln!(out, "}}");
554
555 out.next_section();
556 writeln!(
557 out,
558 "bool {0}::operator>=(const {0} &rhs) const noexcept {{",
559 strct.name.cxx,
560 );
561 if derive::contains(&strct.derives, Trait::Ord) {
562 writeln!(out, " return !(*this < rhs);");
563 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800564 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800565 writeln!(out, " return {}(*this, rhs);", link_name);
566 }
567 writeln!(out, "}}");
568 }
David Tolnayb960ed22020-11-27 14:34:30 -0800569}
570
David Tolnay0b9b9f82020-11-01 20:41:00 -0800571fn write_cxx_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay04770742020-11-01 13:50:50 -0800572 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800573 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800574 out.begin_block(Block::ExternC);
David Tolnaye1476af2020-11-01 13:47:25 -0800575 if let Some(annotation) = &out.opt.cxx_impl_annotations {
David Tolnaycc1ae762020-10-31 15:53:50 -0700576 write!(out, "{} ", annotation);
Adrian Taylor21f0ff02020-07-21 16:21:48 -0700577 }
David Tolnayebef4a22020-03-17 15:33:47 -0700578 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700579 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700580 write!(out, "::rust::repr::PtrLen ");
David Tolnayebef4a22020-03-17 15:33:47 -0700581 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700582 write_extern_return_type_space(out, &efn.ret);
David Tolnayebef4a22020-03-17 15:33:47 -0700583 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700584 let mangled = mangle::extern_fn(efn, out.types);
David Tolnay3caa50a2020-04-19 21:25:34 -0700585 write!(out, "{}(", mangled);
David Tolnaye439c772020-04-20 00:23:55 -0700586 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800587 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700588 write!(out, "const ");
589 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700590 write!(
591 out,
592 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800593 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700594 );
Joel Galenson3d4f6122020-04-07 15:54:05 -0700595 }
David Tolnay7db73692019-10-20 14:51:12 -0400596 for (i, arg) in efn.args.iter().enumerate() {
Joel Galenson3d4f6122020-04-07 15:54:05 -0700597 if i > 0 || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400598 write!(out, ", ");
599 }
David Tolnaya46a2372020-03-06 10:03:48 -0800600 if arg.ty == RustString {
601 write!(out, "const ");
David Tolnay313b10e2020-04-25 16:30:51 -0700602 } else if let Type::RustVec(_) = arg.ty {
603 write!(out, "const ");
David Tolnaya46a2372020-03-06 10:03:48 -0800604 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700605 write_extern_arg(out, arg);
David Tolnay7db73692019-10-20 14:51:12 -0400606 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700607 let indirect_return = indirect_return(efn, out.types);
David Tolnay7db73692019-10-20 14:51:12 -0400608 if indirect_return {
myronahne3b78ea2020-05-23 01:08:13 +0700609 if !efn.args.is_empty() || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400610 write!(out, ", ");
611 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700612 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400613 write!(out, "*return$");
614 }
615 writeln!(out, ") noexcept {{");
616 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700617 write_return_type(out, &efn.ret);
Joel Galenson3d4f6122020-04-07 15:54:05 -0700618 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800619 None => write!(out, "(*{}$)(", efn.name.rust),
Adrian Taylorc8713432020-10-21 18:20:55 -0700620 Some(receiver) => write!(
621 out,
622 "({}::*{}$)(",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700623 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800624 efn.name.rust,
Adrian Taylorc8713432020-10-21 18:20:55 -0700625 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700626 }
David Tolnay7db73692019-10-20 14:51:12 -0400627 for (i, arg) in efn.args.iter().enumerate() {
628 if i > 0 {
629 write!(out, ", ");
630 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700631 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400632 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700633 write!(out, ")");
David Tolnay4e7123f2020-04-19 21:11:37 -0700634 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800635 if !receiver.mutable {
David Tolnay4e7123f2020-04-19 21:11:37 -0700636 write!(out, " const");
637 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700638 }
639 write!(out, " = ");
640 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800641 None => write!(out, "{}", efn.name.to_fully_qualified()),
Adrian Taylorc8713432020-10-21 18:20:55 -0700642 Some(receiver) => write!(
643 out,
644 "&{}::{}",
David Tolnaya7c2ea12020-10-30 21:32:53 -0700645 out.types.resolve(&receiver.ty).to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800646 efn.name.cxx,
Adrian Taylorc8713432020-10-21 18:20:55 -0700647 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700648 }
649 writeln!(out, ";");
David Tolnay7db73692019-10-20 14:51:12 -0400650 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700651 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700652 out.builtin.ptr_len = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700653 out.builtin.trycatch = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700654 writeln!(out, "::rust::repr::PtrLen throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700655 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700656 writeln!(out, " [&] {{");
657 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700658 }
David Tolnay7db73692019-10-20 14:51:12 -0400659 if indirect_return {
David Tolnay0ecd05a2020-07-29 16:32:03 -0700660 out.include.new = true;
David Tolnay7db73692019-10-20 14:51:12 -0400661 write!(out, "new (return$) ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700662 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400663 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700664 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400665 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700666 }
667 match &efn.ret {
668 Some(Type::Ref(_)) => write!(out, "&"),
David Tolnay74d6d512020-10-31 22:22:03 -0700669 Some(Type::Str(_)) if !indirect_return => {
670 out.builtin.rust_str_repr = true;
671 write!(out, "::rust::impl<::rust::Str>::repr(");
672 }
David Tolnay73b72642020-11-25 17:44:05 -0800673 Some(ty @ Type::SliceRef(_)) if !indirect_return => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700674 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800675 write!(out, "::rust::impl<");
676 write_type(out, ty);
677 write!(out, ">::repr(");
David Tolnayeb952ba2020-04-14 15:02:24 -0700678 }
David Tolnay99642622020-03-25 13:07:35 -0700679 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400680 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700681 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800682 None => write!(out, "{}$(", efn.name.rust),
683 Some(_) => write!(out, "(self.*{}$)(", efn.name.rust),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700684 }
David Tolnay7db73692019-10-20 14:51:12 -0400685 for (i, arg) in efn.args.iter().enumerate() {
686 if i > 0 {
687 write!(out, ", ");
688 }
689 if let Type::RustBox(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700690 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400691 write!(out, "::from_raw({})", arg.ident);
692 } else if let Type::UniquePtr(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700693 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400694 write!(out, "({})", arg.ident);
David Tolnay0356d332020-10-31 19:46:41 -0700695 } else if let Type::Str(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700696 out.builtin.rust_str_new_unchecked = true;
David Tolnay0356d332020-10-31 19:46:41 -0700697 write!(
698 out,
699 "::rust::impl<::rust::Str>::new_unchecked({})",
700 arg.ident,
701 );
David Tolnaya46a2372020-03-06 10:03:48 -0800702 } else if arg.ty == RustString {
David Tolnay74d6d512020-10-31 22:22:03 -0700703 out.builtin.unsafe_bitcopy = true;
David Tolnaycc3767f2020-03-06 10:41:51 -0800704 write!(
705 out,
706 "::rust::String(::rust::unsafe_bitcopy, *{})",
707 arg.ident,
708 );
David Tolnay313b10e2020-04-25 16:30:51 -0700709 } else if let Type::RustVec(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700710 out.builtin.unsafe_bitcopy = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700711 write_type(out, &arg.ty);
David Tolnay313b10e2020-04-25 16:30:51 -0700712 write!(out, "(::rust::unsafe_bitcopy, *{})", arg.ident);
David Tolnay73b72642020-11-25 17:44:05 -0800713 } else if let Type::SliceRef(slice) = &arg.ty {
David Tolnayc5629f02020-11-23 18:32:46 -0800714 write_type(out, &arg.ty);
715 write!(out, "(static_cast<");
716 if slice.mutability.is_none() {
717 write!(out, "const ");
718 }
David Tolnay5515a9e2020-11-25 19:07:54 -0800719 write_type_space(out, &slice.inner);
720 write!(out, "*>({0}.ptr), {0}.len)", arg.ident);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700721 } else if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700722 out.include.utility = true;
David Tolnay7e219b82020-03-01 13:14:51 -0800723 write!(out, "::std::move(*{})", arg.ident);
David Tolnay7db73692019-10-20 14:51:12 -0400724 } else {
725 write!(out, "{}", arg.ident);
726 }
727 }
728 write!(out, ")");
729 match &efn.ret {
730 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
731 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800732 Some(Type::Str(_)) | Some(Type::SliceRef(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400733 _ => {}
734 }
735 if indirect_return {
736 write!(out, ")");
737 }
738 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700739 if efn.throws {
740 out.include.cstring = true;
David Tolnay1f010c62020-11-01 20:27:46 -0800741 out.builtin.exception = true;
David Tolnay5d121442020-03-17 22:14:40 -0700742 writeln!(out, " throw$.ptr = nullptr;");
743 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700744 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700745 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700746 writeln!(
747 out,
David Tolnayc5629f02020-11-23 18:32:46 -0800748 " throw$.ptr = const_cast<char *>(::cxxbridge1$exception(catch$, throw$.len));",
David Tolnayebef4a22020-03-17 15:33:47 -0700749 );
David Tolnay5d121442020-03-17 22:14:40 -0700750 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700751 writeln!(out, " return throw$;");
752 }
David Tolnay7db73692019-10-20 14:51:12 -0400753 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700754 for arg in &efn.args {
755 if let Type::Fn(f) = &arg.ty {
756 let var = &arg.ident;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700757 write_function_pointer_trampoline(out, efn, var, f);
David Tolnay75dca2e2020-03-25 20:17:52 -0700758 }
759 }
David Tolnayca563ee2020-11-01 20:12:27 -0800760 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700761}
762
763fn write_function_pointer_trampoline(
764 out: &mut OutFile,
765 efn: &ExternFn,
766 var: &Ident,
767 f: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700768) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700769 let r_trampoline = mangle::r_trampoline(efn, var, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700770 let indirect_call = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700771 write_rust_function_decl_impl(out, &r_trampoline, f, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700772
773 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700774 let c_trampoline = mangle::c_trampoline(efn, var, out.types).to_string();
775 write_rust_function_shim_impl(out, &c_trampoline, f, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400776}
777
David Tolnay0b9b9f82020-11-01 20:41:00 -0800778fn write_rust_function_decl<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800779 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800780 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700781 let link_name = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700782 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700783 write_rust_function_decl_impl(out, &link_name, efn, indirect_call);
David Tolnayca563ee2020-11-01 20:12:27 -0800784 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700785}
786
787fn write_rust_function_decl_impl(
788 out: &mut OutFile,
David Tolnay891061b2020-04-19 22:42:33 -0700789 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700790 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700791 indirect_call: bool,
792) {
David Tolnay04770742020-11-01 13:50:50 -0800793 out.next_section();
David Tolnay75dca2e2020-03-25 20:17:52 -0700794 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700795 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700796 write!(out, "::rust::repr::PtrLen ");
David Tolnay1e548172020-03-16 13:37:09 -0700797 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700798 write_extern_return_type_space(out, &sig.ret);
David Tolnay1e548172020-03-16 13:37:09 -0700799 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700800 write!(out, "{}(", link_name);
801 let mut needs_comma = false;
David Tolnaye439c772020-04-20 00:23:55 -0700802 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800803 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700804 write!(out, "const ");
805 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700806 write!(
807 out,
808 "{} &self",
David Tolnay8faec772020-11-02 00:18:19 -0800809 out.types.resolve(&receiver.ty).to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700810 );
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700811 needs_comma = true;
812 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700813 for arg in &sig.args {
814 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400815 write!(out, ", ");
816 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700817 write_extern_arg(out, arg);
David Tolnay75dca2e2020-03-25 20:17:52 -0700818 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400819 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700820 if indirect_return(sig, out.types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700821 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400822 write!(out, ", ");
823 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700824 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400825 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700826 needs_comma = true;
827 }
828 if indirect_call {
829 if needs_comma {
830 write!(out, ", ");
831 }
832 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400833 }
834 writeln!(out, ") noexcept;");
835}
836
David Tolnay0b9b9f82020-11-01 20:41:00 -0800837fn write_rust_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800838 out.set_namespace(&efn.name.namespace);
David Tolnay7db73692019-10-20 14:51:12 -0400839 for line in efn.doc.to_string().lines() {
840 writeln!(out, "//{}", line);
841 }
David Tolnaya73853b2020-04-20 01:19:56 -0700842 let local_name = match &efn.sig.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800843 None => efn.name.cxx.to_string(),
844 Some(receiver) => format!("{}::{}", out.types.resolve(&receiver.ty).cxx, efn.name.cxx),
David Tolnaya73853b2020-04-20 01:19:56 -0700845 };
David Tolnaya7c2ea12020-10-30 21:32:53 -0700846 let invoke = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700847 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700848 write_rust_function_shim_impl(out, &local_name, efn, &invoke, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700849}
850
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700851fn write_rust_function_shim_decl(
David Tolnay75dca2e2020-03-25 20:17:52 -0700852 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700853 local_name: &str,
David Tolnay75dca2e2020-03-25 20:17:52 -0700854 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700855 indirect_call: bool,
856) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700857 write_return_type(out, &sig.ret);
David Tolnay75dca2e2020-03-25 20:17:52 -0700858 write!(out, "{}(", local_name);
859 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400860 if i > 0 {
861 write!(out, ", ");
862 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700863 write_type_space(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400864 write!(out, "{}", arg.ident);
865 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700866 if indirect_call {
867 if !sig.args.is_empty() {
868 write!(out, ", ");
869 }
870 write!(out, "void *extern$");
871 }
David Tolnay1e548172020-03-16 13:37:09 -0700872 write!(out, ")");
David Tolnay86710612020-04-20 00:30:32 -0700873 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800874 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700875 write!(out, " const");
876 }
877 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700878 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700879 write!(out, " noexcept");
880 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700881}
882
883fn write_rust_function_shim_impl(
884 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700885 local_name: &str,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700886 sig: &Signature,
David Tolnay891061b2020-04-19 22:42:33 -0700887 invoke: &Symbol,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700888 indirect_call: bool,
889) {
890 if out.header && sig.receiver.is_some() {
891 // We've already defined this inside the struct.
892 return;
893 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700894 write_rust_function_shim_decl(out, local_name, sig, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400895 if out.header {
896 writeln!(out, ";");
David Tolnay439cde22020-04-20 00:46:25 -0700897 return;
David Tolnay7db73692019-10-20 14:51:12 -0400898 }
David Tolnay439cde22020-04-20 00:46:25 -0700899 writeln!(out, " {{");
900 for arg in &sig.args {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700901 if arg.ty != RustString && out.types.needs_indirect_abi(&arg.ty) {
David Tolnay439cde22020-04-20 00:46:25 -0700902 out.include.utility = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700903 out.builtin.manually_drop = true;
David Tolnay439cde22020-04-20 00:46:25 -0700904 write!(out, " ::rust::ManuallyDrop<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700905 write_type(out, &arg.ty);
David Tolnay439cde22020-04-20 00:46:25 -0700906 writeln!(out, "> {}$(::std::move({0}));", arg.ident);
907 }
908 }
909 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700910 let indirect_return = indirect_return(sig, out.types);
David Tolnay439cde22020-04-20 00:46:25 -0700911 if indirect_return {
David Tolnay74d6d512020-10-31 22:22:03 -0700912 out.builtin.maybe_uninit = true;
David Tolnay439cde22020-04-20 00:46:25 -0700913 write!(out, "::rust::MaybeUninit<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700914 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay439cde22020-04-20 00:46:25 -0700915 writeln!(out, "> return$;");
916 write!(out, " ");
917 } else if let Some(ret) = &sig.ret {
918 write!(out, "return ");
919 match ret {
920 Type::RustBox(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700921 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700922 write!(out, "::from_raw(");
923 }
924 Type::UniquePtr(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700925 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700926 write!(out, "(");
927 }
928 Type::Ref(_) => write!(out, "*"),
David Tolnay74d6d512020-10-31 22:22:03 -0700929 Type::Str(_) => {
930 out.builtin.rust_str_new_unchecked = true;
931 write!(out, "::rust::impl<::rust::Str>::new_unchecked(");
932 }
David Tolnay73b72642020-11-25 17:44:05 -0800933 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700934 out.builtin.rust_slice_new = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800935 write!(out, "::rust::impl<");
936 write_type(out, ret);
937 write!(out, ">::slice(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700938 }
David Tolnay439cde22020-04-20 00:46:25 -0700939 _ => {}
940 }
941 }
942 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700943 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700944 write!(out, "::rust::repr::PtrLen error$ = ");
David Tolnay439cde22020-04-20 00:46:25 -0700945 }
946 write!(out, "{}(", invoke);
David Tolnay9d840362020-11-10 08:50:40 -0800947 let mut needs_comma = false;
David Tolnay439cde22020-04-20 00:46:25 -0700948 if sig.receiver.is_some() {
949 write!(out, "*this");
David Tolnay9d840362020-11-10 08:50:40 -0800950 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700951 }
David Tolnay9d840362020-11-10 08:50:40 -0800952 for arg in &sig.args {
953 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700954 write!(out, ", ");
955 }
956 match &arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700957 Type::Str(_) => {
958 out.builtin.rust_str_repr = true;
959 write!(out, "::rust::impl<::rust::Str>::repr(");
960 }
David Tolnay73b72642020-11-25 17:44:05 -0800961 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700962 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800963 write!(out, "::rust::impl<");
964 write_type(out, &arg.ty);
965 write!(out, ">::repr(");
David Tolnay36aa9e02020-10-31 23:08:21 -0700966 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700967 ty if out.types.needs_indirect_abi(ty) => write!(out, "&"),
David Tolnay439cde22020-04-20 00:46:25 -0700968 _ => {}
969 }
970 write!(out, "{}", arg.ident);
971 match &arg.ty {
972 Type::RustBox(_) => write!(out, ".into_raw()"),
973 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800974 Type::Str(_) | Type::SliceRef(_) => write!(out, ")"),
David Tolnaya7c2ea12020-10-30 21:32:53 -0700975 ty if ty != RustString && out.types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay439cde22020-04-20 00:46:25 -0700976 _ => {}
977 }
David Tolnay9d840362020-11-10 08:50:40 -0800978 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700979 }
980 if indirect_return {
David Tolnay9d840362020-11-10 08:50:40 -0800981 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700982 write!(out, ", ");
983 }
984 write!(out, "&return$.value");
David Tolnay9d840362020-11-10 08:50:40 -0800985 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -0700986 }
987 if indirect_call {
David Tolnay9d840362020-11-10 08:50:40 -0800988 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -0700989 write!(out, ", ");
990 }
991 write!(out, "extern$");
992 }
993 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -0400994 if !indirect_return {
995 if let Some(ret) = &sig.ret {
David Tolnay73b72642020-11-25 17:44:05 -0800996 if let Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::SliceRef(_) = ret {
David Tolnay22602b42020-09-21 18:04:05 -0400997 write!(out, ")");
998 }
David Tolnay439cde22020-04-20 00:46:25 -0700999 }
1000 }
1001 writeln!(out, ";");
1002 if sig.throws {
David Tolnay74d6d512020-10-31 22:22:03 -07001003 out.builtin.rust_error = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001004 writeln!(out, " if (error$.ptr) {{");
David Tolnay84ddf9e2020-10-31 15:36:48 -07001005 writeln!(out, " throw ::rust::impl<::rust::Error>::error(error$);");
David Tolnay439cde22020-04-20 00:46:25 -07001006 writeln!(out, " }}");
1007 }
1008 if indirect_return {
1009 out.include.utility = true;
1010 writeln!(out, " return ::std::move(return$.value);");
1011 }
1012 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001013}
1014
David Tolnaya7c2ea12020-10-30 21:32:53 -07001015fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001016 match ty {
1017 None => write!(out, "void "),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001018 Some(ty) => write_type_space(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001019 }
1020}
1021
David Tolnay75dca2e2020-03-25 20:17:52 -07001022fn indirect_return(sig: &Signature, types: &Types) -> bool {
1023 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -07001024 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -07001025 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -07001026}
1027
David Tolnaya7c2ea12020-10-30 21:32:53 -07001028fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
David Tolnay99642622020-03-25 13:07:35 -07001029 match ty {
1030 Type::RustBox(ty) | Type::UniquePtr(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001031 write_type_space(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001032 write!(out, "*");
1033 }
1034 Type::Ref(ty) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001035 if !ty.mutable {
David Tolnay99642622020-03-25 13:07:35 -07001036 write!(out, "const ");
1037 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001038 write_type(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001039 write!(out, " *");
1040 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001041 _ => write_type(out, ty),
David Tolnay99642622020-03-25 13:07:35 -07001042 }
1043}
1044
David Tolnaya7c2ea12020-10-30 21:32:53 -07001045fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
1046 write_indirect_return_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001047 match ty {
1048 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
David Tolnay73b72642020-11-25 17:44:05 -08001049 Type::Str(_) | Type::SliceRef(_) => write!(out, " "),
David Tolnay99642622020-03-25 13:07:35 -07001050 _ => write_space_after_type(out, ty),
1051 }
1052}
1053
David Tolnaya7c2ea12020-10-30 21:32:53 -07001054fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001055 match ty {
1056 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001057 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001058 write!(out, "*");
1059 }
David Tolnay4a441222020-01-25 16:24:27 -08001060 Some(Type::Ref(ty)) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001061 if !ty.mutable {
David Tolnay4a441222020-01-25 16:24:27 -08001062 write!(out, "const ");
1063 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001064 write_type(out, &ty.inner);
David Tolnay4a441222020-01-25 16:24:27 -08001065 write!(out, " *");
1066 }
David Tolnay73b72642020-11-25 17:44:05 -08001067 Some(Type::Str(_)) | Some(Type::SliceRef(_)) => {
David Tolnay919085c2020-10-31 22:32:22 -07001068 out.builtin.ptr_len = true;
1069 write!(out, "::rust::repr::PtrLen ");
1070 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001071 Some(ty) if out.types.needs_indirect_abi(ty) => write!(out, "void "),
1072 _ => write_return_type(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001073 }
1074}
1075
David Tolnaya7c2ea12020-10-30 21:32:53 -07001076fn write_extern_arg(out: &mut OutFile, arg: &Var) {
David Tolnay7db73692019-10-20 14:51:12 -04001077 match &arg.ty {
David Tolnay4377a9e2020-04-24 15:20:26 -07001078 Type::RustBox(ty) | Type::UniquePtr(ty) | Type::CxxVector(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001079 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001080 write!(out, "*");
1081 }
David Tolnay73b72642020-11-25 17:44:05 -08001082 Type::Str(_) | Type::SliceRef(_) => {
David Tolnay919085c2020-10-31 22:32:22 -07001083 out.builtin.ptr_len = true;
1084 write!(out, "::rust::repr::PtrLen ");
1085 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001086 _ => write_type_space(out, &arg.ty),
David Tolnay7db73692019-10-20 14:51:12 -04001087 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001088 if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay7db73692019-10-20 14:51:12 -04001089 write!(out, "*");
1090 }
1091 write!(out, "{}", arg.ident);
1092}
1093
David Tolnaya7c2ea12020-10-30 21:32:53 -07001094fn write_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001095 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -07001096 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnayf6a89f22020-05-10 23:39:27 -07001097 Some(atom) => write_atom(out, atom),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001098 None => write!(out, "{}", out.types.resolve(ident).to_fully_qualified()),
David Tolnay7db73692019-10-20 14:51:12 -04001099 },
1100 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -08001101 write!(out, "::rust::Box<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001102 write_type(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001103 write!(out, ">");
1104 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001105 Type::RustVec(ty) => {
1106 write!(out, "::rust::Vec<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001107 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001108 write!(out, ">");
1109 }
David Tolnay7db73692019-10-20 14:51:12 -04001110 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -08001111 write!(out, "::std::unique_ptr<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001112 write_type(out, &ptr.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001113 write!(out, ">");
1114 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001115 Type::SharedPtr(ptr) => {
1116 write!(out, "::std::shared_ptr<");
1117 write_type(out, &ptr.inner);
1118 write!(out, ">");
1119 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001120 Type::CxxVector(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +07001121 write!(out, "::std::vector<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001122 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001123 write!(out, ">");
1124 }
David Tolnay7db73692019-10-20 14:51:12 -04001125 Type::Ref(r) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001126 if !r.mutable {
David Tolnay7db73692019-10-20 14:51:12 -04001127 write!(out, "const ");
1128 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001129 write_type(out, &r.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001130 write!(out, " &");
1131 }
1132 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -08001133 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -04001134 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001135 Type::SliceRef(slice) => {
David Tolnayc5629f02020-11-23 18:32:46 -08001136 write!(out, "::rust::Slice<");
David Tolnay5515a9e2020-11-25 19:07:54 -08001137 if slice.mutability.is_none() {
David Tolnayc5629f02020-11-23 18:32:46 -08001138 write!(out, "const ");
1139 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001140 write_type(out, &slice.inner);
1141 write!(out, ">");
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001142 }
David Tolnay75dca2e2020-03-25 20:17:52 -07001143 Type::Fn(f) => {
David Tolnayf031c322020-11-29 19:41:33 -08001144 write!(out, "::rust::Fn<");
David Tolnay75dca2e2020-03-25 20:17:52 -07001145 match &f.ret {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001146 Some(ret) => write_type(out, ret),
David Tolnay75dca2e2020-03-25 20:17:52 -07001147 None => write!(out, "void"),
1148 }
1149 write!(out, "(");
1150 for (i, arg) in f.args.iter().enumerate() {
1151 if i > 0 {
1152 write!(out, ", ");
1153 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001154 write_type(out, &arg.ty);
David Tolnay75dca2e2020-03-25 20:17:52 -07001155 }
1156 write!(out, ")>");
1157 }
Xiangpeng Hao78762352020-11-12 10:24:18 +08001158 Type::Array(a) => {
1159 write!(out, "::std::array<");
1160 write_type(out, &a.inner);
1161 write!(out, ", {}>", &a.len);
1162 }
David Tolnay2fb14e92020-03-15 23:11:38 -07001163 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001164 }
1165}
1166
David Tolnayf6a89f22020-05-10 23:39:27 -07001167fn write_atom(out: &mut OutFile, atom: Atom) {
1168 match atom {
1169 Bool => write!(out, "bool"),
David Tolnayb3873dc2020-11-25 19:47:49 -08001170 Char => write!(out, "char"),
David Tolnaybe3cbf72020-12-12 22:12:07 -08001171 U8 => write!(out, "::std::uint8_t"),
1172 U16 => write!(out, "::std::uint16_t"),
1173 U32 => write!(out, "::std::uint32_t"),
1174 U64 => write!(out, "::std::uint64_t"),
1175 Usize => write!(out, "::std::size_t"),
1176 I8 => write!(out, "::std::int8_t"),
1177 I16 => write!(out, "::std::int16_t"),
1178 I32 => write!(out, "::std::int32_t"),
1179 I64 => write!(out, "::std::int64_t"),
David Tolnayf6a89f22020-05-10 23:39:27 -07001180 Isize => write!(out, "::rust::isize"),
1181 F32 => write!(out, "float"),
1182 F64 => write!(out, "double"),
1183 CxxString => write!(out, "::std::string"),
1184 RustString => write!(out, "::rust::String"),
1185 }
1186}
1187
David Tolnaya7c2ea12020-10-30 21:32:53 -07001188fn write_type_space(out: &mut OutFile, ty: &Type) {
1189 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001190 write_space_after_type(out, ty);
1191}
1192
1193fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001194 match ty {
David Tolnayeb952ba2020-04-14 15:02:24 -07001195 Type::Ident(_)
1196 | Type::RustBox(_)
1197 | Type::UniquePtr(_)
David Tolnayb3b24a12020-12-01 15:27:43 -08001198 | Type::SharedPtr(_)
David Tolnayeb952ba2020-04-14 15:02:24 -07001199 | Type::Str(_)
David Tolnay4377a9e2020-04-24 15:20:26 -07001200 | Type::CxxVector(_)
Myron Ahneba35cf2020-02-05 19:41:51 +07001201 | Type::RustVec(_)
David Tolnay73b72642020-11-25 17:44:05 -08001202 | Type::SliceRef(_)
Xiangpeng Hao78762352020-11-12 10:24:18 +08001203 | Type::Fn(_)
1204 | Type::Array(_) => write!(out, " "),
David Tolnay7db73692019-10-20 14:51:12 -04001205 Type::Ref(_) => {}
David Tolnaye0dca7b2020-11-25 17:18:57 -08001206 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001207 }
1208}
1209
David Tolnay1bdb4712020-11-25 07:27:54 -08001210#[derive(Copy, Clone)]
1211enum UniquePtr<'a> {
David Tolnay75ea17c2020-12-06 21:08:34 -08001212 Ident(&'a RustName),
1213 CxxVector(&'a RustName),
David Tolnay1bdb4712020-11-25 07:27:54 -08001214}
1215
1216trait ToTypename {
1217 fn to_typename(&self, types: &Types) -> String;
1218}
1219
David Tolnay75ea17c2020-12-06 21:08:34 -08001220impl ToTypename for RustName {
David Tolnay1bdb4712020-11-25 07:27:54 -08001221 fn to_typename(&self, types: &Types) -> String {
1222 types.resolve(self).to_fully_qualified()
David Tolnay2eca4a02020-04-24 19:50:51 -07001223 }
1224}
1225
David Tolnay1bdb4712020-11-25 07:27:54 -08001226impl<'a> ToTypename for UniquePtr<'a> {
1227 fn to_typename(&self, types: &Types) -> String {
1228 match self {
1229 UniquePtr::Ident(ident) => ident.to_typename(types),
1230 UniquePtr::CxxVector(element) => {
1231 format!("::std::vector<{}>", element.to_typename(types))
1232 }
1233 }
1234 }
1235}
1236
1237trait ToMangled {
1238 fn to_mangled(&self, types: &Types) -> Symbol;
1239}
1240
David Tolnay75ea17c2020-12-06 21:08:34 -08001241impl ToMangled for RustName {
David Tolnay1bdb4712020-11-25 07:27:54 -08001242 fn to_mangled(&self, types: &Types) -> Symbol {
1243 self.to_symbol(types)
1244 }
1245}
1246
1247impl<'a> ToMangled for UniquePtr<'a> {
1248 fn to_mangled(&self, types: &Types) -> Symbol {
1249 match self {
1250 UniquePtr::Ident(ident) => ident.to_mangled(types),
1251 UniquePtr::CxxVector(element) => element.to_mangled(types).prefix_with("std$vector$"),
1252 }
David Tolnaybae50ef2020-04-25 12:38:41 -07001253 }
1254}
1255
David Tolnaya7c2ea12020-10-30 21:32:53 -07001256fn write_generic_instantiations(out: &mut OutFile) {
David Tolnay169bb472020-11-01 21:04:24 -08001257 if out.header {
1258 return;
1259 }
1260
1261 out.next_section();
David Tolnay078c90f2020-11-01 13:31:08 -08001262 out.set_namespace(Default::default());
David Tolnay0c033e32020-11-01 15:15:48 -08001263 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -07001264 for ty in out.types {
David Tolnayf33bc242020-12-04 18:27:02 -08001265 if let Type::RustBox(ptr) = ty {
1266 if let Type::Ident(inner) = &ptr.inner {
1267 if Atom::from(&inner.rust).is_none()
1268 && (!out.types.aliases.contains_key(&inner.rust)
1269 || out.types.explicit_impls.contains(ty))
1270 {
1271 out.next_section();
1272 write_rust_box_extern(out, &out.types.resolve(&inner));
1273 }
David Tolnay7db73692019-10-20 14:51:12 -04001274 }
David Tolnayf33bc242020-12-04 18:27:02 -08001275 } else if let Type::RustVec(vec) = ty {
1276 if let Type::Ident(inner) = &vec.inner {
1277 if Atom::from(&inner.rust).is_none()
1278 && (!out.types.aliases.contains_key(&inner.rust)
1279 || out.types.explicit_impls.contains(ty))
1280 {
David Tolnay6787be62020-04-25 11:01:02 -07001281 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001282 write_rust_vec_extern(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001283 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001284 }
David Tolnay7db73692019-10-20 14:51:12 -04001285 } else if let Type::UniquePtr(ptr) = ty {
1286 if let Type::Ident(inner) = &ptr.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001287 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001288 && (!out.types.aliases.contains_key(&inner.rust)
1289 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001290 {
David Tolnay7db73692019-10-20 14:51:12 -04001291 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -07001292 write_unique_ptr(out, inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001293 }
1294 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001295 } else if let Type::SharedPtr(ptr) = ty {
1296 if let Type::Ident(inner) = &ptr.inner {
1297 if Atom::from(&inner.rust).is_none()
David Tolnayb7453812020-12-01 21:46:55 -08001298 && (!out.types.aliases.contains_key(&inner.rust)
David Tolnayb3b24a12020-12-01 15:27:43 -08001299 || out.types.explicit_impls.contains(ty))
1300 {
1301 out.next_section();
1302 write_shared_ptr(out, inner);
1303 }
1304 }
David Tolnayf33bc242020-12-04 18:27:02 -08001305 } else if let Type::CxxVector(vector) = ty {
1306 if let Type::Ident(inner) = &vector.inner {
Adrian Taylorc8713432020-10-21 18:20:55 -07001307 if Atom::from(&inner.rust).is_none()
David Tolnaya7c2ea12020-10-30 21:32:53 -07001308 && (!out.types.aliases.contains_key(&inner.rust)
1309 || out.types.explicit_impls.contains(ty))
David Tolnay7e69f892020-10-03 22:20:22 -07001310 {
Myron Ahneba35cf2020-02-05 19:41:51 +07001311 out.next_section();
David Tolnay1bdb4712020-11-25 07:27:54 -08001312 write_cxx_vector(out, inner);
David Tolnay7db73692019-10-20 14:51:12 -04001313 }
1314 }
1315 }
1316 }
David Tolnay0c033e32020-11-01 15:15:48 -08001317 out.end_block(Block::ExternC);
David Tolnay7db73692019-10-20 14:51:12 -04001318
David Tolnay0c033e32020-11-01 15:15:48 -08001319 out.begin_block(Block::Namespace("rust"));
David Tolnay0f0162f2020-11-16 23:43:37 -08001320 out.begin_block(Block::InlineNamespace("cxxbridge1"));
David Tolnaya7c2ea12020-10-30 21:32:53 -07001321 for ty in out.types {
David Tolnayf33bc242020-12-04 18:27:02 -08001322 if let Type::RustBox(ptr) = ty {
1323 if let Type::Ident(inner) = &ptr.inner {
1324 if Atom::from(&inner.rust).is_none()
1325 && (!out.types.aliases.contains_key(&inner.rust)
1326 || out.types.explicit_impls.contains(ty))
1327 {
1328 write_rust_box_impl(out, &out.types.resolve(&inner));
1329 }
David Tolnay7db73692019-10-20 14:51:12 -04001330 }
David Tolnayf33bc242020-12-04 18:27:02 -08001331 } else if let Type::RustVec(vec) = ty {
1332 if let Type::Ident(inner) = &vec.inner {
1333 if Atom::from(&inner.rust).is_none()
1334 && (!out.types.aliases.contains_key(&inner.rust)
1335 || out.types.explicit_impls.contains(ty))
1336 {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001337 write_rust_vec_impl(out, inner);
David Tolnay6787be62020-04-25 11:01:02 -07001338 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001339 }
David Tolnay7db73692019-10-20 14:51:12 -04001340 }
1341 }
David Tolnay0f0162f2020-11-16 23:43:37 -08001342 out.end_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay0c033e32020-11-01 15:15:48 -08001343 out.end_block(Block::Namespace("rust"));
David Tolnay7db73692019-10-20 14:51:12 -04001344}
1345
David Tolnay8faec772020-11-02 00:18:19 -08001346fn write_rust_box_extern(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001347 let inner = ident.to_fully_qualified();
1348 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001349
David Tolnay0f0162f2020-11-16 23:43:37 -08001350 writeln!(out, "#ifndef CXXBRIDGE1_RUST_BOX_{}", instance);
1351 writeln!(out, "#define CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001352 writeln!(
1353 out,
David Tolnayc4b34222020-12-12 13:06:26 -08001354 "{} *cxxbridge1$box${}$alloc() noexcept;",
1355 inner, instance,
David Tolnay7db73692019-10-20 14:51:12 -04001356 );
1357 writeln!(
1358 out,
David Tolnay25b3c1d2020-12-12 15:50:10 -08001359 "void cxxbridge1$box${}$dealloc({} *) noexcept;",
1360 instance, inner,
1361 );
1362 writeln!(
1363 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001364 "void cxxbridge1$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001365 instance, inner,
1366 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001367 writeln!(out, "#endif // CXXBRIDGE1_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001368}
1369
David Tolnay75ea17c2020-12-06 21:08:34 -08001370fn write_rust_vec_extern(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001371 let inner = element.to_typename(out.types);
1372 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001373
David Tolnay12320e12020-12-09 23:09:36 -08001374 out.include.cstddef = true;
1375
David Tolnay0f0162f2020-11-16 23:43:37 -08001376 writeln!(out, "#ifndef CXXBRIDGE1_RUST_VEC_{}", instance);
1377 writeln!(out, "#define CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001378 writeln!(
1379 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001380 "void cxxbridge1$rust_vec${}$new(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnayf97c2d52020-04-25 16:37:48 -07001381 instance, inner,
1382 );
1383 writeln!(
1384 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001385 "void cxxbridge1$rust_vec${}$drop(::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001386 instance, inner,
1387 );
1388 writeln!(
1389 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001390 "::std::size_t cxxbridge1$rust_vec${}$len(const ::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001391 instance, inner,
1392 );
David Tolnay219c0792020-04-24 20:31:37 -07001393 writeln!(
1394 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001395 "::std::size_t cxxbridge1$rust_vec${}$capacity(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnaydc62d712020-12-11 13:51:53 -08001396 instance, inner,
1397 );
1398 writeln!(
1399 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001400 "const {} *cxxbridge1$rust_vec${}$data(const ::rust::Vec<{0}> *ptr) noexcept;",
David Tolnay219c0792020-04-24 20:31:37 -07001401 inner, instance,
1402 );
David Tolnay503d0192020-04-24 22:18:56 -07001403 writeln!(
1404 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001405 "void cxxbridge1$rust_vec${}$reserve_total(::rust::Vec<{}> *ptr, ::std::size_t cap) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001406 instance, inner,
1407 );
1408 writeln!(
1409 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001410 "void cxxbridge1$rust_vec${}$set_len(::rust::Vec<{}> *ptr, ::std::size_t len) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001411 instance, inner,
1412 );
1413 writeln!(
1414 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001415 "::std::size_t cxxbridge1$rust_vec${}$stride() noexcept;",
David Tolnay503d0192020-04-24 22:18:56 -07001416 instance,
1417 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001418 writeln!(out, "#endif // CXXBRIDGE1_RUST_VEC_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001419}
1420
David Tolnay8faec772020-11-02 00:18:19 -08001421fn write_rust_box_impl(out: &mut OutFile, ident: &Pair) {
Adrian Taylorc8713432020-10-21 18:20:55 -07001422 let inner = ident.to_fully_qualified();
1423 let instance = ident.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001424
1425 writeln!(out, "template <>");
David Tolnaye5703162020-12-12 16:26:35 -08001426 writeln!(
1427 out,
1428 "{} *Box<{}>::allocation::alloc() noexcept {{",
1429 inner, inner,
1430 );
David Tolnayc4b34222020-12-12 13:06:26 -08001431 writeln!(out, " return cxxbridge1$box${}$alloc();", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001432 writeln!(out, "}}");
1433
1434 writeln!(out, "template <>");
David Tolnay25b3c1d2020-12-12 15:50:10 -08001435 writeln!(
1436 out,
David Tolnaye5703162020-12-12 16:26:35 -08001437 "void Box<{}>::allocation::dealloc({} *ptr) noexcept {{",
David Tolnay25b3c1d2020-12-12 15:50:10 -08001438 inner, inner,
1439 );
1440 writeln!(out, " cxxbridge1$box${}$dealloc(ptr);", instance);
1441 writeln!(out, "}}");
1442
1443 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -08001444 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001445 writeln!(out, " cxxbridge1$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001446 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001447}
1448
David Tolnay75ea17c2020-12-06 21:08:34 -08001449fn write_rust_vec_impl(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001450 let inner = element.to_typename(out.types);
1451 let instance = element.to_mangled(out.types);
David Tolnay4791f1c2020-03-17 21:53:16 -07001452
David Tolnay12320e12020-12-09 23:09:36 -08001453 out.include.cstddef = true;
1454
Myron Ahneba35cf2020-02-05 19:41:51 +07001455 writeln!(out, "template <>");
David Tolnayf97c2d52020-04-25 16:37:48 -07001456 writeln!(out, "Vec<{}>::Vec() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001457 writeln!(out, " cxxbridge1$rust_vec${}$new(this);", instance);
David Tolnayf97c2d52020-04-25 16:37:48 -07001458 writeln!(out, "}}");
1459
1460 writeln!(out, "template <>");
Myron Ahneba35cf2020-02-05 19:41:51 +07001461 writeln!(out, "void Vec<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001462 writeln!(out, " return cxxbridge1$rust_vec${}$drop(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001463 writeln!(out, "}}");
1464
1465 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001466 writeln!(
1467 out,
1468 "::std::size_t Vec<{}>::size() const noexcept {{",
1469 inner,
1470 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001471 writeln!(out, " return cxxbridge1$rust_vec${}$len(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001472 writeln!(out, "}}");
David Tolnay219c0792020-04-24 20:31:37 -07001473
1474 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001475 writeln!(
1476 out,
1477 "::std::size_t Vec<{}>::capacity() const noexcept {{",
1478 inner,
1479 );
David Tolnay381d9532020-12-11 13:59:45 -08001480 writeln!(
1481 out,
1482 " return cxxbridge1$rust_vec${}$capacity(this);",
1483 instance,
1484 );
David Tolnaydc62d712020-12-11 13:51:53 -08001485 writeln!(out, "}}");
1486
1487 writeln!(out, "template <>");
David Tolnay219c0792020-04-24 20:31:37 -07001488 writeln!(out, "const {} *Vec<{0}>::data() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001489 writeln!(out, " return cxxbridge1$rust_vec${}$data(this);", instance);
David Tolnay219c0792020-04-24 20:31:37 -07001490 writeln!(out, "}}");
David Tolnay503d0192020-04-24 22:18:56 -07001491
1492 writeln!(out, "template <>");
David Tolnayfb6b73c2020-11-10 14:32:16 -08001493 writeln!(
1494 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001495 "void Vec<{}>::reserve_total(::std::size_t cap) noexcept {{",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001496 inner,
1497 );
1498 writeln!(
1499 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001500 " return cxxbridge1$rust_vec${}$reserve_total(this, cap);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001501 instance,
1502 );
1503 writeln!(out, "}}");
1504
1505 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001506 writeln!(
1507 out,
1508 "void Vec<{}>::set_len(::std::size_t len) noexcept {{",
1509 inner,
1510 );
David Tolnayfb6b73c2020-11-10 14:32:16 -08001511 writeln!(
1512 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001513 " return cxxbridge1$rust_vec${}$set_len(this, len);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001514 instance,
1515 );
1516 writeln!(out, "}}");
1517
1518 writeln!(out, "template <>");
David Tolnaybe3cbf72020-12-12 22:12:07 -08001519 writeln!(out, "::std::size_t Vec<{}>::stride() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001520 writeln!(out, " return cxxbridge1$rust_vec${}$stride();", instance);
David Tolnay503d0192020-04-24 22:18:56 -07001521 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001522}
1523
David Tolnay75ea17c2020-12-06 21:08:34 -08001524fn write_unique_ptr(out: &mut OutFile, ident: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001525 let ty = UniquePtr::Ident(ident);
1526 let instance = ty.to_mangled(out.types);
David Tolnay63da4d32020-04-25 09:41:12 -07001527
David Tolnay0f0162f2020-11-16 23:43:37 -08001528 writeln!(out, "#ifndef CXXBRIDGE1_UNIQUE_PTR_{}", instance);
1529 writeln!(out, "#define CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001530
David Tolnay1bdb4712020-11-25 07:27:54 -08001531 write_unique_ptr_common(out, ty);
David Tolnay63da4d32020-04-25 09:41:12 -07001532
David Tolnay0f0162f2020-11-16 23:43:37 -08001533 writeln!(out, "#endif // CXXBRIDGE1_UNIQUE_PTR_{}", instance);
David Tolnay63da4d32020-04-25 09:41:12 -07001534}
1535
1536// Shared by UniquePtr<T> and UniquePtr<CxxVector<T>>.
David Tolnay1bdb4712020-11-25 07:27:54 -08001537fn write_unique_ptr_common(out: &mut OutFile, ty: UniquePtr) {
David Tolnay0ecd05a2020-07-29 16:32:03 -07001538 out.include.new = true;
Myron Ahneba35cf2020-02-05 19:41:51 +07001539 out.include.utility = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001540 let inner = ty.to_typename(out.types);
1541 let instance = ty.to_mangled(out.types);
David Tolnay7db73692019-10-20 14:51:12 -04001542
David Tolnay63da4d32020-04-25 09:41:12 -07001543 let can_construct_from_value = match ty {
David Tolnayca0f9da2020-10-16 13:16:17 -07001544 // Some aliases are to opaque types; some are to trivial types. We can't
1545 // know at code generation time, so we generate both C++ and Rust side
1546 // bindings for a "new" method anyway. But the Rust code can't be called
1547 // for Opaque types because the 'new' method is not implemented.
David Tolnay1bdb4712020-11-25 07:27:54 -08001548 UniquePtr::Ident(ident) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001549 out.types.structs.contains_key(&ident.rust)
David Tolnay15609ab2020-11-25 07:14:12 -08001550 || out.types.enums.contains_key(&ident.rust)
David Tolnaya7c2ea12020-10-30 21:32:53 -07001551 || out.types.aliases.contains_key(&ident.rust)
David Tolnayca0f9da2020-10-16 13:16:17 -07001552 }
David Tolnay1bdb4712020-11-25 07:27:54 -08001553 UniquePtr::CxxVector(_) => false,
David Tolnay63da4d32020-04-25 09:41:12 -07001554 };
1555
David Tolnay53462762020-11-25 07:08:10 -08001556 let conditional_delete = match ty {
1557 UniquePtr::Ident(ident) => {
1558 !out.types.structs.contains_key(&ident.rust)
1559 && !out.types.enums.contains_key(&ident.rust)
1560 }
1561 UniquePtr::CxxVector(_) => false,
1562 };
1563
1564 if conditional_delete {
1565 out.builtin.is_complete = true;
1566 let definition = match ty {
1567 UniquePtr::Ident(ty) => &out.types.resolve(ty).cxx,
1568 UniquePtr::CxxVector(_) => unreachable!(),
1569 };
1570 writeln!(
1571 out,
1572 "static_assert(::rust::is_complete<{}>::value, \"definition of {} is required\");",
1573 inner, definition,
1574 );
1575 }
David Tolnay7db73692019-10-20 14:51:12 -04001576 writeln!(
1577 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001578 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001579 inner,
1580 );
1581 writeln!(
1582 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001583 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001584 inner,
1585 );
1586 writeln!(
1587 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001588 "void cxxbridge1$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001589 instance, inner,
1590 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001591 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001592 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001593 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001594 out.builtin.maybe_uninit = true;
David Tolnay63da4d32020-04-25 09:41:12 -07001595 writeln!(
David Tolnay53838912020-04-09 20:56:44 -07001596 out,
David Tolnay0b881402020-12-01 21:05:08 -08001597 "{} *cxxbridge1$unique_ptr${}$uninit(::std::unique_ptr<{}> *ptr) noexcept {{",
1598 inner, instance, inner,
David Tolnay53838912020-04-09 20:56:44 -07001599 );
David Tolnay63da4d32020-04-25 09:41:12 -07001600 writeln!(
1601 out,
David Tolnay0b881402020-12-01 21:05:08 -08001602 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1603 inner, inner, inner,
David Tolnay63da4d32020-04-25 09:41:12 -07001604 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001605 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001606 writeln!(out, " return uninit;");
David Tolnay63da4d32020-04-25 09:41:12 -07001607 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -07001608 }
David Tolnay7db73692019-10-20 14:51:12 -04001609 writeln!(
1610 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001611 "void cxxbridge1$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001612 instance, inner, inner,
1613 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001614 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001615 writeln!(out, "}}");
1616 writeln!(
1617 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001618 "const {} *cxxbridge1$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001619 inner, instance, inner,
1620 );
1621 writeln!(out, " return ptr.get();");
1622 writeln!(out, "}}");
1623 writeln!(
1624 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001625 "{} *cxxbridge1$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001626 inner, instance, inner,
1627 );
1628 writeln!(out, " return ptr.release();");
1629 writeln!(out, "}}");
1630 writeln!(
1631 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001632 "void cxxbridge1$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001633 instance, inner,
1634 );
David Tolnay53462762020-11-25 07:08:10 -08001635 if conditional_delete {
1636 out.builtin.deleter_if = true;
1637 writeln!(
1638 out,
1639 " ::rust::deleter_if<::rust::is_complete<{}>::value>{{}}(ptr);",
1640 inner,
1641 );
1642 } else {
1643 writeln!(out, " ptr->~unique_ptr();");
1644 }
David Tolnay7db73692019-10-20 14:51:12 -04001645 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001646}
Myron Ahneba35cf2020-02-05 19:41:51 +07001647
David Tolnay75ea17c2020-12-06 21:08:34 -08001648fn write_shared_ptr(out: &mut OutFile, ident: &RustName) {
David Tolnayb3b24a12020-12-01 15:27:43 -08001649 let resolved = out.types.resolve(ident);
1650 let inner = resolved.to_fully_qualified();
1651 let instance = ident.to_symbol(out.types);
1652
1653 writeln!(out, "#ifndef CXXBRIDGE1_SHARED_PTR_{}", instance);
1654 writeln!(out, "#define CXXBRIDGE1_SHARED_PTR_{}", instance);
1655 out.include.new = true;
1656 out.include.utility = true;
1657
1658 // Some aliases are to opaque types; some are to trivial types. We can't
1659 // know at code generation time, so we generate both C++ and Rust side
1660 // bindings for a "new" method anyway. But the Rust code can't be called for
1661 // Opaque types because the 'new' method is not implemented.
1662 let can_construct_from_value = out.types.structs.contains_key(&ident.rust)
1663 || out.types.enums.contains_key(&ident.rust)
1664 || out.types.aliases.contains_key(&ident.rust);
1665
David Tolnayb3b24a12020-12-01 15:27:43 -08001666 writeln!(
1667 out,
1668 "static_assert(sizeof(::std::shared_ptr<{}>) == 2 * sizeof(void *), \"\");",
1669 inner,
1670 );
1671 writeln!(
1672 out,
1673 "static_assert(alignof(::std::shared_ptr<{}>) == alignof(void *), \"\");",
1674 inner,
1675 );
1676 writeln!(
1677 out,
1678 "void cxxbridge1$shared_ptr${}$null(::std::shared_ptr<{}> *ptr) noexcept {{",
1679 instance, inner,
1680 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001681 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>();", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001682 writeln!(out, "}}");
1683 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001684 out.builtin.maybe_uninit = true;
David Tolnayb3b24a12020-12-01 15:27:43 -08001685 writeln!(
1686 out,
David Tolnay0b881402020-12-01 21:05:08 -08001687 "{} *cxxbridge1$shared_ptr${}$uninit(::std::shared_ptr<{}> *ptr) noexcept {{",
1688 inner, instance, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001689 );
1690 writeln!(
1691 out,
David Tolnay0b881402020-12-01 21:05:08 -08001692 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1693 inner, inner, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001694 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001695 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001696 writeln!(out, " return uninit;");
David Tolnayb3b24a12020-12-01 15:27:43 -08001697 writeln!(out, "}}");
1698 }
1699 writeln!(
1700 out,
1701 "void cxxbridge1$shared_ptr${}$clone(const ::std::shared_ptr<{}>& self, ::std::shared_ptr<{}> *ptr) noexcept {{",
1702 instance, inner, inner,
1703 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001704 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(self);", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001705 writeln!(out, "}}");
1706 writeln!(
1707 out,
1708 "const {} *cxxbridge1$shared_ptr${}$get(const ::std::shared_ptr<{}>& self) noexcept {{",
1709 inner, instance, inner,
1710 );
1711 writeln!(out, " return self.get();");
1712 writeln!(out, "}}");
1713 writeln!(
1714 out,
1715 "void cxxbridge1$shared_ptr${}$drop(::std::shared_ptr<{}> *self) noexcept {{",
1716 instance, inner,
1717 );
1718 writeln!(out, " self->~shared_ptr();");
1719 writeln!(out, "}}");
1720
1721 writeln!(out, "#endif // CXXBRIDGE1_SHARED_PTR_{}", instance);
1722}
1723
David Tolnay75ea17c2020-12-06 21:08:34 -08001724fn write_cxx_vector(out: &mut OutFile, element: &RustName) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001725 let inner = element.to_typename(out.types);
1726 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001727
David Tolnay12320e12020-12-09 23:09:36 -08001728 out.include.cstddef = true;
1729
David Tolnay0f0162f2020-11-16 23:43:37 -08001730 writeln!(out, "#ifndef CXXBRIDGE1_VECTOR_{}", instance);
1731 writeln!(out, "#define CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001732 writeln!(
1733 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001734 "::std::size_t cxxbridge1$std$vector${}$size(const ::std::vector<{}> &s) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001735 instance, inner,
1736 );
1737 writeln!(out, " return s.size();");
1738 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001739 writeln!(
1740 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001741 "const {} *cxxbridge1$std$vector${}$get_unchecked(const ::std::vector<{}> &s, ::std::size_t pos) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001742 inner, instance, inner,
1743 );
David Tolnayb3fcf7b2020-04-30 22:58:28 -07001744 writeln!(out, " return &s[pos];");
Myron Ahneba35cf2020-02-05 19:41:51 +07001745 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001746
David Tolnay70820672020-12-07 11:15:14 -08001747 out.include.memory = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001748 write_unique_ptr_common(out, UniquePtr::CxxVector(element));
David Tolnay63da4d32020-04-25 09:41:12 -07001749
David Tolnay0f0162f2020-11-16 23:43:37 -08001750 writeln!(out, "#endif // CXXBRIDGE1_VECTOR_{}", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001751}