blob: 2404f9c0c4b70d7c6f0f8c7c02759f6612add5cf [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 Tolnay3abed472020-12-31 23:34:53 -08006use crate::syntax::instantiate::ImplKey;
David Tolnaye352c1e2020-12-31 16:41:05 -08007use crate::syntax::map::UnorderedMap as Map;
David Tolnaye5cd1482021-01-02 21:04:32 -08008use crate::syntax::set::UnorderedSet;
David Tolnay891061b2020-04-19 22:42:33 -07009use crate::syntax::symbol::Symbol;
David Tolnay5b1d8632020-12-20 22:05:11 -080010use crate::syntax::trivial::{self, TrivialReason};
Adrian Taylorc8713432020-10-21 18:20:55 -070011use crate::syntax::{
David Tolnay3abed472020-12-31 23:34:53 -080012 derive, mangle, Api, Enum, ExternFn, ExternType, Pair, Signature, Struct, Trait, Type,
13 TypeAlias, Types, Var,
Adrian Taylorc8713432020-10-21 18:20:55 -070014};
David Tolnay7db73692019-10-20 14:51:12 -040015use proc_macro2::Ident;
16
David Tolnayac7188c2020-11-01 16:58:16 -080017pub(super) fn gen(apis: &[Api], types: &Types, opt: &Opt, header: bool) -> Vec<u8> {
David Tolnaye1476af2020-11-01 13:47:25 -080018 let mut out_file = OutFile::new(header, opt, types);
David Tolnay7db73692019-10-20 14:51:12 -040019 let out = &mut out_file;
20
David Tolnaydfb82d72020-11-02 00:10:10 -080021 pick_includes_and_builtins(out, apis);
David Tolnay4aae7c02020-10-28 12:35:42 -070022 out.include.extend(&opt.include);
David Tolnay7db73692019-10-20 14:51:12 -040023
David Tolnay7b0e5102020-11-01 23:22:12 -080024 write_forward_declarations(out, apis);
David Tolnaye1109d92020-11-01 20:51:56 -080025 write_data_structures(out, apis);
26 write_functions(out, apis);
David Tolnay169bb472020-11-01 21:04:24 -080027 write_generic_instantiations(out);
David Tolnay7db73692019-10-20 14:51:12 -040028
David Tolnay3374d8d2020-10-31 22:18:45 -070029 builtin::write(out);
David Tolnay2f3e90b2020-10-31 22:16:51 -070030 include::write(out);
Adrian Taylorc8713432020-10-21 18:20:55 -070031
David Tolnayac7188c2020-11-01 16:58:16 -080032 out_file.content()
Adrian Taylorc8713432020-10-21 18:20:55 -070033}
34
David Tolnay7b0e5102020-11-01 23:22:12 -080035fn write_forward_declarations(out: &mut OutFile, apis: &[Api]) {
36 let needs_forward_declaration = |api: &&Api| match api {
David Tolnay4bfca112020-11-01 23:59:11 -080037 Api::Struct(_) | Api::CxxType(_) | Api::RustType(_) => true,
David Tolnay17a934c2020-11-02 00:40:04 -080038 Api::Enum(enm) => !out.types.cxx.contains(&enm.name.rust),
David Tolnay7b0e5102020-11-01 23:22:12 -080039 _ => false,
40 };
Adrian Taylorc8713432020-10-21 18:20:55 -070041
David Tolnay7b0e5102020-11-01 23:22:12 -080042 let apis_by_namespace =
43 NamespaceEntries::new(apis.iter().filter(needs_forward_declaration).collect());
44
David Tolnayd920be52020-11-01 23:34:30 -080045 write(out, &apis_by_namespace, 0);
David Tolnay7b0e5102020-11-01 23:22:12 -080046
David Tolnayd920be52020-11-01 23:34:30 -080047 fn write(out: &mut OutFile, ns_entries: &NamespaceEntries, indent: usize) {
David Tolnay7b0e5102020-11-01 23:22:12 -080048 let apis = ns_entries.direct_content();
49
50 for api in apis {
David Tolnayd920be52020-11-01 23:34:30 -080051 write!(out, "{:1$}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080052 match api {
David Tolnayed6ba4a2021-01-01 14:59:40 -080053 Api::Struct(strct) => write_struct_decl(out, &strct.name),
David Tolnay0e3ee8e2020-11-01 23:46:52 -080054 Api::Enum(enm) => write_enum_decl(out, enm),
David Tolnay17a934c2020-11-02 00:40:04 -080055 Api::CxxType(ety) => write_struct_using(out, &ety.name),
David Tolnayed6ba4a2021-01-01 14:59:40 -080056 Api::RustType(ety) => write_struct_decl(out, &ety.name),
David Tolnay7b0e5102020-11-01 23:22:12 -080057 _ => unreachable!(),
58 }
David Tolnay7db73692019-10-20 14:51:12 -040059 }
David Tolnay7db73692019-10-20 14:51:12 -040060
David Tolnay7b0e5102020-11-01 23:22:12 -080061 for (namespace, nested_ns_entries) in ns_entries.nested_content() {
David Tolnayd920be52020-11-01 23:34:30 -080062 writeln!(out, "{:2$}namespace {} {{", "", namespace, indent);
63 write(out, nested_ns_entries, indent + 2);
64 writeln!(out, "{:1$}}}", "", indent);
David Tolnay7b0e5102020-11-01 23:22:12 -080065 }
Adrian Taylorf9213622020-10-31 22:25:42 -070066 }
67}
68
David Tolnaye1109d92020-11-01 20:51:56 -080069fn write_data_structures<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnaye352c1e2020-12-31 16:41:05 -080070 let mut methods_for_type = Map::new();
David Tolnay630af882020-10-31 22:03:47 -070071 for api in apis {
David Tolnay102c7ea2020-11-08 18:58:09 -080072 if let Api::CxxFunction(efn) | Api::RustFunction(efn) = api {
David Tolnayf94bef12020-04-17 14:46:42 -070073 if let Some(receiver) = &efn.sig.receiver {
74 methods_for_type
Adrian Taylorc8713432020-10-21 18:20:55 -070075 .entry(&receiver.ty.rust)
David Tolnayf94bef12020-04-17 14:46:42 -070076 .or_insert_with(Vec::new)
77 .push(efn);
78 }
79 }
80 }
Joel Galenson968738f2020-04-15 14:19:33 -070081
David Tolnaye352c1e2020-12-31 16:41:05 -080082 let mut structs_written = UnorderedSet::new();
David Tolnay5439fa12020-11-03 18:45:01 -080083 let mut toposorted_structs = out.types.toposorted_structs.iter();
David Tolnay9622b462020-11-02 21:34:42 -080084 for api in apis {
Joel Galensonc1c4e7a2020-04-15 10:21:00 -070085 match api {
David Tolnay5439fa12020-11-03 18:45:01 -080086 Api::Struct(strct) if !structs_written.contains(&strct.name.rust) => {
87 for next in &mut toposorted_structs {
88 if !out.types.cxx.contains(&strct.name.rust) {
89 out.next_section();
David Tolnay102c7ea2020-11-08 18:58:09 -080090 let methods = methods_for_type
91 .get(&strct.name.rust)
92 .map(Vec::as_slice)
93 .unwrap_or_default();
94 write_struct(out, next, methods);
David Tolnay5439fa12020-11-03 18:45:01 -080095 }
96 structs_written.insert(&next.name.rust);
97 if next.name.rust == strct.name.rust {
98 break;
99 }
100 }
101 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700102 Api::Enum(enm) => {
103 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800104 if out.types.cxx.contains(&enm.name.rust) {
Joel Galenson905eb2e2020-05-04 14:58:14 -0700105 check_enum(out, enm);
106 } else {
107 write_enum(out, enm);
108 }
Joel Galensonc03402a2020-04-23 17:31:09 -0700109 }
David Tolnayc1fe0052020-04-17 15:15:06 -0700110 Api::RustType(ety) => {
David Tolnay8d067de2020-12-26 16:18:23 -0800111 out.next_section();
112 let methods = methods_for_type
113 .get(&ety.name.rust)
114 .map(Vec::as_slice)
115 .unwrap_or_default();
116 write_opaque_type(out, ety, methods);
David Tolnayc1fe0052020-04-17 15:15:06 -0700117 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700118 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400119 }
120 }
121
David Tolnay8b229532021-01-02 18:30:45 -0800122 if out.header {
123 return;
124 }
125
David Tolnay440e24a2021-01-01 23:25:34 -0800126 out.set_namespace(Default::default());
127
David Tolnay8b229532021-01-02 18:30:45 -0800128 out.next_section();
129 for api in apis {
130 if let Api::TypeAlias(ety) = api {
131 if let Some(reasons) = out.types.required_trivial.get(&ety.name.rust) {
132 check_trivial_extern_type(out, ety, reasons)
David Tolnayfabca772020-10-03 21:25:41 -0700133 }
134 }
135 }
David Tolnay1b0339c2020-11-01 20:37:50 -0800136}
137
David Tolnaye1109d92020-11-01 20:51:56 -0800138fn write_functions<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayce5a91f2020-10-31 22:42:08 -0700139 if !out.header {
David Tolnay7db73692019-10-20 14:51:12 -0400140 for api in apis {
David Tolnay04770742020-11-01 13:50:50 -0800141 match api {
David Tolnayb960ed22020-11-27 14:34:30 -0800142 Api::Struct(strct) => write_struct_operator_decls(out, strct),
David Tolnay358bc4b2020-12-26 22:37:57 -0800143 Api::RustType(ety) => write_opaque_type_layout_decls(out, ety),
David Tolnay04770742020-11-01 13:50:50 -0800144 Api::CxxFunction(efn) => write_cxx_function_shim(out, efn),
145 Api::RustFunction(efn) => write_rust_function_decl(out, efn),
146 _ => {}
147 }
David Tolnay7db73692019-10-20 14:51:12 -0400148 }
David Tolnay7da38202020-11-27 17:36:16 -0800149
150 write_std_specializations(out, apis);
David Tolnay7db73692019-10-20 14:51:12 -0400151 }
152
153 for api in apis {
David Tolnayb960ed22020-11-27 14:34:30 -0800154 match api {
155 Api::Struct(strct) => write_struct_operators(out, strct),
David Tolnay358bc4b2020-12-26 22:37:57 -0800156 Api::RustType(ety) => write_opaque_type_layout(out, ety),
David Tolnayb960ed22020-11-27 14:34:30 -0800157 Api::RustFunction(efn) => {
158 out.next_section();
159 write_rust_function_shim(out, efn);
160 }
161 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400162 }
163 }
David Tolnay7db73692019-10-20 14:51:12 -0400164}
165
David Tolnay7da38202020-11-27 17:36:16 -0800166fn write_std_specializations(out: &mut OutFile, apis: &[Api]) {
167 out.set_namespace(Default::default());
168 out.begin_block(Block::Namespace("std"));
169
170 for api in apis {
171 if let Api::Struct(strct) = api {
172 if derive::contains(&strct.derives, Trait::Hash) {
173 out.next_section();
David Tolnay12320e12020-12-09 23:09:36 -0800174 out.include.cstddef = true;
David Tolnay08a03db2020-12-09 23:04:36 -0800175 out.include.functional = true;
David Tolnay7da38202020-11-27 17:36:16 -0800176 let qualified = strct.name.to_fully_qualified();
177 writeln!(out, "template <> struct hash<{}> {{", qualified);
178 writeln!(
179 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800180 " ::std::size_t operator()(const {} &self) const noexcept {{",
David Tolnay7da38202020-11-27 17:36:16 -0800181 qualified,
182 );
183 let link_name = mangle::operator(&strct.name, "hash");
184 write!(out, " return ::");
185 for name in &strct.name.namespace {
186 write!(out, "{}::", name);
187 }
188 writeln!(out, "{}(self);", link_name);
189 writeln!(out, " }}");
190 writeln!(out, "}};");
191 }
192 }
193 }
194
195 out.end_block(Block::Namespace("std"));
196}
197
David Tolnaydfb82d72020-11-02 00:10:10 -0800198fn pick_includes_and_builtins(out: &mut OutFile, apis: &[Api]) {
199 for api in apis {
200 if let Api::Include(include) = api {
201 out.include.insert(include);
202 }
203 }
204
David Tolnaya7c2ea12020-10-30 21:32:53 -0700205 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -0400206 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700207 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnay89e386d2020-10-03 19:02:19 -0700208 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
209 | Some(I64) => out.include.cstdint = true,
210 Some(Usize) => out.include.cstddef = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800211 Some(Isize) => out.builtin.rust_isize = true,
David Tolnay89e386d2020-10-03 19:02:19 -0700212 Some(CxxString) => out.include.string = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800213 Some(RustString) => out.builtin.rust_string = true,
David Tolnayb3873dc2020-11-25 19:47:49 -0800214 Some(Bool) | Some(Char) | Some(F32) | Some(F64) | None => {}
David Tolnay89e386d2020-10-03 19:02:19 -0700215 },
David Tolnaydcfa8e92020-11-02 09:50:06 -0800216 Type::RustBox(_) => out.builtin.rust_box = true,
217 Type::RustVec(_) => out.builtin.rust_vec = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700218 Type::UniquePtr(_) => out.include.memory = true,
David Tolnay215e77f2020-12-28 17:09:48 -0800219 Type::SharedPtr(_) | Type::WeakPtr(_) => out.include.memory = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800220 Type::Str(_) => out.builtin.rust_str = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700221 Type::CxxVector(_) => out.include.vector = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800222 Type::Fn(_) => out.builtin.rust_fn = true,
David Tolnay5515a9e2020-11-25 19:07:54 -0800223 Type::SliceRef(_) => out.builtin.rust_slice = true,
David Tolnaye8b1bb42020-11-24 20:37:37 -0800224 Type::Array(_) => out.include.array = true,
David Tolnay11bd7ff2020-11-01 19:44:58 -0800225 Type::Ref(_) | Type::Void(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -0400226 }
227 }
David Tolnayec66d112020-10-31 21:00:26 -0700228}
David Tolnayf51447e2020-03-06 14:14:27 -0800229
David Tolnay102c7ea2020-11-08 18:58:09 -0800230fn write_struct<'a>(out: &mut OutFile<'a>, strct: &'a Struct, methods: &[&ExternFn]) {
David Tolnayb960ed22020-11-27 14:34:30 -0800231 let operator_eq = derive::contains(&strct.derives, Trait::PartialEq);
David Tolnay84389352020-11-27 17:12:10 -0800232 let operator_ord = derive::contains(&strct.derives, Trait::PartialOrd);
David Tolnayb960ed22020-11-27 14:34:30 -0800233
David Tolnay17a934c2020-11-02 00:40:04 -0800234 out.set_namespace(&strct.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800235 let guard = format!("CXXBRIDGE1_STRUCT_{}", strct.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700236 writeln!(out, "#ifndef {}", guard);
237 writeln!(out, "#define {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400238 for line in strct.doc.to_string().lines() {
239 writeln!(out, "//{}", line);
240 }
David Tolnay17a934c2020-11-02 00:40:04 -0800241 writeln!(out, "struct {} final {{", strct.name.cxx);
David Tolnay84389352020-11-27 17:12:10 -0800242
David Tolnay7db73692019-10-20 14:51:12 -0400243 for field in &strct.fields {
David Tolnay5573e522020-12-31 11:07:38 -0800244 for line in field.doc.to_string().lines() {
245 writeln!(out, " //{}", line);
246 }
David Tolnay7db73692019-10-20 14:51:12 -0400247 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700248 write_type_space(out, &field.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800249 writeln!(out, "{};", field.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400250 }
David Tolnay84389352020-11-27 17:12:10 -0800251
David Tolnay5554ebd2020-12-10 11:34:41 -0800252 writeln!(out);
David Tolnay84389352020-11-27 17:12:10 -0800253
David Tolnay102c7ea2020-11-08 18:58:09 -0800254 for method in methods {
255 write!(out, " ");
256 let sig = &method.sig;
257 let local_name = method.name.cxx.to_string();
258 write_rust_function_shim_decl(out, &local_name, sig, false);
259 writeln!(out, ";");
260 }
David Tolnay84389352020-11-27 17:12:10 -0800261
David Tolnayb960ed22020-11-27 14:34:30 -0800262 if operator_eq {
263 writeln!(
264 out,
265 " bool operator==(const {} &) const noexcept;",
266 strct.name.cxx,
267 );
268 writeln!(
269 out,
270 " bool operator!=(const {} &) const noexcept;",
271 strct.name.cxx,
272 );
273 }
David Tolnay84389352020-11-27 17:12:10 -0800274
275 if operator_ord {
276 writeln!(
277 out,
278 " bool operator<(const {} &) const noexcept;",
279 strct.name.cxx,
280 );
281 writeln!(
282 out,
283 " bool operator<=(const {} &) const noexcept;",
284 strct.name.cxx,
285 );
286 writeln!(
287 out,
288 " bool operator>(const {} &) const noexcept;",
289 strct.name.cxx,
290 );
291 writeln!(
292 out,
293 " bool operator>=(const {} &) const noexcept;",
294 strct.name.cxx,
295 );
296 }
297
David Tolnay5554ebd2020-12-10 11:34:41 -0800298 out.include.type_traits = true;
299 writeln!(out, " using IsRelocatable = ::std::true_type;");
300
David Tolnay7db73692019-10-20 14:51:12 -0400301 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700302 writeln!(out, "#endif // {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400303}
304
David Tolnayed6ba4a2021-01-01 14:59:40 -0800305fn write_struct_decl(out: &mut OutFile, ident: &Pair) {
306 writeln!(out, "struct {};", ident.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400307}
308
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800309fn write_enum_decl(out: &mut OutFile, enm: &Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800310 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800311 write_atom(out, enm.repr);
312 writeln!(out, ";");
313}
314
David Tolnay8faec772020-11-02 00:18:19 -0800315fn write_struct_using(out: &mut OutFile, ident: &Pair) {
316 writeln!(out, "using {} = {};", ident.cxx, ident.to_fully_qualified());
David Tolnay8861bee2020-01-20 18:39:24 -0800317}
318
David Tolnay8d067de2020-12-26 16:18:23 -0800319fn write_opaque_type<'a>(out: &mut OutFile<'a>, ety: &'a ExternType, methods: &[&ExternFn]) {
David Tolnay17a934c2020-11-02 00:40:04 -0800320 out.set_namespace(&ety.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800321 let guard = format!("CXXBRIDGE1_STRUCT_{}", ety.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700322 writeln!(out, "#ifndef {}", guard);
323 writeln!(out, "#define {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700324 for line in ety.doc.to_string().lines() {
325 writeln!(out, "//{}", line);
326 }
David Tolnay6ba262f2020-12-26 22:23:50 -0800327
David Tolnay365fc7c2020-11-25 16:08:13 -0800328 out.builtin.opaque = true;
David Tolnay6ba262f2020-12-26 22:23:50 -0800329 writeln!(
David Tolnay7c06b862020-11-25 16:59:09 -0800330 out,
331 "struct {} final : public ::rust::Opaque {{",
332 ety.name.cxx,
333 );
David Tolnay6ba262f2020-12-26 22:23:50 -0800334
Joel Galenson968738f2020-04-15 14:19:33 -0700335 for method in methods {
David Tolnay6ba262f2020-12-26 22:23:50 -0800336 write!(out, " ");
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700337 let sig = &method.sig;
David Tolnay17a934c2020-11-02 00:40:04 -0800338 let local_name = method.name.cxx.to_string();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700339 write_rust_function_shim_decl(out, &local_name, sig, false);
David Tolnay6ba262f2020-12-26 22:23:50 -0800340 writeln!(out, ";");
David Tolnay8d067de2020-12-26 16:18:23 -0800341 }
David Tolnay6ba262f2020-12-26 22:23:50 -0800342
David Tolnay8f940cf2021-01-02 18:45:00 -0800343 writeln!(out, " ~{}() = delete;", ety.name.cxx);
344 writeln!(out);
David Tolnay6ba262f2020-12-26 22:23:50 -0800345
David Tolnayee6ecfc2020-12-26 21:54:37 -0800346 out.builtin.layout = true;
David Tolnay6ba262f2020-12-26 22:23:50 -0800347 out.include.cstddef = true;
348 writeln!(out, "private:");
David Tolnayee6ecfc2020-12-26 21:54:37 -0800349 writeln!(out, " friend ::rust::layout;");
David Tolnay6ba262f2020-12-26 22:23:50 -0800350 writeln!(out, " struct layout {{");
351 writeln!(out, " static ::std::size_t size() noexcept;");
352 writeln!(out, " static ::std::size_t align() noexcept;");
353 writeln!(out, " }};");
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700354 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700355 writeln!(out, "#endif // {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700356}
357
David Tolnay0b9b9f82020-11-01 20:41:00 -0800358fn write_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800359 out.set_namespace(&enm.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800360 let guard = format!("CXXBRIDGE1_ENUM_{}", enm.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700361 writeln!(out, "#ifndef {}", guard);
362 writeln!(out, "#define {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700363 for line in enm.doc.to_string().lines() {
364 writeln!(out, "//{}", line);
365 }
David Tolnay17a934c2020-11-02 00:40:04 -0800366 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700367 write_atom(out, enm.repr);
368 writeln!(out, " {{");
Joel Galensonc03402a2020-04-23 17:31:09 -0700369 for variant in &enm.variants {
David Tolnay4486f722020-12-30 00:01:26 -0800370 for line in variant.doc.to_string().lines() {
371 writeln!(out, " //{}", line);
372 }
David Tolnaye6f62142020-12-21 16:00:41 -0800373 writeln!(out, " {} = {},", variant.name.cxx, variant.discriminant);
Joel Galensonc03402a2020-04-23 17:31:09 -0700374 }
375 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700376 writeln!(out, "#endif // {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700377}
378
David Tolnay0b9b9f82020-11-01 20:41:00 -0800379fn check_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800380 out.set_namespace(&enm.name.namespace);
David Tolnay06711bc2020-11-19 19:25:14 -0800381 out.include.type_traits = true;
382 writeln!(
383 out,
384 "static_assert(::std::is_enum<{}>::value, \"expected enum\");",
385 enm.name.cxx,
386 );
David Tolnay17a934c2020-11-02 00:40:04 -0800387 write!(out, "static_assert(sizeof({}) == sizeof(", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700388 write_atom(out, enm.repr);
389 writeln!(out, "), \"incorrect size\");");
Joel Galenson0f654ff2020-05-04 20:04:21 -0700390 for variant in &enm.variants {
David Tolnayf6a89f22020-05-10 23:39:27 -0700391 write!(out, "static_assert(static_cast<");
392 write_atom(out, enm.repr);
Joel Galenson0f654ff2020-05-04 20:04:21 -0700393 writeln!(
394 out,
David Tolnayf6a89f22020-05-10 23:39:27 -0700395 ">({}::{}) == {}, \"disagrees with the value in #[cxx::bridge]\");",
David Tolnaye6f62142020-12-21 16:00:41 -0800396 enm.name.cxx, variant.name.cxx, variant.discriminant,
Joel Galenson0f654ff2020-05-04 20:04:21 -0700397 );
Joel Galenson0f654ff2020-05-04 20:04:21 -0700398 }
Joel Galenson905eb2e2020-05-04 14:58:14 -0700399}
400
David Tolnay5b1d8632020-12-20 22:05:11 -0800401fn check_trivial_extern_type(out: &mut OutFile, alias: &TypeAlias, reasons: &[TrivialReason]) {
David Tolnay174bd952020-11-02 09:23:12 -0800402 // NOTE: The following static assertion is just nice-to-have and not
David Tolnayfd0034e2020-10-04 13:15:34 -0700403 // necessary for soundness. That's because triviality is always declared by
404 // the user in the form of an unsafe impl of cxx::ExternType:
405 //
406 // unsafe impl ExternType for MyType {
407 // type Id = cxx::type_id!("...");
408 // type Kind = cxx::kind::Trivial;
409 // }
410 //
411 // Since the user went on the record with their unsafe impl to unsafely
412 // claim they KNOW that the type is trivial, it's fine for that to be on
David Tolnay174bd952020-11-02 09:23:12 -0800413 // them if that were wrong. However, in practice correctly reasoning about
414 // the relocatability of C++ types is challenging, particularly if the type
415 // definition were to change over time, so for now we add this check.
David Tolnayfd0034e2020-10-04 13:15:34 -0700416 //
David Tolnay174bd952020-11-02 09:23:12 -0800417 // There may be legitimate reasons to opt out of this assertion for support
418 // of types that the programmer knows are soundly Rust-movable despite not
419 // being recognized as such by the C++ type system due to a move constructor
420 // or destructor. To opt out of the relocatability check, they need to do
421 // one of the following things in any header used by `include!` in their
422 // bridge.
423 //
424 // --- if they define the type:
425 // struct MyType {
426 // ...
427 // + using IsRelocatable = std::true_type;
428 // };
429 //
430 // --- otherwise:
431 // + template <>
432 // + struct rust::IsRelocatable<MyType> : std::true_type {};
433 //
David Tolnayfd0034e2020-10-04 13:15:34 -0700434
David Tolnay5b1d8632020-12-20 22:05:11 -0800435 let id = alias.name.to_fully_qualified();
David Tolnay174bd952020-11-02 09:23:12 -0800436 out.builtin.relocatable = true;
David Tolnay5b1d8632020-12-20 22:05:11 -0800437 writeln!(out, "static_assert(");
438 writeln!(out, " ::rust::IsRelocatable<{}>::value,", id);
439 writeln!(
440 out,
441 " \"type {} should be trivially move constructible and trivially destructible in C++ to be used as {} in Rust\");",
442 id.trim_start_matches("::"),
443 trivial::as_what(&alias.name, reasons),
444 );
Adrian Taylor405e8742020-09-25 14:01:25 -0700445}
446
David Tolnayb960ed22020-11-27 14:34:30 -0800447fn write_struct_operator_decls<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
448 out.set_namespace(&strct.name.namespace);
449 out.begin_block(Block::ExternC);
450
451 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800452 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800453 writeln!(
454 out,
455 "bool {}(const {1} &, const {1} &) noexcept;",
456 link_name, strct.name.cxx,
457 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800458
459 if !derive::contains(&strct.derives, Trait::Eq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800460 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800461 writeln!(
462 out,
463 "bool {}(const {1} &, const {1} &) noexcept;",
464 link_name, strct.name.cxx,
465 );
466 }
David Tolnayb960ed22020-11-27 14:34:30 -0800467 }
468
David Tolnay84389352020-11-27 17:12:10 -0800469 if derive::contains(&strct.derives, Trait::PartialOrd) {
David Tolnaya05f9402020-11-27 17:44:05 -0800470 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800471 writeln!(
472 out,
473 "bool {}(const {1} &, const {1} &) noexcept;",
474 link_name, strct.name.cxx,
475 );
476
David Tolnaya05f9402020-11-27 17:44:05 -0800477 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800478 writeln!(
479 out,
480 "bool {}(const {1} &, const {1} &) noexcept;",
481 link_name, strct.name.cxx,
482 );
483
484 if !derive::contains(&strct.derives, Trait::Ord) {
David Tolnaya05f9402020-11-27 17:44:05 -0800485 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800486 writeln!(
487 out,
488 "bool {}(const {1} &, const {1} &) noexcept;",
489 link_name, strct.name.cxx,
490 );
491
David Tolnaya05f9402020-11-27 17:44:05 -0800492 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800493 writeln!(
494 out,
495 "bool {}(const {1} &, const {1} &) noexcept;",
496 link_name, strct.name.cxx,
497 );
498 }
499 }
500
David Tolnay7da38202020-11-27 17:36:16 -0800501 if derive::contains(&strct.derives, Trait::Hash) {
David Tolnay12320e12020-12-09 23:09:36 -0800502 out.include.cstddef = true;
David Tolnay7da38202020-11-27 17:36:16 -0800503 let link_name = mangle::operator(&strct.name, "hash");
504 writeln!(
505 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800506 "::std::size_t {}(const {} &) noexcept;",
David Tolnay7da38202020-11-27 17:36:16 -0800507 link_name, strct.name.cxx,
508 );
509 }
510
David Tolnayb960ed22020-11-27 14:34:30 -0800511 out.end_block(Block::ExternC);
512}
513
514fn write_struct_operators<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
515 if out.header {
516 return;
517 }
518
519 out.set_namespace(&strct.name.namespace);
520
521 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnayb960ed22020-11-27 14:34:30 -0800522 out.next_section();
523 writeln!(
524 out,
525 "bool {0}::operator==(const {0} &rhs) const noexcept {{",
526 strct.name.cxx,
527 );
David Tolnaya05f9402020-11-27 17:44:05 -0800528 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800529 writeln!(out, " return {}(*this, rhs);", link_name);
530 writeln!(out, "}}");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800531
David Tolnayb960ed22020-11-27 14:34:30 -0800532 out.next_section();
533 writeln!(
534 out,
535 "bool {0}::operator!=(const {0} &rhs) const noexcept {{",
536 strct.name.cxx,
537 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800538 if derive::contains(&strct.derives, Trait::Eq) {
539 writeln!(out, " return !(*this == rhs);");
540 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800541 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800542 writeln!(out, " return {}(*this, rhs);", link_name);
543 }
David Tolnayb960ed22020-11-27 14:34:30 -0800544 writeln!(out, "}}");
545 }
David Tolnay84389352020-11-27 17:12:10 -0800546
547 if derive::contains(&strct.derives, Trait::PartialOrd) {
548 out.next_section();
549 writeln!(
550 out,
551 "bool {0}::operator<(const {0} &rhs) const noexcept {{",
552 strct.name.cxx,
553 );
David Tolnaya05f9402020-11-27 17:44:05 -0800554 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800555 writeln!(out, " return {}(*this, rhs);", link_name);
556 writeln!(out, "}}");
557
558 out.next_section();
559 writeln!(
560 out,
561 "bool {0}::operator<=(const {0} &rhs) const noexcept {{",
562 strct.name.cxx,
563 );
David Tolnaya05f9402020-11-27 17:44:05 -0800564 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800565 writeln!(out, " return {}(*this, rhs);", link_name);
566 writeln!(out, "}}");
567
568 out.next_section();
569 writeln!(
570 out,
571 "bool {0}::operator>(const {0} &rhs) const noexcept {{",
572 strct.name.cxx,
573 );
574 if derive::contains(&strct.derives, Trait::Ord) {
575 writeln!(out, " return !(*this <= rhs);");
576 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800577 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800578 writeln!(out, " return {}(*this, rhs);", link_name);
579 }
580 writeln!(out, "}}");
581
582 out.next_section();
583 writeln!(
584 out,
585 "bool {0}::operator>=(const {0} &rhs) const noexcept {{",
586 strct.name.cxx,
587 );
588 if derive::contains(&strct.derives, Trait::Ord) {
589 writeln!(out, " return !(*this < rhs);");
590 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800591 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800592 writeln!(out, " return {}(*this, rhs);", link_name);
593 }
594 writeln!(out, "}}");
595 }
David Tolnayb960ed22020-11-27 14:34:30 -0800596}
597
David Tolnay358bc4b2020-12-26 22:37:57 -0800598fn write_opaque_type_layout_decls<'a>(out: &mut OutFile<'a>, ety: &'a ExternType) {
599 out.set_namespace(&ety.name.namespace);
600 out.begin_block(Block::ExternC);
601
602 let link_name = mangle::operator(&ety.name, "sizeof");
603 writeln!(out, "::std::size_t {}() noexcept;", link_name);
604
605 let link_name = mangle::operator(&ety.name, "alignof");
606 writeln!(out, "::std::size_t {}() noexcept;", link_name);
607
608 out.end_block(Block::ExternC);
609}
610
611fn write_opaque_type_layout<'a>(out: &mut OutFile<'a>, ety: &'a ExternType) {
612 if out.header {
613 return;
614 }
615
616 out.set_namespace(&ety.name.namespace);
617
618 out.next_section();
619 let link_name = mangle::operator(&ety.name, "sizeof");
620 writeln!(
621 out,
622 "::std::size_t {}::layout::size() noexcept {{",
623 ety.name.cxx,
624 );
625 writeln!(out, " return {}();", link_name);
626 writeln!(out, "}}");
627
628 out.next_section();
629 let link_name = mangle::operator(&ety.name, "alignof");
630 writeln!(
631 out,
632 "::std::size_t {}::layout::align() noexcept {{",
633 ety.name.cxx,
634 );
635 writeln!(out, " return {}();", link_name);
636 writeln!(out, "}}");
637}
638
David Tolnay1eadd262020-12-29 16:42:08 -0800639fn begin_function_definition(out: &mut OutFile) {
640 if let Some(annotation) = &out.opt.cxx_impl_annotations {
641 write!(out, "{} ", annotation);
642 }
643}
644
David Tolnay0b9b9f82020-11-01 20:41:00 -0800645fn write_cxx_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay04770742020-11-01 13:50:50 -0800646 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800647 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800648 out.begin_block(Block::ExternC);
David Tolnay1eadd262020-12-29 16:42:08 -0800649 begin_function_definition(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 Tolnayd68dfa82020-10-31 16:01:24 -0700652 write!(out, "::rust::repr::PtrLen ");
David Tolnayebef4a22020-03-17 15:33:47 -0700653 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700654 write_extern_return_type_space(out, &efn.ret);
David Tolnayebef4a22020-03-17 15:33:47 -0700655 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700656 let mangled = mangle::extern_fn(efn, out.types);
David Tolnay3caa50a2020-04-19 21:25:34 -0700657 write!(out, "{}(", mangled);
David Tolnaye439c772020-04-20 00:23:55 -0700658 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800659 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700660 write!(out, "const ");
661 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700662 write!(
663 out,
664 "{} &self",
David Tolnay1e5fe232021-01-01 18:11:40 -0800665 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700666 );
Joel Galenson3d4f6122020-04-07 15:54:05 -0700667 }
David Tolnay7db73692019-10-20 14:51:12 -0400668 for (i, arg) in efn.args.iter().enumerate() {
Joel Galenson3d4f6122020-04-07 15:54:05 -0700669 if i > 0 || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400670 write!(out, ", ");
671 }
David Tolnaya46a2372020-03-06 10:03:48 -0800672 if arg.ty == RustString {
673 write!(out, "const ");
David Tolnay313b10e2020-04-25 16:30:51 -0700674 } else if let Type::RustVec(_) = arg.ty {
675 write!(out, "const ");
David Tolnaya46a2372020-03-06 10:03:48 -0800676 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700677 write_extern_arg(out, arg);
David Tolnay7db73692019-10-20 14:51:12 -0400678 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700679 let indirect_return = indirect_return(efn, out.types);
David Tolnay7db73692019-10-20 14:51:12 -0400680 if indirect_return {
myronahne3b78ea2020-05-23 01:08:13 +0700681 if !efn.args.is_empty() || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400682 write!(out, ", ");
683 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700684 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400685 write!(out, "*return$");
686 }
687 writeln!(out, ") noexcept {{");
688 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700689 write_return_type(out, &efn.ret);
Joel Galenson3d4f6122020-04-07 15:54:05 -0700690 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800691 None => write!(out, "(*{}$)(", efn.name.rust),
Adrian Taylorc8713432020-10-21 18:20:55 -0700692 Some(receiver) => write!(
693 out,
694 "({}::*{}$)(",
David Tolnay1e5fe232021-01-01 18:11:40 -0800695 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800696 efn.name.rust,
Adrian Taylorc8713432020-10-21 18:20:55 -0700697 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700698 }
David Tolnay7db73692019-10-20 14:51:12 -0400699 for (i, arg) in efn.args.iter().enumerate() {
700 if i > 0 {
701 write!(out, ", ");
702 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700703 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400704 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700705 write!(out, ")");
David Tolnay4e7123f2020-04-19 21:11:37 -0700706 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800707 if !receiver.mutable {
David Tolnay4e7123f2020-04-19 21:11:37 -0700708 write!(out, " const");
709 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700710 }
711 write!(out, " = ");
712 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800713 None => write!(out, "{}", efn.name.to_fully_qualified()),
Adrian Taylorc8713432020-10-21 18:20:55 -0700714 Some(receiver) => write!(
715 out,
716 "&{}::{}",
David Tolnay1e5fe232021-01-01 18:11:40 -0800717 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800718 efn.name.cxx,
Adrian Taylorc8713432020-10-21 18:20:55 -0700719 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700720 }
721 writeln!(out, ";");
David Tolnay7db73692019-10-20 14:51:12 -0400722 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700723 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700724 out.builtin.ptr_len = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700725 out.builtin.trycatch = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700726 writeln!(out, "::rust::repr::PtrLen throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700727 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700728 writeln!(out, " [&] {{");
729 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700730 }
David Tolnay7db73692019-10-20 14:51:12 -0400731 if indirect_return {
David Tolnay0ecd05a2020-07-29 16:32:03 -0700732 out.include.new = true;
David Tolnay7db73692019-10-20 14:51:12 -0400733 write!(out, "new (return$) ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700734 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400735 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700736 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400737 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700738 }
David Tolnayfe77f332021-01-02 17:39:51 -0800739 match &efn.ret {
740 Some(Type::Ref(_)) => write!(out, "&"),
David Tolnaye5cd1482021-01-02 21:04:32 -0800741 Some(Type::Str(_)) if !indirect_return => {
742 out.builtin.rust_str_repr = true;
743 write!(out, "::rust::impl<::rust::Str>::repr(");
744 }
745 Some(ty @ Type::SliceRef(_)) if !indirect_return => {
746 out.builtin.rust_slice_repr = true;
747 write!(out, "::rust::impl<");
748 write_type(out, ty);
749 write!(out, ">::repr(");
750 }
David Tolnayfe77f332021-01-02 17:39:51 -0800751 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400752 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700753 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800754 None => write!(out, "{}$(", efn.name.rust),
755 Some(_) => write!(out, "(self.*{}$)(", efn.name.rust),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700756 }
David Tolnay7db73692019-10-20 14:51:12 -0400757 for (i, arg) in efn.args.iter().enumerate() {
758 if i > 0 {
759 write!(out, ", ");
760 }
761 if let Type::RustBox(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700762 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800763 write!(out, "::from_raw({})", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400764 } else if let Type::UniquePtr(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700765 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800766 write!(out, "({})", arg.name.cxx);
David Tolnaya46a2372020-03-06 10:03:48 -0800767 } else if arg.ty == RustString {
David Tolnay74d6d512020-10-31 22:22:03 -0700768 out.builtin.unsafe_bitcopy = true;
David Tolnaycc3767f2020-03-06 10:41:51 -0800769 write!(
770 out,
771 "::rust::String(::rust::unsafe_bitcopy, *{})",
David Tolnay84ed6ad2021-01-01 15:30:14 -0800772 arg.name.cxx,
David Tolnaycc3767f2020-03-06 10:41:51 -0800773 );
David Tolnay313b10e2020-04-25 16:30:51 -0700774 } else if let Type::RustVec(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700775 out.builtin.unsafe_bitcopy = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700776 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800777 write!(out, "(::rust::unsafe_bitcopy, *{})", arg.name.cxx);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700778 } else if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700779 out.include.utility = true;
David Tolnay84ed6ad2021-01-01 15:30:14 -0800780 write!(out, "::std::move(*{})", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400781 } else {
David Tolnay84ed6ad2021-01-01 15:30:14 -0800782 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400783 }
784 }
785 write!(out, ")");
786 match &efn.ret {
787 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
788 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnaye5cd1482021-01-02 21:04:32 -0800789 Some(Type::Str(_)) | Some(Type::SliceRef(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400790 _ => {}
791 }
792 if indirect_return {
793 write!(out, ")");
794 }
795 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700796 if efn.throws {
797 out.include.cstring = true;
David Tolnay1f010c62020-11-01 20:27:46 -0800798 out.builtin.exception = true;
David Tolnay5d121442020-03-17 22:14:40 -0700799 writeln!(out, " throw$.ptr = nullptr;");
800 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700801 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700802 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700803 writeln!(
804 out,
David Tolnayc5629f02020-11-23 18:32:46 -0800805 " throw$.ptr = const_cast<char *>(::cxxbridge1$exception(catch$, throw$.len));",
David Tolnayebef4a22020-03-17 15:33:47 -0700806 );
David Tolnay5d121442020-03-17 22:14:40 -0700807 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700808 writeln!(out, " return throw$;");
809 }
David Tolnay7db73692019-10-20 14:51:12 -0400810 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700811 for arg in &efn.args {
812 if let Type::Fn(f) = &arg.ty {
David Tolnay84ed6ad2021-01-01 15:30:14 -0800813 let var = &arg.name;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700814 write_function_pointer_trampoline(out, efn, var, f);
David Tolnay75dca2e2020-03-25 20:17:52 -0700815 }
816 }
David Tolnayca563ee2020-11-01 20:12:27 -0800817 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700818}
819
David Tolnay84ed6ad2021-01-01 15:30:14 -0800820fn write_function_pointer_trampoline(out: &mut OutFile, efn: &ExternFn, var: &Pair, f: &Signature) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700821 let r_trampoline = mangle::r_trampoline(efn, var, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700822 let indirect_call = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700823 write_rust_function_decl_impl(out, &r_trampoline, f, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700824
825 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700826 let c_trampoline = mangle::c_trampoline(efn, var, out.types).to_string();
827 write_rust_function_shim_impl(out, &c_trampoline, f, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400828}
829
David Tolnay0b9b9f82020-11-01 20:41:00 -0800830fn write_rust_function_decl<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800831 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800832 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700833 let link_name = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700834 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700835 write_rust_function_decl_impl(out, &link_name, efn, indirect_call);
David Tolnayca563ee2020-11-01 20:12:27 -0800836 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700837}
838
839fn write_rust_function_decl_impl(
840 out: &mut OutFile,
David Tolnay891061b2020-04-19 22:42:33 -0700841 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700842 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700843 indirect_call: bool,
844) {
David Tolnay04770742020-11-01 13:50:50 -0800845 out.next_section();
David Tolnay75dca2e2020-03-25 20:17:52 -0700846 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700847 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700848 write!(out, "::rust::repr::PtrLen ");
David Tolnay1e548172020-03-16 13:37:09 -0700849 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700850 write_extern_return_type_space(out, &sig.ret);
David Tolnay1e548172020-03-16 13:37:09 -0700851 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700852 write!(out, "{}(", link_name);
853 let mut needs_comma = false;
David Tolnaye439c772020-04-20 00:23:55 -0700854 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800855 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700856 write!(out, "const ");
857 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700858 write!(
859 out,
860 "{} &self",
David Tolnay1e5fe232021-01-01 18:11:40 -0800861 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700862 );
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700863 needs_comma = true;
864 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700865 for arg in &sig.args {
866 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400867 write!(out, ", ");
868 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700869 write_extern_arg(out, arg);
David Tolnay75dca2e2020-03-25 20:17:52 -0700870 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400871 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700872 if indirect_return(sig, out.types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700873 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400874 write!(out, ", ");
875 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700876 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400877 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700878 needs_comma = true;
879 }
880 if indirect_call {
881 if needs_comma {
882 write!(out, ", ");
883 }
884 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400885 }
886 writeln!(out, ") noexcept;");
887}
888
David Tolnay0b9b9f82020-11-01 20:41:00 -0800889fn write_rust_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800890 out.set_namespace(&efn.name.namespace);
David Tolnay7db73692019-10-20 14:51:12 -0400891 for line in efn.doc.to_string().lines() {
892 writeln!(out, "//{}", line);
893 }
David Tolnaya73853b2020-04-20 01:19:56 -0700894 let local_name = match &efn.sig.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800895 None => efn.name.cxx.to_string(),
David Tolnay1e5fe232021-01-01 18:11:40 -0800896 Some(receiver) => format!(
897 "{}::{}",
898 out.types.resolve(&receiver.ty).name.cxx,
899 efn.name.cxx,
900 ),
David Tolnaya73853b2020-04-20 01:19:56 -0700901 };
David Tolnaya7c2ea12020-10-30 21:32:53 -0700902 let invoke = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700903 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700904 write_rust_function_shim_impl(out, &local_name, efn, &invoke, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700905}
906
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700907fn write_rust_function_shim_decl(
David Tolnay75dca2e2020-03-25 20:17:52 -0700908 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700909 local_name: &str,
David Tolnay75dca2e2020-03-25 20:17:52 -0700910 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700911 indirect_call: bool,
912) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700913 write_return_type(out, &sig.ret);
David Tolnay75dca2e2020-03-25 20:17:52 -0700914 write!(out, "{}(", local_name);
915 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400916 if i > 0 {
917 write!(out, ", ");
918 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700919 write_type_space(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800920 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400921 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700922 if indirect_call {
923 if !sig.args.is_empty() {
924 write!(out, ", ");
925 }
926 write!(out, "void *extern$");
927 }
David Tolnay1e548172020-03-16 13:37:09 -0700928 write!(out, ")");
David Tolnay86710612020-04-20 00:30:32 -0700929 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800930 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700931 write!(out, " const");
932 }
933 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700934 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700935 write!(out, " noexcept");
936 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700937}
938
939fn write_rust_function_shim_impl(
940 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700941 local_name: &str,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700942 sig: &Signature,
David Tolnay891061b2020-04-19 22:42:33 -0700943 invoke: &Symbol,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700944 indirect_call: bool,
945) {
946 if out.header && sig.receiver.is_some() {
947 // We've already defined this inside the struct.
948 return;
949 }
David Tolnayb478fcb2020-12-29 16:48:02 -0800950 if !out.header {
951 begin_function_definition(out);
952 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700953 write_rust_function_shim_decl(out, local_name, sig, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400954 if out.header {
955 writeln!(out, ";");
David Tolnay439cde22020-04-20 00:46:25 -0700956 return;
David Tolnay7db73692019-10-20 14:51:12 -0400957 }
David Tolnay439cde22020-04-20 00:46:25 -0700958 writeln!(out, " {{");
959 for arg in &sig.args {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700960 if arg.ty != RustString && out.types.needs_indirect_abi(&arg.ty) {
David Tolnay439cde22020-04-20 00:46:25 -0700961 out.include.utility = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700962 out.builtin.manually_drop = true;
David Tolnay439cde22020-04-20 00:46:25 -0700963 write!(out, " ::rust::ManuallyDrop<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700964 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800965 writeln!(out, "> {}$(::std::move({0}));", arg.name.cxx);
David Tolnay439cde22020-04-20 00:46:25 -0700966 }
967 }
968 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700969 let indirect_return = indirect_return(sig, out.types);
David Tolnay439cde22020-04-20 00:46:25 -0700970 if indirect_return {
David Tolnay74d6d512020-10-31 22:22:03 -0700971 out.builtin.maybe_uninit = true;
David Tolnay439cde22020-04-20 00:46:25 -0700972 write!(out, "::rust::MaybeUninit<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700973 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay439cde22020-04-20 00:46:25 -0700974 writeln!(out, "> return$;");
975 write!(out, " ");
976 } else if let Some(ret) = &sig.ret {
977 write!(out, "return ");
978 match ret {
979 Type::RustBox(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700980 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700981 write!(out, "::from_raw(");
982 }
983 Type::UniquePtr(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700984 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -0700985 write!(out, "(");
986 }
987 Type::Ref(_) => write!(out, "*"),
David Tolnaye5cd1482021-01-02 21:04:32 -0800988 Type::Str(_) => {
989 out.builtin.rust_str_new_unchecked = true;
990 write!(out, "::rust::impl<::rust::Str>::new_unchecked(");
991 }
992 Type::SliceRef(_) => {
993 out.builtin.rust_slice_new = true;
994 write!(out, "::rust::impl<");
995 write_type(out, ret);
996 write!(out, ">::slice(");
997 }
David Tolnay439cde22020-04-20 00:46:25 -0700998 _ => {}
999 }
1000 }
1001 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -07001002 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001003 write!(out, "::rust::repr::PtrLen error$ = ");
David Tolnay439cde22020-04-20 00:46:25 -07001004 }
1005 write!(out, "{}(", invoke);
David Tolnay9d840362020-11-10 08:50:40 -08001006 let mut needs_comma = false;
David Tolnay439cde22020-04-20 00:46:25 -07001007 if sig.receiver.is_some() {
1008 write!(out, "*this");
David Tolnay9d840362020-11-10 08:50:40 -08001009 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001010 }
David Tolnay9d840362020-11-10 08:50:40 -08001011 for arg in &sig.args {
1012 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001013 write!(out, ", ");
1014 }
David Tolnay681f5c82021-01-01 22:56:27 -08001015 if out.types.needs_indirect_abi(&arg.ty) {
1016 write!(out, "&");
David Tolnay439cde22020-04-20 00:46:25 -07001017 }
David Tolnay84ed6ad2021-01-01 15:30:14 -08001018 write!(out, "{}", arg.name.cxx);
David Tolnay439cde22020-04-20 00:46:25 -07001019 match &arg.ty {
1020 Type::RustBox(_) => write!(out, ".into_raw()"),
1021 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001022 ty if ty != RustString && out.types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay439cde22020-04-20 00:46:25 -07001023 _ => {}
1024 }
David Tolnay9d840362020-11-10 08:50:40 -08001025 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001026 }
1027 if indirect_return {
David Tolnay9d840362020-11-10 08:50:40 -08001028 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001029 write!(out, ", ");
1030 }
1031 write!(out, "&return$.value");
David Tolnay9d840362020-11-10 08:50:40 -08001032 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001033 }
1034 if indirect_call {
David Tolnay9d840362020-11-10 08:50:40 -08001035 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001036 write!(out, ", ");
1037 }
1038 write!(out, "extern$");
1039 }
1040 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -04001041 if !indirect_return {
1042 if let Some(ret) = &sig.ret {
David Tolnaye5cd1482021-01-02 21:04:32 -08001043 if let Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::SliceRef(_) = ret {
1044 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -04001045 }
David Tolnay439cde22020-04-20 00:46:25 -07001046 }
1047 }
1048 writeln!(out, ";");
1049 if sig.throws {
David Tolnay74d6d512020-10-31 22:22:03 -07001050 out.builtin.rust_error = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001051 writeln!(out, " if (error$.ptr) {{");
David Tolnay84ddf9e2020-10-31 15:36:48 -07001052 writeln!(out, " throw ::rust::impl<::rust::Error>::error(error$);");
David Tolnay439cde22020-04-20 00:46:25 -07001053 writeln!(out, " }}");
1054 }
1055 if indirect_return {
1056 out.include.utility = true;
1057 writeln!(out, " return ::std::move(return$.value);");
1058 }
1059 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001060}
1061
David Tolnaya7c2ea12020-10-30 21:32:53 -07001062fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001063 match ty {
1064 None => write!(out, "void "),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001065 Some(ty) => write_type_space(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001066 }
1067}
1068
David Tolnay75dca2e2020-03-25 20:17:52 -07001069fn indirect_return(sig: &Signature, types: &Types) -> bool {
1070 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -07001071 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -07001072 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -07001073}
1074
David Tolnaya7c2ea12020-10-30 21:32:53 -07001075fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
David Tolnay99642622020-03-25 13:07:35 -07001076 match ty {
1077 Type::RustBox(ty) | Type::UniquePtr(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001078 write_type_space(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001079 write!(out, "*");
1080 }
1081 Type::Ref(ty) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001082 if !ty.mutable {
David Tolnay99642622020-03-25 13:07:35 -07001083 write!(out, "const ");
1084 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001085 write_type(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001086 write!(out, " *");
1087 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001088 _ => write_type(out, ty),
David Tolnay99642622020-03-25 13:07:35 -07001089 }
1090}
1091
David Tolnaya7c2ea12020-10-30 21:32:53 -07001092fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
1093 write_indirect_return_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001094 match ty {
1095 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
David Tolnay73b72642020-11-25 17:44:05 -08001096 Type::Str(_) | Type::SliceRef(_) => write!(out, " "),
David Tolnay99642622020-03-25 13:07:35 -07001097 _ => write_space_after_type(out, ty),
1098 }
1099}
1100
David Tolnaya7c2ea12020-10-30 21:32:53 -07001101fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001102 match ty {
1103 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001104 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001105 write!(out, "*");
1106 }
David Tolnay4a441222020-01-25 16:24:27 -08001107 Some(Type::Ref(ty)) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001108 if !ty.mutable {
David Tolnay4a441222020-01-25 16:24:27 -08001109 write!(out, "const ");
1110 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001111 write_type(out, &ty.inner);
David Tolnay4a441222020-01-25 16:24:27 -08001112 write!(out, " *");
1113 }
David Tolnaye5cd1482021-01-02 21:04:32 -08001114 Some(Type::Str(_)) | Some(Type::SliceRef(_)) => {
David Tolnayfe77f332021-01-02 17:39:51 -08001115 out.builtin.repr_fat = true;
David Tolnaye5cd1482021-01-02 21:04:32 -08001116 write!(out, "::rust::repr::Fat ");
David Tolnayfe77f332021-01-02 17:39:51 -08001117 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001118 Some(ty) if out.types.needs_indirect_abi(ty) => write!(out, "void "),
1119 _ => write_return_type(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001120 }
1121}
1122
David Tolnaya7c2ea12020-10-30 21:32:53 -07001123fn write_extern_arg(out: &mut OutFile, arg: &Var) {
David Tolnay7db73692019-10-20 14:51:12 -04001124 match &arg.ty {
David Tolnay4377a9e2020-04-24 15:20:26 -07001125 Type::RustBox(ty) | Type::UniquePtr(ty) | Type::CxxVector(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001126 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001127 write!(out, "*");
1128 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001129 _ => write_type_space(out, &arg.ty),
David Tolnay7db73692019-10-20 14:51:12 -04001130 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001131 if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay7db73692019-10-20 14:51:12 -04001132 write!(out, "*");
1133 }
David Tolnay84ed6ad2021-01-01 15:30:14 -08001134 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -04001135}
1136
David Tolnaya7c2ea12020-10-30 21:32:53 -07001137fn write_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001138 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -07001139 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnayf6a89f22020-05-10 23:39:27 -07001140 Some(atom) => write_atom(out, atom),
David Tolnay1e5fe232021-01-01 18:11:40 -08001141 None => write!(
1142 out,
1143 "{}",
1144 out.types.resolve(ident).name.to_fully_qualified(),
1145 ),
David Tolnay7db73692019-10-20 14:51:12 -04001146 },
1147 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -08001148 write!(out, "::rust::Box<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001149 write_type(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001150 write!(out, ">");
1151 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001152 Type::RustVec(ty) => {
1153 write!(out, "::rust::Vec<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001154 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001155 write!(out, ">");
1156 }
David Tolnay7db73692019-10-20 14:51:12 -04001157 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -08001158 write!(out, "::std::unique_ptr<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001159 write_type(out, &ptr.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001160 write!(out, ">");
1161 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001162 Type::SharedPtr(ptr) => {
1163 write!(out, "::std::shared_ptr<");
1164 write_type(out, &ptr.inner);
1165 write!(out, ">");
1166 }
David Tolnay215e77f2020-12-28 17:09:48 -08001167 Type::WeakPtr(ptr) => {
1168 write!(out, "::std::weak_ptr<");
1169 write_type(out, &ptr.inner);
1170 write!(out, ">");
1171 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001172 Type::CxxVector(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +07001173 write!(out, "::std::vector<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001174 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001175 write!(out, ">");
1176 }
David Tolnay7db73692019-10-20 14:51:12 -04001177 Type::Ref(r) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001178 if !r.mutable {
David Tolnay7db73692019-10-20 14:51:12 -04001179 write!(out, "const ");
1180 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001181 write_type(out, &r.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001182 write!(out, " &");
1183 }
1184 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -08001185 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -04001186 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001187 Type::SliceRef(slice) => {
David Tolnayc5629f02020-11-23 18:32:46 -08001188 write!(out, "::rust::Slice<");
David Tolnay5515a9e2020-11-25 19:07:54 -08001189 if slice.mutability.is_none() {
David Tolnayc5629f02020-11-23 18:32:46 -08001190 write!(out, "const ");
1191 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001192 write_type(out, &slice.inner);
1193 write!(out, ">");
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001194 }
David Tolnay75dca2e2020-03-25 20:17:52 -07001195 Type::Fn(f) => {
David Tolnayf031c322020-11-29 19:41:33 -08001196 write!(out, "::rust::Fn<");
David Tolnay75dca2e2020-03-25 20:17:52 -07001197 match &f.ret {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001198 Some(ret) => write_type(out, ret),
David Tolnay75dca2e2020-03-25 20:17:52 -07001199 None => write!(out, "void"),
1200 }
1201 write!(out, "(");
1202 for (i, arg) in f.args.iter().enumerate() {
1203 if i > 0 {
1204 write!(out, ", ");
1205 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001206 write_type(out, &arg.ty);
David Tolnay75dca2e2020-03-25 20:17:52 -07001207 }
1208 write!(out, ")>");
1209 }
Xiangpeng Hao78762352020-11-12 10:24:18 +08001210 Type::Array(a) => {
1211 write!(out, "::std::array<");
1212 write_type(out, &a.inner);
1213 write!(out, ", {}>", &a.len);
1214 }
David Tolnay2fb14e92020-03-15 23:11:38 -07001215 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001216 }
1217}
1218
David Tolnayf6a89f22020-05-10 23:39:27 -07001219fn write_atom(out: &mut OutFile, atom: Atom) {
1220 match atom {
1221 Bool => write!(out, "bool"),
David Tolnayb3873dc2020-11-25 19:47:49 -08001222 Char => write!(out, "char"),
David Tolnaybe3cbf72020-12-12 22:12:07 -08001223 U8 => write!(out, "::std::uint8_t"),
1224 U16 => write!(out, "::std::uint16_t"),
1225 U32 => write!(out, "::std::uint32_t"),
1226 U64 => write!(out, "::std::uint64_t"),
1227 Usize => write!(out, "::std::size_t"),
1228 I8 => write!(out, "::std::int8_t"),
1229 I16 => write!(out, "::std::int16_t"),
1230 I32 => write!(out, "::std::int32_t"),
1231 I64 => write!(out, "::std::int64_t"),
David Tolnayf6a89f22020-05-10 23:39:27 -07001232 Isize => write!(out, "::rust::isize"),
1233 F32 => write!(out, "float"),
1234 F64 => write!(out, "double"),
1235 CxxString => write!(out, "::std::string"),
1236 RustString => write!(out, "::rust::String"),
1237 }
1238}
1239
David Tolnaya7c2ea12020-10-30 21:32:53 -07001240fn write_type_space(out: &mut OutFile, ty: &Type) {
1241 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001242 write_space_after_type(out, ty);
1243}
1244
1245fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001246 match ty {
David Tolnayeb952ba2020-04-14 15:02:24 -07001247 Type::Ident(_)
1248 | Type::RustBox(_)
1249 | Type::UniquePtr(_)
David Tolnayb3b24a12020-12-01 15:27:43 -08001250 | Type::SharedPtr(_)
David Tolnay215e77f2020-12-28 17:09:48 -08001251 | Type::WeakPtr(_)
David Tolnayeb952ba2020-04-14 15:02:24 -07001252 | Type::Str(_)
David Tolnay4377a9e2020-04-24 15:20:26 -07001253 | Type::CxxVector(_)
Myron Ahneba35cf2020-02-05 19:41:51 +07001254 | Type::RustVec(_)
David Tolnay73b72642020-11-25 17:44:05 -08001255 | Type::SliceRef(_)
Xiangpeng Hao78762352020-11-12 10:24:18 +08001256 | Type::Fn(_)
1257 | Type::Array(_) => write!(out, " "),
David Tolnay7db73692019-10-20 14:51:12 -04001258 Type::Ref(_) => {}
David Tolnaye0dca7b2020-11-25 17:18:57 -08001259 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001260 }
1261}
1262
David Tolnay1bdb4712020-11-25 07:27:54 -08001263#[derive(Copy, Clone)]
1264enum UniquePtr<'a> {
David Tolnay3abed472020-12-31 23:34:53 -08001265 Ident(&'a Ident),
1266 CxxVector(&'a Ident),
David Tolnay1bdb4712020-11-25 07:27:54 -08001267}
1268
1269trait ToTypename {
1270 fn to_typename(&self, types: &Types) -> String;
1271}
1272
David Tolnay3abed472020-12-31 23:34:53 -08001273impl ToTypename for Ident {
David Tolnay1bdb4712020-11-25 07:27:54 -08001274 fn to_typename(&self, types: &Types) -> String {
David Tolnay1e5fe232021-01-01 18:11:40 -08001275 types.resolve(self).name.to_fully_qualified()
David Tolnay2eca4a02020-04-24 19:50:51 -07001276 }
1277}
1278
David Tolnay1bdb4712020-11-25 07:27:54 -08001279impl<'a> ToTypename for UniquePtr<'a> {
1280 fn to_typename(&self, types: &Types) -> String {
1281 match self {
1282 UniquePtr::Ident(ident) => ident.to_typename(types),
1283 UniquePtr::CxxVector(element) => {
1284 format!("::std::vector<{}>", element.to_typename(types))
1285 }
1286 }
1287 }
1288}
1289
1290trait ToMangled {
1291 fn to_mangled(&self, types: &Types) -> Symbol;
1292}
1293
David Tolnay3abed472020-12-31 23:34:53 -08001294impl ToMangled for Ident {
David Tolnay1bdb4712020-11-25 07:27:54 -08001295 fn to_mangled(&self, types: &Types) -> Symbol {
David Tolnay1e5fe232021-01-01 18:11:40 -08001296 types.resolve(self).name.to_symbol()
David Tolnay1bdb4712020-11-25 07:27:54 -08001297 }
1298}
1299
1300impl<'a> ToMangled for UniquePtr<'a> {
1301 fn to_mangled(&self, types: &Types) -> Symbol {
1302 match self {
1303 UniquePtr::Ident(ident) => ident.to_mangled(types),
1304 UniquePtr::CxxVector(element) => element.to_mangled(types).prefix_with("std$vector$"),
1305 }
David Tolnaybae50ef2020-04-25 12:38:41 -07001306 }
1307}
1308
David Tolnaya7c2ea12020-10-30 21:32:53 -07001309fn write_generic_instantiations(out: &mut OutFile) {
David Tolnay169bb472020-11-01 21:04:24 -08001310 if out.header {
1311 return;
1312 }
1313
1314 out.next_section();
David Tolnay078c90f2020-11-01 13:31:08 -08001315 out.set_namespace(Default::default());
David Tolnay0c033e32020-11-01 15:15:48 -08001316 out.begin_block(Block::ExternC);
David Tolnay3abed472020-12-31 23:34:53 -08001317 for impl_key in out.types.impls.keys() {
1318 out.next_section();
1319 match impl_key {
1320 ImplKey::RustBox(ident) => write_rust_box_extern(out, ident),
1321 ImplKey::RustVec(ident) => write_rust_vec_extern(out, ident),
1322 ImplKey::UniquePtr(ident) => write_unique_ptr(out, ident),
1323 ImplKey::SharedPtr(ident) => write_shared_ptr(out, ident),
1324 ImplKey::WeakPtr(ident) => write_weak_ptr(out, ident),
1325 ImplKey::CxxVector(ident) => write_cxx_vector(out, ident),
David Tolnay7db73692019-10-20 14:51:12 -04001326 }
1327 }
David Tolnay0c033e32020-11-01 15:15:48 -08001328 out.end_block(Block::ExternC);
David Tolnay7db73692019-10-20 14:51:12 -04001329
David Tolnay0c033e32020-11-01 15:15:48 -08001330 out.begin_block(Block::Namespace("rust"));
David Tolnay0f0162f2020-11-16 23:43:37 -08001331 out.begin_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay3abed472020-12-31 23:34:53 -08001332 for impl_key in out.types.impls.keys() {
1333 match impl_key {
1334 ImplKey::RustBox(ident) => write_rust_box_impl(out, ident),
1335 ImplKey::RustVec(ident) => write_rust_vec_impl(out, ident),
1336 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -04001337 }
1338 }
David Tolnay0f0162f2020-11-16 23:43:37 -08001339 out.end_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay0c033e32020-11-01 15:15:48 -08001340 out.end_block(Block::Namespace("rust"));
David Tolnay7db73692019-10-20 14:51:12 -04001341}
1342
David Tolnay3abed472020-12-31 23:34:53 -08001343fn write_rust_box_extern(out: &mut OutFile, ident: &Ident) {
1344 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001345 let inner = resolve.name.to_fully_qualified();
1346 let instance = resolve.name.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001347
David Tolnay7db73692019-10-20 14:51:12 -04001348 writeln!(
1349 out,
David Tolnayc4b34222020-12-12 13:06:26 -08001350 "{} *cxxbridge1$box${}$alloc() noexcept;",
1351 inner, instance,
David Tolnay7db73692019-10-20 14:51:12 -04001352 );
1353 writeln!(
1354 out,
David Tolnay25b3c1d2020-12-12 15:50:10 -08001355 "void cxxbridge1$box${}$dealloc({} *) noexcept;",
1356 instance, inner,
1357 );
1358 writeln!(
1359 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001360 "void cxxbridge1$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001361 instance, inner,
1362 );
David Tolnay7db73692019-10-20 14:51:12 -04001363}
1364
David Tolnay3abed472020-12-31 23:34:53 -08001365fn write_rust_vec_extern(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001366 let inner = element.to_typename(out.types);
1367 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001368
David Tolnay12320e12020-12-09 23:09:36 -08001369 out.include.cstddef = true;
1370
Myron Ahneba35cf2020-02-05 19:41:51 +07001371 writeln!(
1372 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001373 "void cxxbridge1$rust_vec${}$new(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnayf97c2d52020-04-25 16:37:48 -07001374 instance, inner,
1375 );
1376 writeln!(
1377 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001378 "void cxxbridge1$rust_vec${}$drop(::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001379 instance, inner,
1380 );
1381 writeln!(
1382 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001383 "::std::size_t cxxbridge1$rust_vec${}$len(const ::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001384 instance, inner,
1385 );
David Tolnay219c0792020-04-24 20:31:37 -07001386 writeln!(
1387 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001388 "::std::size_t cxxbridge1$rust_vec${}$capacity(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnaydc62d712020-12-11 13:51:53 -08001389 instance, inner,
1390 );
1391 writeln!(
1392 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001393 "const {} *cxxbridge1$rust_vec${}$data(const ::rust::Vec<{0}> *ptr) noexcept;",
David Tolnay219c0792020-04-24 20:31:37 -07001394 inner, instance,
1395 );
David Tolnay503d0192020-04-24 22:18:56 -07001396 writeln!(
1397 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001398 "void cxxbridge1$rust_vec${}$reserve_total(::rust::Vec<{}> *ptr, ::std::size_t cap) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001399 instance, inner,
1400 );
1401 writeln!(
1402 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001403 "void cxxbridge1$rust_vec${}$set_len(::rust::Vec<{}> *ptr, ::std::size_t len) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001404 instance, inner,
1405 );
Myron Ahneba35cf2020-02-05 19:41:51 +07001406}
1407
David Tolnay3abed472020-12-31 23:34:53 -08001408fn write_rust_box_impl(out: &mut OutFile, ident: &Ident) {
1409 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001410 let inner = resolve.name.to_fully_qualified();
1411 let instance = resolve.name.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001412
1413 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001414 begin_function_definition(out);
David Tolnaye5703162020-12-12 16:26:35 -08001415 writeln!(
1416 out,
1417 "{} *Box<{}>::allocation::alloc() noexcept {{",
1418 inner, inner,
1419 );
David Tolnayc4b34222020-12-12 13:06:26 -08001420 writeln!(out, " return cxxbridge1$box${}$alloc();", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001421 writeln!(out, "}}");
1422
1423 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001424 begin_function_definition(out);
David Tolnay25b3c1d2020-12-12 15:50:10 -08001425 writeln!(
1426 out,
David Tolnaye5703162020-12-12 16:26:35 -08001427 "void Box<{}>::allocation::dealloc({} *ptr) noexcept {{",
David Tolnay25b3c1d2020-12-12 15:50:10 -08001428 inner, inner,
1429 );
1430 writeln!(out, " cxxbridge1$box${}$dealloc(ptr);", instance);
1431 writeln!(out, "}}");
1432
1433 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001434 begin_function_definition(out);
David Tolnay324437a2020-03-01 13:02:24 -08001435 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001436 writeln!(out, " cxxbridge1$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001437 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001438}
1439
David Tolnay3abed472020-12-31 23:34:53 -08001440fn write_rust_vec_impl(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001441 let inner = element.to_typename(out.types);
1442 let instance = element.to_mangled(out.types);
David Tolnay4791f1c2020-03-17 21:53:16 -07001443
David Tolnay12320e12020-12-09 23:09:36 -08001444 out.include.cstddef = true;
1445
Myron Ahneba35cf2020-02-05 19:41:51 +07001446 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001447 begin_function_definition(out);
David Tolnayf97c2d52020-04-25 16:37:48 -07001448 writeln!(out, "Vec<{}>::Vec() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001449 writeln!(out, " cxxbridge1$rust_vec${}$new(this);", instance);
David Tolnayf97c2d52020-04-25 16:37:48 -07001450 writeln!(out, "}}");
1451
1452 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001453 begin_function_definition(out);
Myron Ahneba35cf2020-02-05 19:41:51 +07001454 writeln!(out, "void Vec<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001455 writeln!(out, " return cxxbridge1$rust_vec${}$drop(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001456 writeln!(out, "}}");
1457
1458 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001459 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001460 writeln!(
1461 out,
1462 "::std::size_t Vec<{}>::size() const noexcept {{",
1463 inner,
1464 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001465 writeln!(out, " return cxxbridge1$rust_vec${}$len(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001466 writeln!(out, "}}");
David Tolnay219c0792020-04-24 20:31:37 -07001467
1468 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001469 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001470 writeln!(
1471 out,
1472 "::std::size_t Vec<{}>::capacity() const noexcept {{",
1473 inner,
1474 );
David Tolnay381d9532020-12-11 13:59:45 -08001475 writeln!(
1476 out,
1477 " return cxxbridge1$rust_vec${}$capacity(this);",
1478 instance,
1479 );
David Tolnaydc62d712020-12-11 13:51:53 -08001480 writeln!(out, "}}");
1481
1482 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001483 begin_function_definition(out);
David Tolnay219c0792020-04-24 20:31:37 -07001484 writeln!(out, "const {} *Vec<{0}>::data() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001485 writeln!(out, " return cxxbridge1$rust_vec${}$data(this);", instance);
David Tolnay219c0792020-04-24 20:31:37 -07001486 writeln!(out, "}}");
David Tolnay503d0192020-04-24 22:18:56 -07001487
1488 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001489 begin_function_definition(out);
David Tolnayfb6b73c2020-11-10 14:32:16 -08001490 writeln!(
1491 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001492 "void Vec<{}>::reserve_total(::std::size_t cap) noexcept {{",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001493 inner,
1494 );
1495 writeln!(
1496 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001497 " return cxxbridge1$rust_vec${}$reserve_total(this, cap);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001498 instance,
1499 );
1500 writeln!(out, "}}");
1501
1502 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001503 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001504 writeln!(
1505 out,
1506 "void Vec<{}>::set_len(::std::size_t len) noexcept {{",
1507 inner,
1508 );
David Tolnayfb6b73c2020-11-10 14:32:16 -08001509 writeln!(
1510 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001511 " return cxxbridge1$rust_vec${}$set_len(this, len);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001512 instance,
1513 );
1514 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001515}
1516
David Tolnay3abed472020-12-31 23:34:53 -08001517fn write_unique_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001518 let ty = UniquePtr::Ident(ident);
David Tolnay1bdb4712020-11-25 07:27:54 -08001519 write_unique_ptr_common(out, ty);
David Tolnay63da4d32020-04-25 09:41:12 -07001520}
1521
1522// Shared by UniquePtr<T> and UniquePtr<CxxVector<T>>.
David Tolnay1bdb4712020-11-25 07:27:54 -08001523fn write_unique_ptr_common(out: &mut OutFile, ty: UniquePtr) {
David Tolnay0ecd05a2020-07-29 16:32:03 -07001524 out.include.new = true;
Myron Ahneba35cf2020-02-05 19:41:51 +07001525 out.include.utility = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001526 let inner = ty.to_typename(out.types);
1527 let instance = ty.to_mangled(out.types);
David Tolnay7db73692019-10-20 14:51:12 -04001528
David Tolnay63da4d32020-04-25 09:41:12 -07001529 let can_construct_from_value = match ty {
David Tolnayca0f9da2020-10-16 13:16:17 -07001530 // Some aliases are to opaque types; some are to trivial types. We can't
1531 // know at code generation time, so we generate both C++ and Rust side
1532 // bindings for a "new" method anyway. But the Rust code can't be called
1533 // for Opaque types because the 'new' method is not implemented.
David Tolnay1bdb4712020-11-25 07:27:54 -08001534 UniquePtr::Ident(ident) => {
David Tolnay3abed472020-12-31 23:34:53 -08001535 out.types.structs.contains_key(ident)
1536 || out.types.enums.contains_key(ident)
1537 || out.types.aliases.contains_key(ident)
David Tolnayca0f9da2020-10-16 13:16:17 -07001538 }
David Tolnay1bdb4712020-11-25 07:27:54 -08001539 UniquePtr::CxxVector(_) => false,
David Tolnay63da4d32020-04-25 09:41:12 -07001540 };
1541
David Tolnay53462762020-11-25 07:08:10 -08001542 let conditional_delete = match ty {
1543 UniquePtr::Ident(ident) => {
David Tolnay3abed472020-12-31 23:34:53 -08001544 !out.types.structs.contains_key(ident) && !out.types.enums.contains_key(ident)
David Tolnay53462762020-11-25 07:08:10 -08001545 }
1546 UniquePtr::CxxVector(_) => false,
1547 };
1548
1549 if conditional_delete {
1550 out.builtin.is_complete = true;
1551 let definition = match ty {
David Tolnay1e5fe232021-01-01 18:11:40 -08001552 UniquePtr::Ident(ty) => &out.types.resolve(ty).name.cxx,
David Tolnay53462762020-11-25 07:08:10 -08001553 UniquePtr::CxxVector(_) => unreachable!(),
1554 };
1555 writeln!(
1556 out,
David Tolnay75068632020-12-26 22:15:17 -08001557 "static_assert(::rust::detail::is_complete<{}>::value, \"definition of {} is required\");",
David Tolnay53462762020-11-25 07:08:10 -08001558 inner, definition,
1559 );
1560 }
David Tolnay7db73692019-10-20 14:51:12 -04001561 writeln!(
1562 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001563 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001564 inner,
1565 );
1566 writeln!(
1567 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001568 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001569 inner,
1570 );
1571 writeln!(
1572 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001573 "void cxxbridge1$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001574 instance, inner,
1575 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001576 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001577 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001578 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001579 out.builtin.maybe_uninit = true;
David Tolnay63da4d32020-04-25 09:41:12 -07001580 writeln!(
David Tolnay53838912020-04-09 20:56:44 -07001581 out,
David Tolnay0b881402020-12-01 21:05:08 -08001582 "{} *cxxbridge1$unique_ptr${}$uninit(::std::unique_ptr<{}> *ptr) noexcept {{",
1583 inner, instance, inner,
David Tolnay53838912020-04-09 20:56:44 -07001584 );
David Tolnay63da4d32020-04-25 09:41:12 -07001585 writeln!(
1586 out,
David Tolnay0b881402020-12-01 21:05:08 -08001587 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1588 inner, inner, inner,
David Tolnay63da4d32020-04-25 09:41:12 -07001589 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001590 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001591 writeln!(out, " return uninit;");
David Tolnay63da4d32020-04-25 09:41:12 -07001592 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -07001593 }
David Tolnay7db73692019-10-20 14:51:12 -04001594 writeln!(
1595 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001596 "void cxxbridge1$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001597 instance, inner, inner,
1598 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001599 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001600 writeln!(out, "}}");
1601 writeln!(
1602 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001603 "const {} *cxxbridge1$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001604 inner, instance, inner,
1605 );
1606 writeln!(out, " return ptr.get();");
1607 writeln!(out, "}}");
1608 writeln!(
1609 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001610 "{} *cxxbridge1$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001611 inner, instance, inner,
1612 );
1613 writeln!(out, " return ptr.release();");
1614 writeln!(out, "}}");
1615 writeln!(
1616 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001617 "void cxxbridge1$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001618 instance, inner,
1619 );
David Tolnay53462762020-11-25 07:08:10 -08001620 if conditional_delete {
1621 out.builtin.deleter_if = true;
1622 writeln!(
1623 out,
David Tolnay75068632020-12-26 22:15:17 -08001624 " ::rust::deleter_if<::rust::detail::is_complete<{}>::value>{{}}(ptr);",
David Tolnay53462762020-11-25 07:08:10 -08001625 inner,
1626 );
1627 } else {
1628 writeln!(out, " ptr->~unique_ptr();");
1629 }
David Tolnay7db73692019-10-20 14:51:12 -04001630 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001631}
Myron Ahneba35cf2020-02-05 19:41:51 +07001632
David Tolnay3abed472020-12-31 23:34:53 -08001633fn write_shared_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay95bc57f2021-01-01 13:29:44 -08001634 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001635 let inner = resolve.name.to_fully_qualified();
1636 let instance = resolve.name.to_symbol();
David Tolnayb3b24a12020-12-01 15:27:43 -08001637
David Tolnayb3b24a12020-12-01 15:27:43 -08001638 out.include.new = true;
1639 out.include.utility = true;
1640
1641 // Some aliases are to opaque types; some are to trivial types. We can't
1642 // know at code generation time, so we generate both C++ and Rust side
1643 // bindings for a "new" method anyway. But the Rust code can't be called for
1644 // Opaque types because the 'new' method is not implemented.
David Tolnay3abed472020-12-31 23:34:53 -08001645 let can_construct_from_value = out.types.structs.contains_key(ident)
1646 || out.types.enums.contains_key(ident)
1647 || out.types.aliases.contains_key(ident);
David Tolnayb3b24a12020-12-01 15:27:43 -08001648
David Tolnayb3b24a12020-12-01 15:27:43 -08001649 writeln!(
1650 out,
1651 "static_assert(sizeof(::std::shared_ptr<{}>) == 2 * sizeof(void *), \"\");",
1652 inner,
1653 );
1654 writeln!(
1655 out,
1656 "static_assert(alignof(::std::shared_ptr<{}>) == alignof(void *), \"\");",
1657 inner,
1658 );
1659 writeln!(
1660 out,
1661 "void cxxbridge1$shared_ptr${}$null(::std::shared_ptr<{}> *ptr) noexcept {{",
1662 instance, inner,
1663 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001664 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>();", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001665 writeln!(out, "}}");
1666 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001667 out.builtin.maybe_uninit = true;
David Tolnayb3b24a12020-12-01 15:27:43 -08001668 writeln!(
1669 out,
David Tolnay0b881402020-12-01 21:05:08 -08001670 "{} *cxxbridge1$shared_ptr${}$uninit(::std::shared_ptr<{}> *ptr) noexcept {{",
1671 inner, instance, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001672 );
1673 writeln!(
1674 out,
David Tolnay0b881402020-12-01 21:05:08 -08001675 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1676 inner, inner, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001677 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001678 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001679 writeln!(out, " return uninit;");
David Tolnayb3b24a12020-12-01 15:27:43 -08001680 writeln!(out, "}}");
1681 }
1682 writeln!(
1683 out,
1684 "void cxxbridge1$shared_ptr${}$clone(const ::std::shared_ptr<{}>& self, ::std::shared_ptr<{}> *ptr) noexcept {{",
1685 instance, inner, inner,
1686 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001687 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(self);", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001688 writeln!(out, "}}");
1689 writeln!(
1690 out,
1691 "const {} *cxxbridge1$shared_ptr${}$get(const ::std::shared_ptr<{}>& self) noexcept {{",
1692 inner, instance, inner,
1693 );
1694 writeln!(out, " return self.get();");
1695 writeln!(out, "}}");
1696 writeln!(
1697 out,
1698 "void cxxbridge1$shared_ptr${}$drop(::std::shared_ptr<{}> *self) noexcept {{",
1699 instance, inner,
1700 );
1701 writeln!(out, " self->~shared_ptr();");
1702 writeln!(out, "}}");
David Tolnayb3b24a12020-12-01 15:27:43 -08001703}
1704
David Tolnay3abed472020-12-31 23:34:53 -08001705fn write_weak_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay95bc57f2021-01-01 13:29:44 -08001706 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001707 let inner = resolve.name.to_fully_qualified();
1708 let instance = resolve.name.to_symbol();
David Tolnay215e77f2020-12-28 17:09:48 -08001709
1710 out.include.new = true;
1711 out.include.utility = true;
1712
1713 writeln!(
1714 out,
1715 "static_assert(sizeof(::std::weak_ptr<{}>) == 2 * sizeof(void *), \"\");",
1716 inner,
1717 );
1718 writeln!(
1719 out,
1720 "static_assert(alignof(::std::weak_ptr<{}>) == alignof(void *), \"\");",
1721 inner,
1722 );
1723 writeln!(
1724 out,
1725 "void cxxbridge1$weak_ptr${}$null(::std::weak_ptr<{}> *ptr) noexcept {{",
1726 instance, inner,
1727 );
1728 writeln!(out, " ::new (ptr) ::std::weak_ptr<{}>();", inner);
1729 writeln!(out, "}}");
1730 writeln!(
1731 out,
1732 "void cxxbridge1$weak_ptr${}$clone(const ::std::weak_ptr<{}>& self, ::std::weak_ptr<{}> *ptr) noexcept {{",
1733 instance, inner, inner,
1734 );
1735 writeln!(out, " ::new (ptr) ::std::weak_ptr<{}>(self);", inner);
1736 writeln!(out, "}}");
1737 writeln!(
1738 out,
David Tolnay85b6bc42020-12-28 17:47:51 -08001739 "void cxxbridge1$weak_ptr${}$downgrade(const ::std::shared_ptr<{}>& shared, ::std::weak_ptr<{}> *weak) noexcept {{",
1740 instance, inner, inner,
1741 );
1742 writeln!(out, " ::new (weak) ::std::weak_ptr<{}>(shared);", inner);
1743 writeln!(out, "}}");
1744 writeln!(
1745 out,
David Tolnay7a487852020-12-28 18:02:25 -08001746 "void cxxbridge1$weak_ptr${}$upgrade(const ::std::weak_ptr<{}>& weak, ::std::shared_ptr<{}> *shared) noexcept {{",
1747 instance, inner, inner,
1748 );
1749 writeln!(
1750 out,
1751 " ::new (shared) ::std::shared_ptr<{}>(weak.lock());",
1752 inner,
1753 );
1754 writeln!(out, "}}");
1755 writeln!(
1756 out,
David Tolnay215e77f2020-12-28 17:09:48 -08001757 "void cxxbridge1$weak_ptr${}$drop(::std::weak_ptr<{}> *self) noexcept {{",
1758 instance, inner,
1759 );
1760 writeln!(out, " self->~weak_ptr();");
1761 writeln!(out, "}}");
1762}
1763
David Tolnay3abed472020-12-31 23:34:53 -08001764fn write_cxx_vector(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001765 let inner = element.to_typename(out.types);
1766 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001767
David Tolnay12320e12020-12-09 23:09:36 -08001768 out.include.cstddef = true;
1769
Myron Ahneba35cf2020-02-05 19:41:51 +07001770 writeln!(
1771 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001772 "::std::size_t cxxbridge1$std$vector${}$size(const ::std::vector<{}> &s) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001773 instance, inner,
1774 );
1775 writeln!(out, " return s.size();");
1776 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001777 writeln!(
1778 out,
David Tolnay767e00d2020-12-21 17:12:27 -08001779 "{} *cxxbridge1$std$vector${}$get_unchecked(::std::vector<{}> *s, ::std::size_t pos) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001780 inner, instance, inner,
1781 );
David Tolnay767e00d2020-12-21 17:12:27 -08001782 writeln!(out, " return &(*s)[pos];");
Myron Ahneba35cf2020-02-05 19:41:51 +07001783 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001784
David Tolnay70820672020-12-07 11:15:14 -08001785 out.include.memory = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001786 write_unique_ptr_common(out, UniquePtr::CxxVector(element));
Myron Ahneba35cf2020-02-05 19:41:51 +07001787}