blob: 3e79548581b12af223ad526cb111d6431ea79fe4 [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;
8use 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 Tolnay440e24a2021-01-01 23:25:34 -0800122 out.set_namespace(Default::default());
123
124 out.next_section();
125 for ty in out.types {
126 // MSVC workaround for "C linkage function cannot return C++ class"
127 // error. Apparently the compiler fails to perform implicit
128 // instantiations as part of an extern declaration. Instead we
129 // instantiate explicitly.
130 // See https://stackoverflow.com/a/57429504/6086311.
131 if let Type::SliceRef(_) = ty {
132 write!(out, "template struct ");
133 write_type(out, ty);
134 writeln!(out, ";");
135 }
136 }
137
David Tolnayfabca772020-10-03 21:25:41 -0700138 out.next_section();
139 for api in apis {
140 if let Api::TypeAlias(ety) = api {
Adrian Taylorf831a5a2020-12-07 16:49:19 -0800141 if let Some(reasons) = out.types.required_trivial.get(&ety.name.rust) {
David Tolnay5b1d8632020-12-20 22:05:11 -0800142 check_trivial_extern_type(out, ety, reasons)
David Tolnayfabca772020-10-03 21:25:41 -0700143 }
144 }
145 }
David Tolnay1b0339c2020-11-01 20:37:50 -0800146}
147
David Tolnaye1109d92020-11-01 20:51:56 -0800148fn write_functions<'a>(out: &mut OutFile<'a>, apis: &'a [Api]) {
David Tolnayce5a91f2020-10-31 22:42:08 -0700149 if !out.header {
David Tolnay7db73692019-10-20 14:51:12 -0400150 for api in apis {
David Tolnay04770742020-11-01 13:50:50 -0800151 match api {
David Tolnayb960ed22020-11-27 14:34:30 -0800152 Api::Struct(strct) => write_struct_operator_decls(out, strct),
David Tolnay358bc4b2020-12-26 22:37:57 -0800153 Api::RustType(ety) => write_opaque_type_layout_decls(out, ety),
David Tolnay04770742020-11-01 13:50:50 -0800154 Api::CxxFunction(efn) => write_cxx_function_shim(out, efn),
155 Api::RustFunction(efn) => write_rust_function_decl(out, efn),
156 _ => {}
157 }
David Tolnay7db73692019-10-20 14:51:12 -0400158 }
David Tolnay7da38202020-11-27 17:36:16 -0800159
160 write_std_specializations(out, apis);
David Tolnay7db73692019-10-20 14:51:12 -0400161 }
162
163 for api in apis {
David Tolnayb960ed22020-11-27 14:34:30 -0800164 match api {
165 Api::Struct(strct) => write_struct_operators(out, strct),
David Tolnay358bc4b2020-12-26 22:37:57 -0800166 Api::RustType(ety) => write_opaque_type_layout(out, ety),
David Tolnayb960ed22020-11-27 14:34:30 -0800167 Api::RustFunction(efn) => {
168 out.next_section();
169 write_rust_function_shim(out, efn);
170 }
171 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400172 }
173 }
David Tolnay7db73692019-10-20 14:51:12 -0400174}
175
David Tolnay7da38202020-11-27 17:36:16 -0800176fn write_std_specializations(out: &mut OutFile, apis: &[Api]) {
177 out.set_namespace(Default::default());
178 out.begin_block(Block::Namespace("std"));
179
180 for api in apis {
181 if let Api::Struct(strct) = api {
182 if derive::contains(&strct.derives, Trait::Hash) {
183 out.next_section();
David Tolnay12320e12020-12-09 23:09:36 -0800184 out.include.cstddef = true;
David Tolnay08a03db2020-12-09 23:04:36 -0800185 out.include.functional = true;
David Tolnay7da38202020-11-27 17:36:16 -0800186 let qualified = strct.name.to_fully_qualified();
187 writeln!(out, "template <> struct hash<{}> {{", qualified);
188 writeln!(
189 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800190 " ::std::size_t operator()(const {} &self) const noexcept {{",
David Tolnay7da38202020-11-27 17:36:16 -0800191 qualified,
192 );
193 let link_name = mangle::operator(&strct.name, "hash");
194 write!(out, " return ::");
195 for name in &strct.name.namespace {
196 write!(out, "{}::", name);
197 }
198 writeln!(out, "{}(self);", link_name);
199 writeln!(out, " }}");
200 writeln!(out, "}};");
201 }
202 }
203 }
204
205 out.end_block(Block::Namespace("std"));
206}
207
David Tolnaydfb82d72020-11-02 00:10:10 -0800208fn pick_includes_and_builtins(out: &mut OutFile, apis: &[Api]) {
209 for api in apis {
210 if let Api::Include(include) = api {
211 out.include.insert(include);
212 }
213 }
214
David Tolnaya7c2ea12020-10-30 21:32:53 -0700215 for ty in out.types {
David Tolnay7db73692019-10-20 14:51:12 -0400216 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -0700217 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnay89e386d2020-10-03 19:02:19 -0700218 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
219 | Some(I64) => out.include.cstdint = true,
220 Some(Usize) => out.include.cstddef = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800221 Some(Isize) => out.builtin.rust_isize = true,
David Tolnay89e386d2020-10-03 19:02:19 -0700222 Some(CxxString) => out.include.string = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800223 Some(RustString) => out.builtin.rust_string = true,
David Tolnayb3873dc2020-11-25 19:47:49 -0800224 Some(Bool) | Some(Char) | Some(F32) | Some(F64) | None => {}
David Tolnay89e386d2020-10-03 19:02:19 -0700225 },
David Tolnaydcfa8e92020-11-02 09:50:06 -0800226 Type::RustBox(_) => out.builtin.rust_box = true,
227 Type::RustVec(_) => out.builtin.rust_vec = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700228 Type::UniquePtr(_) => out.include.memory = true,
David Tolnay215e77f2020-12-28 17:09:48 -0800229 Type::SharedPtr(_) | Type::WeakPtr(_) => out.include.memory = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800230 Type::Str(_) => out.builtin.rust_str = true,
David Tolnayb9da1462020-10-31 21:03:41 -0700231 Type::CxxVector(_) => out.include.vector = true,
David Tolnaydcfa8e92020-11-02 09:50:06 -0800232 Type::Fn(_) => out.builtin.rust_fn = true,
David Tolnay5515a9e2020-11-25 19:07:54 -0800233 Type::SliceRef(_) => out.builtin.rust_slice = true,
David Tolnaye8b1bb42020-11-24 20:37:37 -0800234 Type::Array(_) => out.include.array = true,
David Tolnay11bd7ff2020-11-01 19:44:58 -0800235 Type::Ref(_) | Type::Void(_) => {}
David Tolnay7db73692019-10-20 14:51:12 -0400236 }
237 }
David Tolnayec66d112020-10-31 21:00:26 -0700238}
David Tolnayf51447e2020-03-06 14:14:27 -0800239
David Tolnay102c7ea2020-11-08 18:58:09 -0800240fn write_struct<'a>(out: &mut OutFile<'a>, strct: &'a Struct, methods: &[&ExternFn]) {
David Tolnayb960ed22020-11-27 14:34:30 -0800241 let operator_eq = derive::contains(&strct.derives, Trait::PartialEq);
David Tolnay84389352020-11-27 17:12:10 -0800242 let operator_ord = derive::contains(&strct.derives, Trait::PartialOrd);
David Tolnayb960ed22020-11-27 14:34:30 -0800243
David Tolnay17a934c2020-11-02 00:40:04 -0800244 out.set_namespace(&strct.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800245 let guard = format!("CXXBRIDGE1_STRUCT_{}", strct.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700246 writeln!(out, "#ifndef {}", guard);
247 writeln!(out, "#define {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400248 for line in strct.doc.to_string().lines() {
249 writeln!(out, "//{}", line);
250 }
David Tolnay17a934c2020-11-02 00:40:04 -0800251 writeln!(out, "struct {} final {{", strct.name.cxx);
David Tolnay84389352020-11-27 17:12:10 -0800252
David Tolnay7db73692019-10-20 14:51:12 -0400253 for field in &strct.fields {
David Tolnay5573e522020-12-31 11:07:38 -0800254 for line in field.doc.to_string().lines() {
255 writeln!(out, " //{}", line);
256 }
David Tolnay7db73692019-10-20 14:51:12 -0400257 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700258 write_type_space(out, &field.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800259 writeln!(out, "{};", field.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400260 }
David Tolnay84389352020-11-27 17:12:10 -0800261
David Tolnay5554ebd2020-12-10 11:34:41 -0800262 writeln!(out);
David Tolnay84389352020-11-27 17:12:10 -0800263
David Tolnay102c7ea2020-11-08 18:58:09 -0800264 for method in methods {
265 write!(out, " ");
266 let sig = &method.sig;
267 let local_name = method.name.cxx.to_string();
268 write_rust_function_shim_decl(out, &local_name, sig, false);
269 writeln!(out, ";");
270 }
David Tolnay84389352020-11-27 17:12:10 -0800271
David Tolnayb960ed22020-11-27 14:34:30 -0800272 if operator_eq {
273 writeln!(
274 out,
275 " bool operator==(const {} &) const noexcept;",
276 strct.name.cxx,
277 );
278 writeln!(
279 out,
280 " bool operator!=(const {} &) const noexcept;",
281 strct.name.cxx,
282 );
283 }
David Tolnay84389352020-11-27 17:12:10 -0800284
285 if operator_ord {
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 writeln!(
297 out,
298 " bool operator>(const {} &) const noexcept;",
299 strct.name.cxx,
300 );
301 writeln!(
302 out,
303 " bool operator>=(const {} &) const noexcept;",
304 strct.name.cxx,
305 );
306 }
307
David Tolnay5554ebd2020-12-10 11:34:41 -0800308 out.include.type_traits = true;
309 writeln!(out, " using IsRelocatable = ::std::true_type;");
310
David Tolnay7db73692019-10-20 14:51:12 -0400311 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700312 writeln!(out, "#endif // {}", guard);
David Tolnay7db73692019-10-20 14:51:12 -0400313}
314
David Tolnayed6ba4a2021-01-01 14:59:40 -0800315fn write_struct_decl(out: &mut OutFile, ident: &Pair) {
316 writeln!(out, "struct {};", ident.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400317}
318
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800319fn write_enum_decl(out: &mut OutFile, enm: &Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800320 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnay0e3ee8e2020-11-01 23:46:52 -0800321 write_atom(out, enm.repr);
322 writeln!(out, ";");
323}
324
David Tolnay8faec772020-11-02 00:18:19 -0800325fn write_struct_using(out: &mut OutFile, ident: &Pair) {
326 writeln!(out, "using {} = {};", ident.cxx, ident.to_fully_qualified());
David Tolnay8861bee2020-01-20 18:39:24 -0800327}
328
David Tolnay8d067de2020-12-26 16:18:23 -0800329fn write_opaque_type<'a>(out: &mut OutFile<'a>, ety: &'a ExternType, methods: &[&ExternFn]) {
David Tolnay17a934c2020-11-02 00:40:04 -0800330 out.set_namespace(&ety.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800331 let guard = format!("CXXBRIDGE1_STRUCT_{}", ety.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700332 writeln!(out, "#ifndef {}", guard);
333 writeln!(out, "#define {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700334 for line in ety.doc.to_string().lines() {
335 writeln!(out, "//{}", line);
336 }
David Tolnay6ba262f2020-12-26 22:23:50 -0800337
David Tolnay365fc7c2020-11-25 16:08:13 -0800338 out.builtin.opaque = true;
David Tolnay6ba262f2020-12-26 22:23:50 -0800339 writeln!(
David Tolnay7c06b862020-11-25 16:59:09 -0800340 out,
341 "struct {} final : public ::rust::Opaque {{",
342 ety.name.cxx,
343 );
David Tolnay6ba262f2020-12-26 22:23:50 -0800344
Joel Galenson968738f2020-04-15 14:19:33 -0700345 for method in methods {
David Tolnay6ba262f2020-12-26 22:23:50 -0800346 write!(out, " ");
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700347 let sig = &method.sig;
David Tolnay17a934c2020-11-02 00:40:04 -0800348 let local_name = method.name.cxx.to_string();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700349 write_rust_function_shim_decl(out, &local_name, sig, false);
David Tolnay6ba262f2020-12-26 22:23:50 -0800350 writeln!(out, ";");
David Tolnay8d067de2020-12-26 16:18:23 -0800351 }
David Tolnay6ba262f2020-12-26 22:23:50 -0800352
David Tolnay8d067de2020-12-26 16:18:23 -0800353 if !methods.is_empty() {
354 writeln!(out);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700355 }
David Tolnay6ba262f2020-12-26 22:23:50 -0800356
David Tolnayee6ecfc2020-12-26 21:54:37 -0800357 out.builtin.layout = true;
David Tolnay6ba262f2020-12-26 22:23:50 -0800358 out.include.cstddef = true;
359 writeln!(out, "private:");
David Tolnayee6ecfc2020-12-26 21:54:37 -0800360 writeln!(out, " friend ::rust::layout;");
David Tolnay6ba262f2020-12-26 22:23:50 -0800361 writeln!(out, " struct layout {{");
362 writeln!(out, " static ::std::size_t size() noexcept;");
363 writeln!(out, " static ::std::size_t align() noexcept;");
364 writeln!(out, " }};");
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700365 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700366 writeln!(out, "#endif // {}", guard);
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700367}
368
David Tolnay0b9b9f82020-11-01 20:41:00 -0800369fn write_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800370 out.set_namespace(&enm.name.namespace);
David Tolnay0f0162f2020-11-16 23:43:37 -0800371 let guard = format!("CXXBRIDGE1_ENUM_{}", enm.name.to_symbol());
David Tolnaya25ea9c2020-08-27 22:59:38 -0700372 writeln!(out, "#ifndef {}", guard);
373 writeln!(out, "#define {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700374 for line in enm.doc.to_string().lines() {
375 writeln!(out, "//{}", line);
376 }
David Tolnay17a934c2020-11-02 00:40:04 -0800377 write!(out, "enum class {} : ", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700378 write_atom(out, enm.repr);
379 writeln!(out, " {{");
Joel Galensonc03402a2020-04-23 17:31:09 -0700380 for variant in &enm.variants {
David Tolnay4486f722020-12-30 00:01:26 -0800381 for line in variant.doc.to_string().lines() {
382 writeln!(out, " //{}", line);
383 }
David Tolnaye6f62142020-12-21 16:00:41 -0800384 writeln!(out, " {} = {},", variant.name.cxx, variant.discriminant);
Joel Galensonc03402a2020-04-23 17:31:09 -0700385 }
386 writeln!(out, "}};");
David Tolnaya25ea9c2020-08-27 22:59:38 -0700387 writeln!(out, "#endif // {}", guard);
Joel Galensonc03402a2020-04-23 17:31:09 -0700388}
389
David Tolnay0b9b9f82020-11-01 20:41:00 -0800390fn check_enum<'a>(out: &mut OutFile<'a>, enm: &'a Enum) {
David Tolnay17a934c2020-11-02 00:40:04 -0800391 out.set_namespace(&enm.name.namespace);
David Tolnay06711bc2020-11-19 19:25:14 -0800392 out.include.type_traits = true;
393 writeln!(
394 out,
395 "static_assert(::std::is_enum<{}>::value, \"expected enum\");",
396 enm.name.cxx,
397 );
David Tolnay17a934c2020-11-02 00:40:04 -0800398 write!(out, "static_assert(sizeof({}) == sizeof(", enm.name.cxx);
David Tolnayf6a89f22020-05-10 23:39:27 -0700399 write_atom(out, enm.repr);
400 writeln!(out, "), \"incorrect size\");");
Joel Galenson0f654ff2020-05-04 20:04:21 -0700401 for variant in &enm.variants {
David Tolnayf6a89f22020-05-10 23:39:27 -0700402 write!(out, "static_assert(static_cast<");
403 write_atom(out, enm.repr);
Joel Galenson0f654ff2020-05-04 20:04:21 -0700404 writeln!(
405 out,
David Tolnayf6a89f22020-05-10 23:39:27 -0700406 ">({}::{}) == {}, \"disagrees with the value in #[cxx::bridge]\");",
David Tolnaye6f62142020-12-21 16:00:41 -0800407 enm.name.cxx, variant.name.cxx, variant.discriminant,
Joel Galenson0f654ff2020-05-04 20:04:21 -0700408 );
Joel Galenson0f654ff2020-05-04 20:04:21 -0700409 }
Joel Galenson905eb2e2020-05-04 14:58:14 -0700410}
411
David Tolnay5b1d8632020-12-20 22:05:11 -0800412fn check_trivial_extern_type(out: &mut OutFile, alias: &TypeAlias, reasons: &[TrivialReason]) {
David Tolnay174bd952020-11-02 09:23:12 -0800413 // NOTE: The following static assertion is just nice-to-have and not
David Tolnayfd0034e2020-10-04 13:15:34 -0700414 // necessary for soundness. That's because triviality is always declared by
415 // the user in the form of an unsafe impl of cxx::ExternType:
416 //
417 // unsafe impl ExternType for MyType {
418 // type Id = cxx::type_id!("...");
419 // type Kind = cxx::kind::Trivial;
420 // }
421 //
422 // Since the user went on the record with their unsafe impl to unsafely
423 // claim they KNOW that the type is trivial, it's fine for that to be on
David Tolnay174bd952020-11-02 09:23:12 -0800424 // them if that were wrong. However, in practice correctly reasoning about
425 // the relocatability of C++ types is challenging, particularly if the type
426 // definition were to change over time, so for now we add this check.
David Tolnayfd0034e2020-10-04 13:15:34 -0700427 //
David Tolnay174bd952020-11-02 09:23:12 -0800428 // There may be legitimate reasons to opt out of this assertion for support
429 // of types that the programmer knows are soundly Rust-movable despite not
430 // being recognized as such by the C++ type system due to a move constructor
431 // or destructor. To opt out of the relocatability check, they need to do
432 // one of the following things in any header used by `include!` in their
433 // bridge.
434 //
435 // --- if they define the type:
436 // struct MyType {
437 // ...
438 // + using IsRelocatable = std::true_type;
439 // };
440 //
441 // --- otherwise:
442 // + template <>
443 // + struct rust::IsRelocatable<MyType> : std::true_type {};
444 //
David Tolnayfd0034e2020-10-04 13:15:34 -0700445
David Tolnay5b1d8632020-12-20 22:05:11 -0800446 let id = alias.name.to_fully_qualified();
David Tolnay174bd952020-11-02 09:23:12 -0800447 out.builtin.relocatable = true;
David Tolnay5b1d8632020-12-20 22:05:11 -0800448 writeln!(out, "static_assert(");
449 writeln!(out, " ::rust::IsRelocatable<{}>::value,", id);
450 writeln!(
451 out,
452 " \"type {} should be trivially move constructible and trivially destructible in C++ to be used as {} in Rust\");",
453 id.trim_start_matches("::"),
454 trivial::as_what(&alias.name, reasons),
455 );
Adrian Taylor405e8742020-09-25 14:01:25 -0700456}
457
David Tolnayb960ed22020-11-27 14:34:30 -0800458fn write_struct_operator_decls<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
459 out.set_namespace(&strct.name.namespace);
460 out.begin_block(Block::ExternC);
461
462 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800463 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800464 writeln!(
465 out,
466 "bool {}(const {1} &, const {1} &) noexcept;",
467 link_name, strct.name.cxx,
468 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800469
470 if !derive::contains(&strct.derives, Trait::Eq) {
David Tolnaya05f9402020-11-27 17:44:05 -0800471 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800472 writeln!(
473 out,
474 "bool {}(const {1} &, const {1} &) noexcept;",
475 link_name, strct.name.cxx,
476 );
477 }
David Tolnayb960ed22020-11-27 14:34:30 -0800478 }
479
David Tolnay84389352020-11-27 17:12:10 -0800480 if derive::contains(&strct.derives, Trait::PartialOrd) {
David Tolnaya05f9402020-11-27 17:44:05 -0800481 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800482 writeln!(
483 out,
484 "bool {}(const {1} &, const {1} &) noexcept;",
485 link_name, strct.name.cxx,
486 );
487
David Tolnaya05f9402020-11-27 17:44:05 -0800488 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800489 writeln!(
490 out,
491 "bool {}(const {1} &, const {1} &) noexcept;",
492 link_name, strct.name.cxx,
493 );
494
495 if !derive::contains(&strct.derives, Trait::Ord) {
David Tolnaya05f9402020-11-27 17:44:05 -0800496 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800497 writeln!(
498 out,
499 "bool {}(const {1} &, const {1} &) noexcept;",
500 link_name, strct.name.cxx,
501 );
502
David Tolnaya05f9402020-11-27 17:44:05 -0800503 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800504 writeln!(
505 out,
506 "bool {}(const {1} &, const {1} &) noexcept;",
507 link_name, strct.name.cxx,
508 );
509 }
510 }
511
David Tolnay7da38202020-11-27 17:36:16 -0800512 if derive::contains(&strct.derives, Trait::Hash) {
David Tolnay12320e12020-12-09 23:09:36 -0800513 out.include.cstddef = true;
David Tolnay7da38202020-11-27 17:36:16 -0800514 let link_name = mangle::operator(&strct.name, "hash");
515 writeln!(
516 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -0800517 "::std::size_t {}(const {} &) noexcept;",
David Tolnay7da38202020-11-27 17:36:16 -0800518 link_name, strct.name.cxx,
519 );
520 }
521
David Tolnayb960ed22020-11-27 14:34:30 -0800522 out.end_block(Block::ExternC);
523}
524
525fn write_struct_operators<'a>(out: &mut OutFile<'a>, strct: &'a Struct) {
526 if out.header {
527 return;
528 }
529
530 out.set_namespace(&strct.name.namespace);
531
532 if derive::contains(&strct.derives, Trait::PartialEq) {
David Tolnayb960ed22020-11-27 14:34:30 -0800533 out.next_section();
534 writeln!(
535 out,
536 "bool {0}::operator==(const {0} &rhs) const noexcept {{",
537 strct.name.cxx,
538 );
David Tolnaya05f9402020-11-27 17:44:05 -0800539 let link_name = mangle::operator(&strct.name, "eq");
David Tolnayb960ed22020-11-27 14:34:30 -0800540 writeln!(out, " return {}(*this, rhs);", link_name);
541 writeln!(out, "}}");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800542
David Tolnayb960ed22020-11-27 14:34:30 -0800543 out.next_section();
544 writeln!(
545 out,
546 "bool {0}::operator!=(const {0} &rhs) const noexcept {{",
547 strct.name.cxx,
548 );
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800549 if derive::contains(&strct.derives, Trait::Eq) {
550 writeln!(out, " return !(*this == rhs);");
551 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800552 let link_name = mangle::operator(&strct.name, "ne");
David Tolnaya6f3b6f2020-11-27 16:47:30 -0800553 writeln!(out, " return {}(*this, rhs);", link_name);
554 }
David Tolnayb960ed22020-11-27 14:34:30 -0800555 writeln!(out, "}}");
556 }
David Tolnay84389352020-11-27 17:12:10 -0800557
558 if derive::contains(&strct.derives, Trait::PartialOrd) {
559 out.next_section();
560 writeln!(
561 out,
562 "bool {0}::operator<(const {0} &rhs) const noexcept {{",
563 strct.name.cxx,
564 );
David Tolnaya05f9402020-11-27 17:44:05 -0800565 let link_name = mangle::operator(&strct.name, "lt");
David Tolnay84389352020-11-27 17:12:10 -0800566 writeln!(out, " return {}(*this, rhs);", link_name);
567 writeln!(out, "}}");
568
569 out.next_section();
570 writeln!(
571 out,
572 "bool {0}::operator<=(const {0} &rhs) const noexcept {{",
573 strct.name.cxx,
574 );
David Tolnaya05f9402020-11-27 17:44:05 -0800575 let link_name = mangle::operator(&strct.name, "le");
David Tolnay84389352020-11-27 17:12:10 -0800576 writeln!(out, " return {}(*this, rhs);", link_name);
577 writeln!(out, "}}");
578
579 out.next_section();
580 writeln!(
581 out,
582 "bool {0}::operator>(const {0} &rhs) const noexcept {{",
583 strct.name.cxx,
584 );
585 if derive::contains(&strct.derives, Trait::Ord) {
586 writeln!(out, " return !(*this <= rhs);");
587 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800588 let link_name = mangle::operator(&strct.name, "gt");
David Tolnay84389352020-11-27 17:12:10 -0800589 writeln!(out, " return {}(*this, rhs);", link_name);
590 }
591 writeln!(out, "}}");
592
593 out.next_section();
594 writeln!(
595 out,
596 "bool {0}::operator>=(const {0} &rhs) const noexcept {{",
597 strct.name.cxx,
598 );
599 if derive::contains(&strct.derives, Trait::Ord) {
600 writeln!(out, " return !(*this < rhs);");
601 } else {
David Tolnaya05f9402020-11-27 17:44:05 -0800602 let link_name = mangle::operator(&strct.name, "ge");
David Tolnay84389352020-11-27 17:12:10 -0800603 writeln!(out, " return {}(*this, rhs);", link_name);
604 }
605 writeln!(out, "}}");
606 }
David Tolnayb960ed22020-11-27 14:34:30 -0800607}
608
David Tolnay358bc4b2020-12-26 22:37:57 -0800609fn write_opaque_type_layout_decls<'a>(out: &mut OutFile<'a>, ety: &'a ExternType) {
610 out.set_namespace(&ety.name.namespace);
611 out.begin_block(Block::ExternC);
612
613 let link_name = mangle::operator(&ety.name, "sizeof");
614 writeln!(out, "::std::size_t {}() noexcept;", link_name);
615
616 let link_name = mangle::operator(&ety.name, "alignof");
617 writeln!(out, "::std::size_t {}() noexcept;", link_name);
618
619 out.end_block(Block::ExternC);
620}
621
622fn write_opaque_type_layout<'a>(out: &mut OutFile<'a>, ety: &'a ExternType) {
623 if out.header {
624 return;
625 }
626
627 out.set_namespace(&ety.name.namespace);
628
629 out.next_section();
630 let link_name = mangle::operator(&ety.name, "sizeof");
631 writeln!(
632 out,
633 "::std::size_t {}::layout::size() noexcept {{",
634 ety.name.cxx,
635 );
636 writeln!(out, " return {}();", link_name);
637 writeln!(out, "}}");
638
639 out.next_section();
640 let link_name = mangle::operator(&ety.name, "alignof");
641 writeln!(
642 out,
643 "::std::size_t {}::layout::align() noexcept {{",
644 ety.name.cxx,
645 );
646 writeln!(out, " return {}();", link_name);
647 writeln!(out, "}}");
648}
649
David Tolnay1eadd262020-12-29 16:42:08 -0800650fn begin_function_definition(out: &mut OutFile) {
651 if let Some(annotation) = &out.opt.cxx_impl_annotations {
652 write!(out, "{} ", annotation);
653 }
654}
655
David Tolnay0b9b9f82020-11-01 20:41:00 -0800656fn write_cxx_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay04770742020-11-01 13:50:50 -0800657 out.next_section();
David Tolnay17a934c2020-11-02 00:40:04 -0800658 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800659 out.begin_block(Block::ExternC);
David Tolnay1eadd262020-12-29 16:42:08 -0800660 begin_function_definition(out);
David Tolnayebef4a22020-03-17 15:33:47 -0700661 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700662 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700663 write!(out, "::rust::repr::PtrLen ");
David Tolnayebef4a22020-03-17 15:33:47 -0700664 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700665 write_extern_return_type_space(out, &efn.ret);
David Tolnayebef4a22020-03-17 15:33:47 -0700666 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700667 let mangled = mangle::extern_fn(efn, out.types);
David Tolnay3caa50a2020-04-19 21:25:34 -0700668 write!(out, "{}(", mangled);
David Tolnaye439c772020-04-20 00:23:55 -0700669 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800670 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700671 write!(out, "const ");
672 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700673 write!(
674 out,
675 "{} &self",
David Tolnay1e5fe232021-01-01 18:11:40 -0800676 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700677 );
Joel Galenson3d4f6122020-04-07 15:54:05 -0700678 }
David Tolnay7db73692019-10-20 14:51:12 -0400679 for (i, arg) in efn.args.iter().enumerate() {
Joel Galenson3d4f6122020-04-07 15:54:05 -0700680 if i > 0 || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400681 write!(out, ", ");
682 }
David Tolnaya46a2372020-03-06 10:03:48 -0800683 if arg.ty == RustString {
684 write!(out, "const ");
David Tolnay313b10e2020-04-25 16:30:51 -0700685 } else if let Type::RustVec(_) = arg.ty {
686 write!(out, "const ");
David Tolnaya46a2372020-03-06 10:03:48 -0800687 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700688 write_extern_arg(out, arg);
David Tolnay7db73692019-10-20 14:51:12 -0400689 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700690 let indirect_return = indirect_return(efn, out.types);
David Tolnay7db73692019-10-20 14:51:12 -0400691 if indirect_return {
myronahne3b78ea2020-05-23 01:08:13 +0700692 if !efn.args.is_empty() || efn.receiver.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400693 write!(out, ", ");
694 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700695 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400696 write!(out, "*return$");
697 }
698 writeln!(out, ") noexcept {{");
699 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700700 write_return_type(out, &efn.ret);
Joel Galenson3d4f6122020-04-07 15:54:05 -0700701 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800702 None => write!(out, "(*{}$)(", efn.name.rust),
Adrian Taylorc8713432020-10-21 18:20:55 -0700703 Some(receiver) => write!(
704 out,
705 "({}::*{}$)(",
David Tolnay1e5fe232021-01-01 18:11:40 -0800706 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800707 efn.name.rust,
Adrian Taylorc8713432020-10-21 18:20:55 -0700708 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700709 }
David Tolnay7db73692019-10-20 14:51:12 -0400710 for (i, arg) in efn.args.iter().enumerate() {
711 if i > 0 {
712 write!(out, ", ");
713 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700714 write_type(out, &arg.ty);
David Tolnay7db73692019-10-20 14:51:12 -0400715 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700716 write!(out, ")");
David Tolnay4e7123f2020-04-19 21:11:37 -0700717 if let Some(receiver) = &efn.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800718 if !receiver.mutable {
David Tolnay4e7123f2020-04-19 21:11:37 -0700719 write!(out, " const");
720 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700721 }
722 write!(out, " = ");
723 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800724 None => write!(out, "{}", efn.name.to_fully_qualified()),
Adrian Taylorc8713432020-10-21 18:20:55 -0700725 Some(receiver) => write!(
726 out,
727 "&{}::{}",
David Tolnay1e5fe232021-01-01 18:11:40 -0800728 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
David Tolnay17a934c2020-11-02 00:40:04 -0800729 efn.name.cxx,
Adrian Taylorc8713432020-10-21 18:20:55 -0700730 ),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700731 }
732 writeln!(out, ";");
David Tolnay7db73692019-10-20 14:51:12 -0400733 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700734 if efn.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700735 out.builtin.ptr_len = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700736 out.builtin.trycatch = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700737 writeln!(out, "::rust::repr::PtrLen throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700738 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700739 writeln!(out, " [&] {{");
740 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700741 }
David Tolnay7db73692019-10-20 14:51:12 -0400742 if indirect_return {
David Tolnay0ecd05a2020-07-29 16:32:03 -0700743 out.include.new = true;
David Tolnay7db73692019-10-20 14:51:12 -0400744 write!(out, "new (return$) ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700745 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400746 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700747 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400748 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700749 }
750 match &efn.ret {
751 Some(Type::Ref(_)) => write!(out, "&"),
David Tolnay74d6d512020-10-31 22:22:03 -0700752 Some(Type::Str(_)) if !indirect_return => {
753 out.builtin.rust_str_repr = true;
754 write!(out, "::rust::impl<::rust::Str>::repr(");
755 }
David Tolnay73b72642020-11-25 17:44:05 -0800756 Some(ty @ Type::SliceRef(_)) if !indirect_return => {
David Tolnay36aa9e02020-10-31 23:08:21 -0700757 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -0800758 write!(out, "::rust::impl<");
759 write_type(out, ty);
760 write!(out, ">::repr(");
David Tolnayeb952ba2020-04-14 15:02:24 -0700761 }
David Tolnay99642622020-03-25 13:07:35 -0700762 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400763 }
Joel Galenson3d4f6122020-04-07 15:54:05 -0700764 match &efn.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800765 None => write!(out, "{}$(", efn.name.rust),
766 Some(_) => write!(out, "(self.*{}$)(", efn.name.rust),
Joel Galenson3d4f6122020-04-07 15:54:05 -0700767 }
David Tolnay7db73692019-10-20 14:51:12 -0400768 for (i, arg) in efn.args.iter().enumerate() {
769 if i > 0 {
770 write!(out, ", ");
771 }
772 if let Type::RustBox(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700773 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800774 write!(out, "::from_raw({})", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400775 } else if let Type::UniquePtr(_) = &arg.ty {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700776 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800777 write!(out, "({})", arg.name.cxx);
David Tolnay0356d332020-10-31 19:46:41 -0700778 } else if let Type::Str(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700779 out.builtin.rust_str_new_unchecked = true;
David Tolnay0356d332020-10-31 19:46:41 -0700780 write!(
781 out,
782 "::rust::impl<::rust::Str>::new_unchecked({})",
David Tolnay84ed6ad2021-01-01 15:30:14 -0800783 arg.name.cxx,
David Tolnay0356d332020-10-31 19:46:41 -0700784 );
David Tolnaya46a2372020-03-06 10:03:48 -0800785 } else if arg.ty == RustString {
David Tolnay74d6d512020-10-31 22:22:03 -0700786 out.builtin.unsafe_bitcopy = true;
David Tolnaycc3767f2020-03-06 10:41:51 -0800787 write!(
788 out,
789 "::rust::String(::rust::unsafe_bitcopy, *{})",
David Tolnay84ed6ad2021-01-01 15:30:14 -0800790 arg.name.cxx,
David Tolnaycc3767f2020-03-06 10:41:51 -0800791 );
David Tolnay313b10e2020-04-25 16:30:51 -0700792 } else if let Type::RustVec(_) = arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -0700793 out.builtin.unsafe_bitcopy = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700794 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800795 write!(out, "(::rust::unsafe_bitcopy, *{})", arg.name.cxx);
David Tolnay73b72642020-11-25 17:44:05 -0800796 } else if let Type::SliceRef(slice) = &arg.ty {
David Tolnayc5629f02020-11-23 18:32:46 -0800797 write_type(out, &arg.ty);
798 write!(out, "(static_cast<");
799 if slice.mutability.is_none() {
800 write!(out, "const ");
801 }
David Tolnay5515a9e2020-11-25 19:07:54 -0800802 write_type_space(out, &slice.inner);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800803 write!(out, "*>({0}.ptr), {0}.len)", arg.name.cxx);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700804 } else if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700805 out.include.utility = true;
David Tolnay84ed6ad2021-01-01 15:30:14 -0800806 write!(out, "::std::move(*{})", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400807 } else {
David Tolnay84ed6ad2021-01-01 15:30:14 -0800808 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400809 }
810 }
811 write!(out, ")");
812 match &efn.ret {
813 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
814 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -0800815 Some(Type::Str(_)) | Some(Type::SliceRef(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400816 _ => {}
817 }
818 if indirect_return {
819 write!(out, ")");
820 }
821 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700822 if efn.throws {
823 out.include.cstring = true;
David Tolnay1f010c62020-11-01 20:27:46 -0800824 out.builtin.exception = true;
David Tolnay5d121442020-03-17 22:14:40 -0700825 writeln!(out, " throw$.ptr = nullptr;");
826 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700827 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700828 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700829 writeln!(
830 out,
David Tolnayc5629f02020-11-23 18:32:46 -0800831 " throw$.ptr = const_cast<char *>(::cxxbridge1$exception(catch$, throw$.len));",
David Tolnayebef4a22020-03-17 15:33:47 -0700832 );
David Tolnay5d121442020-03-17 22:14:40 -0700833 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700834 writeln!(out, " return throw$;");
835 }
David Tolnay7db73692019-10-20 14:51:12 -0400836 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700837 for arg in &efn.args {
838 if let Type::Fn(f) = &arg.ty {
David Tolnay84ed6ad2021-01-01 15:30:14 -0800839 let var = &arg.name;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700840 write_function_pointer_trampoline(out, efn, var, f);
David Tolnay75dca2e2020-03-25 20:17:52 -0700841 }
842 }
David Tolnayca563ee2020-11-01 20:12:27 -0800843 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700844}
845
David Tolnay84ed6ad2021-01-01 15:30:14 -0800846fn write_function_pointer_trampoline(out: &mut OutFile, efn: &ExternFn, var: &Pair, f: &Signature) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700847 let r_trampoline = mangle::r_trampoline(efn, var, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700848 let indirect_call = true;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700849 write_rust_function_decl_impl(out, &r_trampoline, f, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700850
851 out.next_section();
David Tolnaya7c2ea12020-10-30 21:32:53 -0700852 let c_trampoline = mangle::c_trampoline(efn, var, out.types).to_string();
853 write_rust_function_shim_impl(out, &c_trampoline, f, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400854}
855
David Tolnay0b9b9f82020-11-01 20:41:00 -0800856fn write_rust_function_decl<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800857 out.set_namespace(&efn.name.namespace);
David Tolnayca563ee2020-11-01 20:12:27 -0800858 out.begin_block(Block::ExternC);
David Tolnaya7c2ea12020-10-30 21:32:53 -0700859 let link_name = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700860 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700861 write_rust_function_decl_impl(out, &link_name, efn, indirect_call);
David Tolnayca563ee2020-11-01 20:12:27 -0800862 out.end_block(Block::ExternC);
David Tolnay75dca2e2020-03-25 20:17:52 -0700863}
864
865fn write_rust_function_decl_impl(
866 out: &mut OutFile,
David Tolnay891061b2020-04-19 22:42:33 -0700867 link_name: &Symbol,
David Tolnay75dca2e2020-03-25 20:17:52 -0700868 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700869 indirect_call: bool,
870) {
David Tolnay04770742020-11-01 13:50:50 -0800871 out.next_section();
David Tolnay75dca2e2020-03-25 20:17:52 -0700872 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -0700873 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -0700874 write!(out, "::rust::repr::PtrLen ");
David Tolnay1e548172020-03-16 13:37:09 -0700875 } else {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700876 write_extern_return_type_space(out, &sig.ret);
David Tolnay1e548172020-03-16 13:37:09 -0700877 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700878 write!(out, "{}(", link_name);
879 let mut needs_comma = false;
David Tolnaye439c772020-04-20 00:23:55 -0700880 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800881 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700882 write!(out, "const ");
883 }
Adrian Taylorc8713432020-10-21 18:20:55 -0700884 write!(
885 out,
886 "{} &self",
David Tolnay1e5fe232021-01-01 18:11:40 -0800887 out.types.resolve(&receiver.ty).name.to_fully_qualified(),
Adrian Taylorc8713432020-10-21 18:20:55 -0700888 );
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700889 needs_comma = true;
890 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700891 for arg in &sig.args {
892 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400893 write!(out, ", ");
894 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700895 write_extern_arg(out, arg);
David Tolnay75dca2e2020-03-25 20:17:52 -0700896 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400897 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700898 if indirect_return(sig, out.types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700899 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400900 write!(out, ", ");
901 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700902 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400903 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700904 needs_comma = true;
905 }
906 if indirect_call {
907 if needs_comma {
908 write!(out, ", ");
909 }
910 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400911 }
912 writeln!(out, ") noexcept;");
913}
914
David Tolnay0b9b9f82020-11-01 20:41:00 -0800915fn write_rust_function_shim<'a>(out: &mut OutFile<'a>, efn: &'a ExternFn) {
David Tolnay17a934c2020-11-02 00:40:04 -0800916 out.set_namespace(&efn.name.namespace);
David Tolnay7db73692019-10-20 14:51:12 -0400917 for line in efn.doc.to_string().lines() {
918 writeln!(out, "//{}", line);
919 }
David Tolnaya73853b2020-04-20 01:19:56 -0700920 let local_name = match &efn.sig.receiver {
David Tolnay17a934c2020-11-02 00:40:04 -0800921 None => efn.name.cxx.to_string(),
David Tolnay1e5fe232021-01-01 18:11:40 -0800922 Some(receiver) => format!(
923 "{}::{}",
924 out.types.resolve(&receiver.ty).name.cxx,
925 efn.name.cxx,
926 ),
David Tolnaya73853b2020-04-20 01:19:56 -0700927 };
David Tolnaya7c2ea12020-10-30 21:32:53 -0700928 let invoke = mangle::extern_fn(efn, out.types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700929 let indirect_call = false;
David Tolnaya7c2ea12020-10-30 21:32:53 -0700930 write_rust_function_shim_impl(out, &local_name, efn, &invoke, indirect_call);
David Tolnay75dca2e2020-03-25 20:17:52 -0700931}
932
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700933fn write_rust_function_shim_decl(
David Tolnay75dca2e2020-03-25 20:17:52 -0700934 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700935 local_name: &str,
David Tolnay75dca2e2020-03-25 20:17:52 -0700936 sig: &Signature,
David Tolnay75dca2e2020-03-25 20:17:52 -0700937 indirect_call: bool,
938) {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700939 write_return_type(out, &sig.ret);
David Tolnay75dca2e2020-03-25 20:17:52 -0700940 write!(out, "{}(", local_name);
941 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400942 if i > 0 {
943 write!(out, ", ");
944 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700945 write_type_space(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800946 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -0400947 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700948 if indirect_call {
949 if !sig.args.is_empty() {
950 write!(out, ", ");
951 }
952 write!(out, "void *extern$");
953 }
David Tolnay1e548172020-03-16 13:37:09 -0700954 write!(out, ")");
David Tolnay86710612020-04-20 00:30:32 -0700955 if let Some(receiver) = &sig.receiver {
David Tolnay9c4ac2e2020-11-15 21:14:03 -0800956 if !receiver.mutable {
David Tolnay86710612020-04-20 00:30:32 -0700957 write!(out, " const");
958 }
959 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700960 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700961 write!(out, " noexcept");
962 }
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700963}
964
965fn write_rust_function_shim_impl(
966 out: &mut OutFile,
David Tolnaya73853b2020-04-20 01:19:56 -0700967 local_name: &str,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700968 sig: &Signature,
David Tolnay891061b2020-04-19 22:42:33 -0700969 invoke: &Symbol,
Joel Galensonc1c4e7a2020-04-15 10:21:00 -0700970 indirect_call: bool,
971) {
972 if out.header && sig.receiver.is_some() {
973 // We've already defined this inside the struct.
974 return;
975 }
David Tolnayb478fcb2020-12-29 16:48:02 -0800976 if !out.header {
977 begin_function_definition(out);
978 }
David Tolnaya7c2ea12020-10-30 21:32:53 -0700979 write_rust_function_shim_decl(out, local_name, sig, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400980 if out.header {
981 writeln!(out, ";");
David Tolnay439cde22020-04-20 00:46:25 -0700982 return;
David Tolnay7db73692019-10-20 14:51:12 -0400983 }
David Tolnay439cde22020-04-20 00:46:25 -0700984 writeln!(out, " {{");
985 for arg in &sig.args {
David Tolnaya7c2ea12020-10-30 21:32:53 -0700986 if arg.ty != RustString && out.types.needs_indirect_abi(&arg.ty) {
David Tolnay439cde22020-04-20 00:46:25 -0700987 out.include.utility = true;
David Tolnay74d6d512020-10-31 22:22:03 -0700988 out.builtin.manually_drop = true;
David Tolnay439cde22020-04-20 00:46:25 -0700989 write!(out, " ::rust::ManuallyDrop<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700990 write_type(out, &arg.ty);
David Tolnay84ed6ad2021-01-01 15:30:14 -0800991 writeln!(out, "> {}$(::std::move({0}));", arg.name.cxx);
David Tolnay439cde22020-04-20 00:46:25 -0700992 }
993 }
994 write!(out, " ");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700995 let indirect_return = indirect_return(sig, out.types);
David Tolnay439cde22020-04-20 00:46:25 -0700996 if indirect_return {
David Tolnay74d6d512020-10-31 22:22:03 -0700997 out.builtin.maybe_uninit = true;
David Tolnay439cde22020-04-20 00:46:25 -0700998 write!(out, "::rust::MaybeUninit<");
David Tolnaya7c2ea12020-10-30 21:32:53 -0700999 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay439cde22020-04-20 00:46:25 -07001000 writeln!(out, "> return$;");
1001 write!(out, " ");
1002 } else if let Some(ret) = &sig.ret {
1003 write!(out, "return ");
1004 match ret {
1005 Type::RustBox(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001006 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -07001007 write!(out, "::from_raw(");
1008 }
1009 Type::UniquePtr(_) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001010 write_type(out, ret);
David Tolnay439cde22020-04-20 00:46:25 -07001011 write!(out, "(");
1012 }
1013 Type::Ref(_) => write!(out, "*"),
David Tolnay74d6d512020-10-31 22:22:03 -07001014 Type::Str(_) => {
1015 out.builtin.rust_str_new_unchecked = true;
1016 write!(out, "::rust::impl<::rust::Str>::new_unchecked(");
1017 }
David Tolnay73b72642020-11-25 17:44:05 -08001018 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -07001019 out.builtin.rust_slice_new = true;
David Tolnayc5629f02020-11-23 18:32:46 -08001020 write!(out, "::rust::impl<");
1021 write_type(out, ret);
1022 write!(out, ">::slice(");
David Tolnay36aa9e02020-10-31 23:08:21 -07001023 }
David Tolnay439cde22020-04-20 00:46:25 -07001024 _ => {}
1025 }
1026 }
1027 if sig.throws {
David Tolnay919085c2020-10-31 22:32:22 -07001028 out.builtin.ptr_len = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001029 write!(out, "::rust::repr::PtrLen error$ = ");
David Tolnay439cde22020-04-20 00:46:25 -07001030 }
1031 write!(out, "{}(", invoke);
David Tolnay9d840362020-11-10 08:50:40 -08001032 let mut needs_comma = false;
David Tolnay439cde22020-04-20 00:46:25 -07001033 if sig.receiver.is_some() {
1034 write!(out, "*this");
David Tolnay9d840362020-11-10 08:50:40 -08001035 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001036 }
David Tolnay9d840362020-11-10 08:50:40 -08001037 for arg in &sig.args {
1038 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001039 write!(out, ", ");
1040 }
1041 match &arg.ty {
David Tolnay74d6d512020-10-31 22:22:03 -07001042 Type::Str(_) => {
1043 out.builtin.rust_str_repr = true;
1044 write!(out, "::rust::impl<::rust::Str>::repr(");
1045 }
David Tolnay73b72642020-11-25 17:44:05 -08001046 Type::SliceRef(_) => {
David Tolnay36aa9e02020-10-31 23:08:21 -07001047 out.builtin.rust_slice_repr = true;
David Tolnayc5629f02020-11-23 18:32:46 -08001048 write!(out, "::rust::impl<");
1049 write_type(out, &arg.ty);
1050 write!(out, ">::repr(");
David Tolnay36aa9e02020-10-31 23:08:21 -07001051 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001052 ty if out.types.needs_indirect_abi(ty) => write!(out, "&"),
David Tolnay439cde22020-04-20 00:46:25 -07001053 _ => {}
1054 }
David Tolnay84ed6ad2021-01-01 15:30:14 -08001055 write!(out, "{}", arg.name.cxx);
David Tolnay439cde22020-04-20 00:46:25 -07001056 match &arg.ty {
1057 Type::RustBox(_) => write!(out, ".into_raw()"),
1058 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnay73b72642020-11-25 17:44:05 -08001059 Type::Str(_) | Type::SliceRef(_) => write!(out, ")"),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001060 ty if ty != RustString && out.types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay439cde22020-04-20 00:46:25 -07001061 _ => {}
1062 }
David Tolnay9d840362020-11-10 08:50:40 -08001063 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001064 }
1065 if indirect_return {
David Tolnay9d840362020-11-10 08:50:40 -08001066 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001067 write!(out, ", ");
1068 }
1069 write!(out, "&return$.value");
David Tolnay9d840362020-11-10 08:50:40 -08001070 needs_comma = true;
David Tolnay439cde22020-04-20 00:46:25 -07001071 }
1072 if indirect_call {
David Tolnay9d840362020-11-10 08:50:40 -08001073 if needs_comma {
David Tolnay439cde22020-04-20 00:46:25 -07001074 write!(out, ", ");
1075 }
1076 write!(out, "extern$");
1077 }
1078 write!(out, ")");
David Tolnay22602b42020-09-21 18:04:05 -04001079 if !indirect_return {
1080 if let Some(ret) = &sig.ret {
David Tolnay73b72642020-11-25 17:44:05 -08001081 if let Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::SliceRef(_) = ret {
David Tolnay22602b42020-09-21 18:04:05 -04001082 write!(out, ")");
1083 }
David Tolnay439cde22020-04-20 00:46:25 -07001084 }
1085 }
1086 writeln!(out, ";");
1087 if sig.throws {
David Tolnay74d6d512020-10-31 22:22:03 -07001088 out.builtin.rust_error = true;
David Tolnayd68dfa82020-10-31 16:01:24 -07001089 writeln!(out, " if (error$.ptr) {{");
David Tolnay84ddf9e2020-10-31 15:36:48 -07001090 writeln!(out, " throw ::rust::impl<::rust::Error>::error(error$);");
David Tolnay439cde22020-04-20 00:46:25 -07001091 writeln!(out, " }}");
1092 }
1093 if indirect_return {
1094 out.include.utility = true;
1095 writeln!(out, " return ::std::move(return$.value);");
1096 }
1097 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001098}
1099
David Tolnaya7c2ea12020-10-30 21:32:53 -07001100fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001101 match ty {
1102 None => write!(out, "void "),
David Tolnaya7c2ea12020-10-30 21:32:53 -07001103 Some(ty) => write_type_space(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001104 }
1105}
1106
David Tolnay75dca2e2020-03-25 20:17:52 -07001107fn indirect_return(sig: &Signature, types: &Types) -> bool {
1108 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -07001109 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -07001110 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -07001111}
1112
David Tolnaya7c2ea12020-10-30 21:32:53 -07001113fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
David Tolnay99642622020-03-25 13:07:35 -07001114 match ty {
1115 Type::RustBox(ty) | Type::UniquePtr(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001116 write_type_space(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001117 write!(out, "*");
1118 }
1119 Type::Ref(ty) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001120 if !ty.mutable {
David Tolnay99642622020-03-25 13:07:35 -07001121 write!(out, "const ");
1122 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001123 write_type(out, &ty.inner);
David Tolnay99642622020-03-25 13:07:35 -07001124 write!(out, " *");
1125 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001126 _ => write_type(out, ty),
David Tolnay99642622020-03-25 13:07:35 -07001127 }
1128}
1129
David Tolnaya7c2ea12020-10-30 21:32:53 -07001130fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
1131 write_indirect_return_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001132 match ty {
1133 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
David Tolnay73b72642020-11-25 17:44:05 -08001134 Type::Str(_) | Type::SliceRef(_) => write!(out, " "),
David Tolnay99642622020-03-25 13:07:35 -07001135 _ => write_space_after_type(out, ty),
1136 }
1137}
1138
David Tolnaya7c2ea12020-10-30 21:32:53 -07001139fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>) {
David Tolnay7db73692019-10-20 14:51:12 -04001140 match ty {
1141 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001142 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001143 write!(out, "*");
1144 }
David Tolnay4a441222020-01-25 16:24:27 -08001145 Some(Type::Ref(ty)) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001146 if !ty.mutable {
David Tolnay4a441222020-01-25 16:24:27 -08001147 write!(out, "const ");
1148 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001149 write_type(out, &ty.inner);
David Tolnay4a441222020-01-25 16:24:27 -08001150 write!(out, " *");
1151 }
David Tolnay73b72642020-11-25 17:44:05 -08001152 Some(Type::Str(_)) | Some(Type::SliceRef(_)) => {
David Tolnay919085c2020-10-31 22:32:22 -07001153 out.builtin.ptr_len = true;
1154 write!(out, "::rust::repr::PtrLen ");
1155 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001156 Some(ty) if out.types.needs_indirect_abi(ty) => write!(out, "void "),
1157 _ => write_return_type(out, ty),
David Tolnay7db73692019-10-20 14:51:12 -04001158 }
1159}
1160
David Tolnaya7c2ea12020-10-30 21:32:53 -07001161fn write_extern_arg(out: &mut OutFile, arg: &Var) {
David Tolnay7db73692019-10-20 14:51:12 -04001162 match &arg.ty {
David Tolnay4377a9e2020-04-24 15:20:26 -07001163 Type::RustBox(ty) | Type::UniquePtr(ty) | Type::CxxVector(ty) => {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001164 write_type_space(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001165 write!(out, "*");
1166 }
David Tolnay73b72642020-11-25 17:44:05 -08001167 Type::Str(_) | Type::SliceRef(_) => {
David Tolnay919085c2020-10-31 22:32:22 -07001168 out.builtin.ptr_len = true;
1169 write!(out, "::rust::repr::PtrLen ");
1170 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001171 _ => write_type_space(out, &arg.ty),
David Tolnay7db73692019-10-20 14:51:12 -04001172 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001173 if out.types.needs_indirect_abi(&arg.ty) {
David Tolnay7db73692019-10-20 14:51:12 -04001174 write!(out, "*");
1175 }
David Tolnay84ed6ad2021-01-01 15:30:14 -08001176 write!(out, "{}", arg.name.cxx);
David Tolnay7db73692019-10-20 14:51:12 -04001177}
1178
David Tolnaya7c2ea12020-10-30 21:32:53 -07001179fn write_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001180 match ty {
Adrian Taylorc8713432020-10-21 18:20:55 -07001181 Type::Ident(ident) => match Atom::from(&ident.rust) {
David Tolnayf6a89f22020-05-10 23:39:27 -07001182 Some(atom) => write_atom(out, atom),
David Tolnay1e5fe232021-01-01 18:11:40 -08001183 None => write!(
1184 out,
1185 "{}",
1186 out.types.resolve(ident).name.to_fully_qualified(),
1187 ),
David Tolnay7db73692019-10-20 14:51:12 -04001188 },
1189 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -08001190 write!(out, "::rust::Box<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001191 write_type(out, &ty.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001192 write!(out, ">");
1193 }
Myron Ahneba35cf2020-02-05 19:41:51 +07001194 Type::RustVec(ty) => {
1195 write!(out, "::rust::Vec<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001196 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001197 write!(out, ">");
1198 }
David Tolnay7db73692019-10-20 14:51:12 -04001199 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -08001200 write!(out, "::std::unique_ptr<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001201 write_type(out, &ptr.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001202 write!(out, ">");
1203 }
David Tolnayb3b24a12020-12-01 15:27:43 -08001204 Type::SharedPtr(ptr) => {
1205 write!(out, "::std::shared_ptr<");
1206 write_type(out, &ptr.inner);
1207 write!(out, ">");
1208 }
David Tolnay215e77f2020-12-28 17:09:48 -08001209 Type::WeakPtr(ptr) => {
1210 write!(out, "::std::weak_ptr<");
1211 write_type(out, &ptr.inner);
1212 write!(out, ">");
1213 }
David Tolnay4377a9e2020-04-24 15:20:26 -07001214 Type::CxxVector(ty) => {
Myron Ahneba35cf2020-02-05 19:41:51 +07001215 write!(out, "::std::vector<");
David Tolnaya7c2ea12020-10-30 21:32:53 -07001216 write_type(out, &ty.inner);
Myron Ahneba35cf2020-02-05 19:41:51 +07001217 write!(out, ">");
1218 }
David Tolnay7db73692019-10-20 14:51:12 -04001219 Type::Ref(r) => {
David Tolnay9c4ac2e2020-11-15 21:14:03 -08001220 if !r.mutable {
David Tolnay7db73692019-10-20 14:51:12 -04001221 write!(out, "const ");
1222 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001223 write_type(out, &r.inner);
David Tolnay7db73692019-10-20 14:51:12 -04001224 write!(out, " &");
1225 }
1226 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -08001227 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -04001228 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001229 Type::SliceRef(slice) => {
David Tolnayc5629f02020-11-23 18:32:46 -08001230 write!(out, "::rust::Slice<");
David Tolnay5515a9e2020-11-25 19:07:54 -08001231 if slice.mutability.is_none() {
David Tolnayc5629f02020-11-23 18:32:46 -08001232 write!(out, "const ");
1233 }
David Tolnay5515a9e2020-11-25 19:07:54 -08001234 write_type(out, &slice.inner);
1235 write!(out, ">");
Adrian Taylorf5dd5522020-04-13 16:50:14 -07001236 }
David Tolnay75dca2e2020-03-25 20:17:52 -07001237 Type::Fn(f) => {
David Tolnayf031c322020-11-29 19:41:33 -08001238 write!(out, "::rust::Fn<");
David Tolnay75dca2e2020-03-25 20:17:52 -07001239 match &f.ret {
David Tolnaya7c2ea12020-10-30 21:32:53 -07001240 Some(ret) => write_type(out, ret),
David Tolnay75dca2e2020-03-25 20:17:52 -07001241 None => write!(out, "void"),
1242 }
1243 write!(out, "(");
1244 for (i, arg) in f.args.iter().enumerate() {
1245 if i > 0 {
1246 write!(out, ", ");
1247 }
David Tolnaya7c2ea12020-10-30 21:32:53 -07001248 write_type(out, &arg.ty);
David Tolnay75dca2e2020-03-25 20:17:52 -07001249 }
1250 write!(out, ")>");
1251 }
Xiangpeng Hao78762352020-11-12 10:24:18 +08001252 Type::Array(a) => {
1253 write!(out, "::std::array<");
1254 write_type(out, &a.inner);
1255 write!(out, ", {}>", &a.len);
1256 }
David Tolnay2fb14e92020-03-15 23:11:38 -07001257 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001258 }
1259}
1260
David Tolnayf6a89f22020-05-10 23:39:27 -07001261fn write_atom(out: &mut OutFile, atom: Atom) {
1262 match atom {
1263 Bool => write!(out, "bool"),
David Tolnayb3873dc2020-11-25 19:47:49 -08001264 Char => write!(out, "char"),
David Tolnaybe3cbf72020-12-12 22:12:07 -08001265 U8 => write!(out, "::std::uint8_t"),
1266 U16 => write!(out, "::std::uint16_t"),
1267 U32 => write!(out, "::std::uint32_t"),
1268 U64 => write!(out, "::std::uint64_t"),
1269 Usize => write!(out, "::std::size_t"),
1270 I8 => write!(out, "::std::int8_t"),
1271 I16 => write!(out, "::std::int16_t"),
1272 I32 => write!(out, "::std::int32_t"),
1273 I64 => write!(out, "::std::int64_t"),
David Tolnayf6a89f22020-05-10 23:39:27 -07001274 Isize => write!(out, "::rust::isize"),
1275 F32 => write!(out, "float"),
1276 F64 => write!(out, "double"),
1277 CxxString => write!(out, "::std::string"),
1278 RustString => write!(out, "::rust::String"),
1279 }
1280}
1281
David Tolnaya7c2ea12020-10-30 21:32:53 -07001282fn write_type_space(out: &mut OutFile, ty: &Type) {
1283 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -07001284 write_space_after_type(out, ty);
1285}
1286
1287fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -04001288 match ty {
David Tolnayeb952ba2020-04-14 15:02:24 -07001289 Type::Ident(_)
1290 | Type::RustBox(_)
1291 | Type::UniquePtr(_)
David Tolnayb3b24a12020-12-01 15:27:43 -08001292 | Type::SharedPtr(_)
David Tolnay215e77f2020-12-28 17:09:48 -08001293 | Type::WeakPtr(_)
David Tolnayeb952ba2020-04-14 15:02:24 -07001294 | Type::Str(_)
David Tolnay4377a9e2020-04-24 15:20:26 -07001295 | Type::CxxVector(_)
Myron Ahneba35cf2020-02-05 19:41:51 +07001296 | Type::RustVec(_)
David Tolnay73b72642020-11-25 17:44:05 -08001297 | Type::SliceRef(_)
Xiangpeng Hao78762352020-11-12 10:24:18 +08001298 | Type::Fn(_)
1299 | Type::Array(_) => write!(out, " "),
David Tolnay7db73692019-10-20 14:51:12 -04001300 Type::Ref(_) => {}
David Tolnaye0dca7b2020-11-25 17:18:57 -08001301 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -04001302 }
1303}
1304
David Tolnay1bdb4712020-11-25 07:27:54 -08001305#[derive(Copy, Clone)]
1306enum UniquePtr<'a> {
David Tolnay3abed472020-12-31 23:34:53 -08001307 Ident(&'a Ident),
1308 CxxVector(&'a Ident),
David Tolnay1bdb4712020-11-25 07:27:54 -08001309}
1310
1311trait ToTypename {
1312 fn to_typename(&self, types: &Types) -> String;
1313}
1314
David Tolnay3abed472020-12-31 23:34:53 -08001315impl ToTypename for Ident {
David Tolnay1bdb4712020-11-25 07:27:54 -08001316 fn to_typename(&self, types: &Types) -> String {
David Tolnay1e5fe232021-01-01 18:11:40 -08001317 types.resolve(self).name.to_fully_qualified()
David Tolnay2eca4a02020-04-24 19:50:51 -07001318 }
1319}
1320
David Tolnay1bdb4712020-11-25 07:27:54 -08001321impl<'a> ToTypename for UniquePtr<'a> {
1322 fn to_typename(&self, types: &Types) -> String {
1323 match self {
1324 UniquePtr::Ident(ident) => ident.to_typename(types),
1325 UniquePtr::CxxVector(element) => {
1326 format!("::std::vector<{}>", element.to_typename(types))
1327 }
1328 }
1329 }
1330}
1331
1332trait ToMangled {
1333 fn to_mangled(&self, types: &Types) -> Symbol;
1334}
1335
David Tolnay3abed472020-12-31 23:34:53 -08001336impl ToMangled for Ident {
David Tolnay1bdb4712020-11-25 07:27:54 -08001337 fn to_mangled(&self, types: &Types) -> Symbol {
David Tolnay1e5fe232021-01-01 18:11:40 -08001338 types.resolve(self).name.to_symbol()
David Tolnay1bdb4712020-11-25 07:27:54 -08001339 }
1340}
1341
1342impl<'a> ToMangled for UniquePtr<'a> {
1343 fn to_mangled(&self, types: &Types) -> Symbol {
1344 match self {
1345 UniquePtr::Ident(ident) => ident.to_mangled(types),
1346 UniquePtr::CxxVector(element) => element.to_mangled(types).prefix_with("std$vector$"),
1347 }
David Tolnaybae50ef2020-04-25 12:38:41 -07001348 }
1349}
1350
David Tolnaya7c2ea12020-10-30 21:32:53 -07001351fn write_generic_instantiations(out: &mut OutFile) {
David Tolnay169bb472020-11-01 21:04:24 -08001352 if out.header {
1353 return;
1354 }
1355
1356 out.next_section();
David Tolnay078c90f2020-11-01 13:31:08 -08001357 out.set_namespace(Default::default());
David Tolnay0c033e32020-11-01 15:15:48 -08001358 out.begin_block(Block::ExternC);
David Tolnay3abed472020-12-31 23:34:53 -08001359 for impl_key in out.types.impls.keys() {
1360 out.next_section();
1361 match impl_key {
1362 ImplKey::RustBox(ident) => write_rust_box_extern(out, ident),
1363 ImplKey::RustVec(ident) => write_rust_vec_extern(out, ident),
1364 ImplKey::UniquePtr(ident) => write_unique_ptr(out, ident),
1365 ImplKey::SharedPtr(ident) => write_shared_ptr(out, ident),
1366 ImplKey::WeakPtr(ident) => write_weak_ptr(out, ident),
1367 ImplKey::CxxVector(ident) => write_cxx_vector(out, ident),
David Tolnay7db73692019-10-20 14:51:12 -04001368 }
1369 }
David Tolnay0c033e32020-11-01 15:15:48 -08001370 out.end_block(Block::ExternC);
David Tolnay7db73692019-10-20 14:51:12 -04001371
David Tolnay0c033e32020-11-01 15:15:48 -08001372 out.begin_block(Block::Namespace("rust"));
David Tolnay0f0162f2020-11-16 23:43:37 -08001373 out.begin_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay3abed472020-12-31 23:34:53 -08001374 for impl_key in out.types.impls.keys() {
1375 match impl_key {
1376 ImplKey::RustBox(ident) => write_rust_box_impl(out, ident),
1377 ImplKey::RustVec(ident) => write_rust_vec_impl(out, ident),
1378 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -04001379 }
1380 }
David Tolnay0f0162f2020-11-16 23:43:37 -08001381 out.end_block(Block::InlineNamespace("cxxbridge1"));
David Tolnay0c033e32020-11-01 15:15:48 -08001382 out.end_block(Block::Namespace("rust"));
David Tolnay7db73692019-10-20 14:51:12 -04001383}
1384
David Tolnay3abed472020-12-31 23:34:53 -08001385fn write_rust_box_extern(out: &mut OutFile, ident: &Ident) {
1386 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001387 let inner = resolve.name.to_fully_qualified();
1388 let instance = resolve.name.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001389
David Tolnay7db73692019-10-20 14:51:12 -04001390 writeln!(
1391 out,
David Tolnayc4b34222020-12-12 13:06:26 -08001392 "{} *cxxbridge1$box${}$alloc() noexcept;",
1393 inner, instance,
David Tolnay7db73692019-10-20 14:51:12 -04001394 );
1395 writeln!(
1396 out,
David Tolnay25b3c1d2020-12-12 15:50:10 -08001397 "void cxxbridge1$box${}$dealloc({} *) noexcept;",
1398 instance, inner,
1399 );
1400 writeln!(
1401 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001402 "void cxxbridge1$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -04001403 instance, inner,
1404 );
David Tolnay7db73692019-10-20 14:51:12 -04001405}
1406
David Tolnay3abed472020-12-31 23:34:53 -08001407fn write_rust_vec_extern(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001408 let inner = element.to_typename(out.types);
1409 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001410
David Tolnay12320e12020-12-09 23:09:36 -08001411 out.include.cstddef = true;
1412
Myron Ahneba35cf2020-02-05 19:41:51 +07001413 writeln!(
1414 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001415 "void cxxbridge1$rust_vec${}$new(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnayf97c2d52020-04-25 16:37:48 -07001416 instance, inner,
1417 );
1418 writeln!(
1419 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001420 "void cxxbridge1$rust_vec${}$drop(::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001421 instance, inner,
1422 );
1423 writeln!(
1424 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001425 "::std::size_t cxxbridge1$rust_vec${}$len(const ::rust::Vec<{}> *ptr) noexcept;",
Myron Ahneba35cf2020-02-05 19:41:51 +07001426 instance, inner,
1427 );
David Tolnay219c0792020-04-24 20:31:37 -07001428 writeln!(
1429 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001430 "::std::size_t cxxbridge1$rust_vec${}$capacity(const ::rust::Vec<{}> *ptr) noexcept;",
David Tolnaydc62d712020-12-11 13:51:53 -08001431 instance, inner,
1432 );
1433 writeln!(
1434 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001435 "const {} *cxxbridge1$rust_vec${}$data(const ::rust::Vec<{0}> *ptr) noexcept;",
David Tolnay219c0792020-04-24 20:31:37 -07001436 inner, instance,
1437 );
David Tolnay503d0192020-04-24 22:18:56 -07001438 writeln!(
1439 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001440 "void cxxbridge1$rust_vec${}$reserve_total(::rust::Vec<{}> *ptr, ::std::size_t cap) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001441 instance, inner,
1442 );
1443 writeln!(
1444 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001445 "void cxxbridge1$rust_vec${}$set_len(::rust::Vec<{}> *ptr, ::std::size_t len) noexcept;",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001446 instance, inner,
1447 );
Myron Ahneba35cf2020-02-05 19:41:51 +07001448}
1449
David Tolnay3abed472020-12-31 23:34:53 -08001450fn write_rust_box_impl(out: &mut OutFile, ident: &Ident) {
1451 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001452 let inner = resolve.name.to_fully_qualified();
1453 let instance = resolve.name.to_symbol();
David Tolnay7db73692019-10-20 14:51:12 -04001454
1455 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001456 begin_function_definition(out);
David Tolnaye5703162020-12-12 16:26:35 -08001457 writeln!(
1458 out,
1459 "{} *Box<{}>::allocation::alloc() noexcept {{",
1460 inner, inner,
1461 );
David Tolnayc4b34222020-12-12 13:06:26 -08001462 writeln!(out, " return cxxbridge1$box${}$alloc();", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001463 writeln!(out, "}}");
1464
1465 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001466 begin_function_definition(out);
David Tolnay25b3c1d2020-12-12 15:50:10 -08001467 writeln!(
1468 out,
David Tolnaye5703162020-12-12 16:26:35 -08001469 "void Box<{}>::allocation::dealloc({} *ptr) noexcept {{",
David Tolnay25b3c1d2020-12-12 15:50:10 -08001470 inner, inner,
1471 );
1472 writeln!(out, " cxxbridge1$box${}$dealloc(ptr);", instance);
1473 writeln!(out, "}}");
1474
1475 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001476 begin_function_definition(out);
David Tolnay324437a2020-03-01 13:02:24 -08001477 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001478 writeln!(out, " cxxbridge1$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -04001479 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001480}
1481
David Tolnay3abed472020-12-31 23:34:53 -08001482fn write_rust_vec_impl(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001483 let inner = element.to_typename(out.types);
1484 let instance = element.to_mangled(out.types);
David Tolnay4791f1c2020-03-17 21:53:16 -07001485
David Tolnay12320e12020-12-09 23:09:36 -08001486 out.include.cstddef = true;
1487
Myron Ahneba35cf2020-02-05 19:41:51 +07001488 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001489 begin_function_definition(out);
David Tolnayf97c2d52020-04-25 16:37:48 -07001490 writeln!(out, "Vec<{}>::Vec() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001491 writeln!(out, " cxxbridge1$rust_vec${}$new(this);", instance);
David Tolnayf97c2d52020-04-25 16:37:48 -07001492 writeln!(out, "}}");
1493
1494 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001495 begin_function_definition(out);
Myron Ahneba35cf2020-02-05 19:41:51 +07001496 writeln!(out, "void Vec<{}>::drop() noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001497 writeln!(out, " return cxxbridge1$rust_vec${}$drop(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001498 writeln!(out, "}}");
1499
1500 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001501 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001502 writeln!(
1503 out,
1504 "::std::size_t Vec<{}>::size() const noexcept {{",
1505 inner,
1506 );
David Tolnay0f0162f2020-11-16 23:43:37 -08001507 writeln!(out, " return cxxbridge1$rust_vec${}$len(this);", instance);
Myron Ahneba35cf2020-02-05 19:41:51 +07001508 writeln!(out, "}}");
David Tolnay219c0792020-04-24 20:31:37 -07001509
1510 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001511 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001512 writeln!(
1513 out,
1514 "::std::size_t Vec<{}>::capacity() const noexcept {{",
1515 inner,
1516 );
David Tolnay381d9532020-12-11 13:59:45 -08001517 writeln!(
1518 out,
1519 " return cxxbridge1$rust_vec${}$capacity(this);",
1520 instance,
1521 );
David Tolnaydc62d712020-12-11 13:51:53 -08001522 writeln!(out, "}}");
1523
1524 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001525 begin_function_definition(out);
David Tolnay219c0792020-04-24 20:31:37 -07001526 writeln!(out, "const {} *Vec<{0}>::data() const noexcept {{", inner);
David Tolnay0f0162f2020-11-16 23:43:37 -08001527 writeln!(out, " return cxxbridge1$rust_vec${}$data(this);", instance);
David Tolnay219c0792020-04-24 20:31:37 -07001528 writeln!(out, "}}");
David Tolnay503d0192020-04-24 22:18:56 -07001529
1530 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001531 begin_function_definition(out);
David Tolnayfb6b73c2020-11-10 14:32:16 -08001532 writeln!(
1533 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001534 "void Vec<{}>::reserve_total(::std::size_t cap) noexcept {{",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001535 inner,
1536 );
1537 writeln!(
1538 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001539 " return cxxbridge1$rust_vec${}$reserve_total(this, cap);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001540 instance,
1541 );
1542 writeln!(out, "}}");
1543
1544 writeln!(out, "template <>");
David Tolnayb478fcb2020-12-29 16:48:02 -08001545 begin_function_definition(out);
David Tolnaybe3cbf72020-12-12 22:12:07 -08001546 writeln!(
1547 out,
1548 "void Vec<{}>::set_len(::std::size_t len) noexcept {{",
1549 inner,
1550 );
David Tolnayfb6b73c2020-11-10 14:32:16 -08001551 writeln!(
1552 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001553 " return cxxbridge1$rust_vec${}$set_len(this, len);",
David Tolnayfb6b73c2020-11-10 14:32:16 -08001554 instance,
1555 );
1556 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001557}
1558
David Tolnay3abed472020-12-31 23:34:53 -08001559fn write_unique_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001560 let ty = UniquePtr::Ident(ident);
David Tolnay1bdb4712020-11-25 07:27:54 -08001561 write_unique_ptr_common(out, ty);
David Tolnay63da4d32020-04-25 09:41:12 -07001562}
1563
1564// Shared by UniquePtr<T> and UniquePtr<CxxVector<T>>.
David Tolnay1bdb4712020-11-25 07:27:54 -08001565fn write_unique_ptr_common(out: &mut OutFile, ty: UniquePtr) {
David Tolnay0ecd05a2020-07-29 16:32:03 -07001566 out.include.new = true;
Myron Ahneba35cf2020-02-05 19:41:51 +07001567 out.include.utility = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001568 let inner = ty.to_typename(out.types);
1569 let instance = ty.to_mangled(out.types);
David Tolnay7db73692019-10-20 14:51:12 -04001570
David Tolnay63da4d32020-04-25 09:41:12 -07001571 let can_construct_from_value = match ty {
David Tolnayca0f9da2020-10-16 13:16:17 -07001572 // Some aliases are to opaque types; some are to trivial types. We can't
1573 // know at code generation time, so we generate both C++ and Rust side
1574 // bindings for a "new" method anyway. But the Rust code can't be called
1575 // for Opaque types because the 'new' method is not implemented.
David Tolnay1bdb4712020-11-25 07:27:54 -08001576 UniquePtr::Ident(ident) => {
David Tolnay3abed472020-12-31 23:34:53 -08001577 out.types.structs.contains_key(ident)
1578 || out.types.enums.contains_key(ident)
1579 || out.types.aliases.contains_key(ident)
David Tolnayca0f9da2020-10-16 13:16:17 -07001580 }
David Tolnay1bdb4712020-11-25 07:27:54 -08001581 UniquePtr::CxxVector(_) => false,
David Tolnay63da4d32020-04-25 09:41:12 -07001582 };
1583
David Tolnay53462762020-11-25 07:08:10 -08001584 let conditional_delete = match ty {
1585 UniquePtr::Ident(ident) => {
David Tolnay3abed472020-12-31 23:34:53 -08001586 !out.types.structs.contains_key(ident) && !out.types.enums.contains_key(ident)
David Tolnay53462762020-11-25 07:08:10 -08001587 }
1588 UniquePtr::CxxVector(_) => false,
1589 };
1590
1591 if conditional_delete {
1592 out.builtin.is_complete = true;
1593 let definition = match ty {
David Tolnay1e5fe232021-01-01 18:11:40 -08001594 UniquePtr::Ident(ty) => &out.types.resolve(ty).name.cxx,
David Tolnay53462762020-11-25 07:08:10 -08001595 UniquePtr::CxxVector(_) => unreachable!(),
1596 };
1597 writeln!(
1598 out,
David Tolnay75068632020-12-26 22:15:17 -08001599 "static_assert(::rust::detail::is_complete<{}>::value, \"definition of {} is required\");",
David Tolnay53462762020-11-25 07:08:10 -08001600 inner, definition,
1601 );
1602 }
David Tolnay7db73692019-10-20 14:51:12 -04001603 writeln!(
1604 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001605 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001606 inner,
1607 );
1608 writeln!(
1609 out,
David Tolnay7e219b82020-03-01 13:14:51 -08001610 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -04001611 inner,
1612 );
1613 writeln!(
1614 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001615 "void cxxbridge1$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001616 instance, inner,
1617 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001618 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001619 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001620 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001621 out.builtin.maybe_uninit = true;
David Tolnay63da4d32020-04-25 09:41:12 -07001622 writeln!(
David Tolnay53838912020-04-09 20:56:44 -07001623 out,
David Tolnay0b881402020-12-01 21:05:08 -08001624 "{} *cxxbridge1$unique_ptr${}$uninit(::std::unique_ptr<{}> *ptr) noexcept {{",
1625 inner, instance, inner,
David Tolnay53838912020-04-09 20:56:44 -07001626 );
David Tolnay63da4d32020-04-25 09:41:12 -07001627 writeln!(
1628 out,
David Tolnay0b881402020-12-01 21:05:08 -08001629 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1630 inner, inner, inner,
David Tolnay63da4d32020-04-25 09:41:12 -07001631 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001632 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001633 writeln!(out, " return uninit;");
David Tolnay63da4d32020-04-25 09:41:12 -07001634 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -07001635 }
David Tolnay7db73692019-10-20 14:51:12 -04001636 writeln!(
1637 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001638 "void cxxbridge1$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001639 instance, inner, inner,
1640 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001641 writeln!(out, " ::new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -04001642 writeln!(out, "}}");
1643 writeln!(
1644 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001645 "const {} *cxxbridge1$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001646 inner, instance, inner,
1647 );
1648 writeln!(out, " return ptr.get();");
1649 writeln!(out, "}}");
1650 writeln!(
1651 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001652 "{} *cxxbridge1$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001653 inner, instance, inner,
1654 );
1655 writeln!(out, " return ptr.release();");
1656 writeln!(out, "}}");
1657 writeln!(
1658 out,
David Tolnay0f0162f2020-11-16 23:43:37 -08001659 "void cxxbridge1$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -04001660 instance, inner,
1661 );
David Tolnay53462762020-11-25 07:08:10 -08001662 if conditional_delete {
1663 out.builtin.deleter_if = true;
1664 writeln!(
1665 out,
David Tolnay75068632020-12-26 22:15:17 -08001666 " ::rust::deleter_if<::rust::detail::is_complete<{}>::value>{{}}(ptr);",
David Tolnay53462762020-11-25 07:08:10 -08001667 inner,
1668 );
1669 } else {
1670 writeln!(out, " ptr->~unique_ptr();");
1671 }
David Tolnay7db73692019-10-20 14:51:12 -04001672 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -04001673}
Myron Ahneba35cf2020-02-05 19:41:51 +07001674
David Tolnay3abed472020-12-31 23:34:53 -08001675fn write_shared_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay95bc57f2021-01-01 13:29:44 -08001676 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001677 let inner = resolve.name.to_fully_qualified();
1678 let instance = resolve.name.to_symbol();
David Tolnayb3b24a12020-12-01 15:27:43 -08001679
David Tolnayb3b24a12020-12-01 15:27:43 -08001680 out.include.new = true;
1681 out.include.utility = true;
1682
1683 // Some aliases are to opaque types; some are to trivial types. We can't
1684 // know at code generation time, so we generate both C++ and Rust side
1685 // bindings for a "new" method anyway. But the Rust code can't be called for
1686 // Opaque types because the 'new' method is not implemented.
David Tolnay3abed472020-12-31 23:34:53 -08001687 let can_construct_from_value = out.types.structs.contains_key(ident)
1688 || out.types.enums.contains_key(ident)
1689 || out.types.aliases.contains_key(ident);
David Tolnayb3b24a12020-12-01 15:27:43 -08001690
David Tolnayb3b24a12020-12-01 15:27:43 -08001691 writeln!(
1692 out,
1693 "static_assert(sizeof(::std::shared_ptr<{}>) == 2 * sizeof(void *), \"\");",
1694 inner,
1695 );
1696 writeln!(
1697 out,
1698 "static_assert(alignof(::std::shared_ptr<{}>) == alignof(void *), \"\");",
1699 inner,
1700 );
1701 writeln!(
1702 out,
1703 "void cxxbridge1$shared_ptr${}$null(::std::shared_ptr<{}> *ptr) noexcept {{",
1704 instance, inner,
1705 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001706 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>();", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001707 writeln!(out, "}}");
1708 if can_construct_from_value {
David Tolnay0b881402020-12-01 21:05:08 -08001709 out.builtin.maybe_uninit = true;
David Tolnayb3b24a12020-12-01 15:27:43 -08001710 writeln!(
1711 out,
David Tolnay0b881402020-12-01 21:05:08 -08001712 "{} *cxxbridge1$shared_ptr${}$uninit(::std::shared_ptr<{}> *ptr) noexcept {{",
1713 inner, instance, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001714 );
1715 writeln!(
1716 out,
David Tolnay0b881402020-12-01 21:05:08 -08001717 " {} *uninit = reinterpret_cast<{} *>(new ::rust::MaybeUninit<{}>);",
1718 inner, inner, inner,
David Tolnayb3b24a12020-12-01 15:27:43 -08001719 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001720 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(uninit);", inner);
David Tolnay0b881402020-12-01 21:05:08 -08001721 writeln!(out, " return uninit;");
David Tolnayb3b24a12020-12-01 15:27:43 -08001722 writeln!(out, "}}");
1723 }
1724 writeln!(
1725 out,
1726 "void cxxbridge1$shared_ptr${}$clone(const ::std::shared_ptr<{}>& self, ::std::shared_ptr<{}> *ptr) noexcept {{",
1727 instance, inner, inner,
1728 );
David Tolnay718ca0f2020-12-09 23:01:11 -08001729 writeln!(out, " ::new (ptr) ::std::shared_ptr<{}>(self);", inner);
David Tolnayb3b24a12020-12-01 15:27:43 -08001730 writeln!(out, "}}");
1731 writeln!(
1732 out,
1733 "const {} *cxxbridge1$shared_ptr${}$get(const ::std::shared_ptr<{}>& self) noexcept {{",
1734 inner, instance, inner,
1735 );
1736 writeln!(out, " return self.get();");
1737 writeln!(out, "}}");
1738 writeln!(
1739 out,
1740 "void cxxbridge1$shared_ptr${}$drop(::std::shared_ptr<{}> *self) noexcept {{",
1741 instance, inner,
1742 );
1743 writeln!(out, " self->~shared_ptr();");
1744 writeln!(out, "}}");
David Tolnayb3b24a12020-12-01 15:27:43 -08001745}
1746
David Tolnay3abed472020-12-31 23:34:53 -08001747fn write_weak_ptr(out: &mut OutFile, ident: &Ident) {
David Tolnay95bc57f2021-01-01 13:29:44 -08001748 let resolve = out.types.resolve(ident);
David Tolnay1e5fe232021-01-01 18:11:40 -08001749 let inner = resolve.name.to_fully_qualified();
1750 let instance = resolve.name.to_symbol();
David Tolnay215e77f2020-12-28 17:09:48 -08001751
1752 out.include.new = true;
1753 out.include.utility = true;
1754
1755 writeln!(
1756 out,
1757 "static_assert(sizeof(::std::weak_ptr<{}>) == 2 * sizeof(void *), \"\");",
1758 inner,
1759 );
1760 writeln!(
1761 out,
1762 "static_assert(alignof(::std::weak_ptr<{}>) == alignof(void *), \"\");",
1763 inner,
1764 );
1765 writeln!(
1766 out,
1767 "void cxxbridge1$weak_ptr${}$null(::std::weak_ptr<{}> *ptr) noexcept {{",
1768 instance, inner,
1769 );
1770 writeln!(out, " ::new (ptr) ::std::weak_ptr<{}>();", inner);
1771 writeln!(out, "}}");
1772 writeln!(
1773 out,
1774 "void cxxbridge1$weak_ptr${}$clone(const ::std::weak_ptr<{}>& self, ::std::weak_ptr<{}> *ptr) noexcept {{",
1775 instance, inner, inner,
1776 );
1777 writeln!(out, " ::new (ptr) ::std::weak_ptr<{}>(self);", inner);
1778 writeln!(out, "}}");
1779 writeln!(
1780 out,
David Tolnay85b6bc42020-12-28 17:47:51 -08001781 "void cxxbridge1$weak_ptr${}$downgrade(const ::std::shared_ptr<{}>& shared, ::std::weak_ptr<{}> *weak) noexcept {{",
1782 instance, inner, inner,
1783 );
1784 writeln!(out, " ::new (weak) ::std::weak_ptr<{}>(shared);", inner);
1785 writeln!(out, "}}");
1786 writeln!(
1787 out,
David Tolnay7a487852020-12-28 18:02:25 -08001788 "void cxxbridge1$weak_ptr${}$upgrade(const ::std::weak_ptr<{}>& weak, ::std::shared_ptr<{}> *shared) noexcept {{",
1789 instance, inner, inner,
1790 );
1791 writeln!(
1792 out,
1793 " ::new (shared) ::std::shared_ptr<{}>(weak.lock());",
1794 inner,
1795 );
1796 writeln!(out, "}}");
1797 writeln!(
1798 out,
David Tolnay215e77f2020-12-28 17:09:48 -08001799 "void cxxbridge1$weak_ptr${}$drop(::std::weak_ptr<{}> *self) noexcept {{",
1800 instance, inner,
1801 );
1802 writeln!(out, " self->~weak_ptr();");
1803 writeln!(out, "}}");
1804}
1805
David Tolnay3abed472020-12-31 23:34:53 -08001806fn write_cxx_vector(out: &mut OutFile, element: &Ident) {
David Tolnay1bdb4712020-11-25 07:27:54 -08001807 let inner = element.to_typename(out.types);
1808 let instance = element.to_mangled(out.types);
Myron Ahneba35cf2020-02-05 19:41:51 +07001809
David Tolnay12320e12020-12-09 23:09:36 -08001810 out.include.cstddef = true;
1811
Myron Ahneba35cf2020-02-05 19:41:51 +07001812 writeln!(
1813 out,
David Tolnaybe3cbf72020-12-12 22:12:07 -08001814 "::std::size_t cxxbridge1$std$vector${}$size(const ::std::vector<{}> &s) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001815 instance, inner,
1816 );
1817 writeln!(out, " return s.size();");
1818 writeln!(out, "}}");
Myron Ahneba35cf2020-02-05 19:41:51 +07001819 writeln!(
1820 out,
David Tolnay767e00d2020-12-21 17:12:27 -08001821 "{} *cxxbridge1$std$vector${}$get_unchecked(::std::vector<{}> *s, ::std::size_t pos) noexcept {{",
Myron Ahneba35cf2020-02-05 19:41:51 +07001822 inner, instance, inner,
1823 );
David Tolnay767e00d2020-12-21 17:12:27 -08001824 writeln!(out, " return &(*s)[pos];");
Myron Ahneba35cf2020-02-05 19:41:51 +07001825 writeln!(out, "}}");
David Tolnay63da4d32020-04-25 09:41:12 -07001826
David Tolnay70820672020-12-07 11:15:14 -08001827 out.include.memory = true;
David Tolnay1bdb4712020-11-25 07:27:54 -08001828 write_unique_ptr_common(out, UniquePtr::CxxVector(element));
Myron Ahneba35cf2020-02-05 19:41:51 +07001829}