blob: d80e957954929c7df1d279f4e82ff3aed65ee41d [file] [log] [blame]
David Tolnay754e21c2020-03-29 20:58:46 -07001use crate::gen::namespace::Namespace;
David Tolnay7db73692019-10-20 14:51:12 -04002use crate::gen::out::OutFile;
David Tolnay33d30292020-03-18 18:02:02 -07003use crate::gen::{include, Opt};
David Tolnay7db73692019-10-20 14:51:12 -04004use crate::syntax::atom::Atom::{self, *};
David Tolnay75dca2e2020-03-25 20:17:52 -07005use crate::syntax::{Api, ExternFn, Signature, Struct, Type, Types, Var};
David Tolnay7db73692019-10-20 14:51:12 -04006use proc_macro2::Ident;
7
David Tolnay33d30292020-03-18 18:02:02 -07008pub(super) fn gen(
David Tolnay754e21c2020-03-29 20:58:46 -07009 namespace: Namespace,
David Tolnay33d30292020-03-18 18:02:02 -070010 apis: &[Api],
11 types: &Types,
12 opt: Opt,
13 header: bool,
14) -> OutFile {
David Tolnay7db73692019-10-20 14:51:12 -040015 let mut out_file = OutFile::new(namespace.clone(), header);
16 let out = &mut out_file;
17
18 if header {
19 writeln!(out, "#pragma once");
20 }
21
David Tolnay33d30292020-03-18 18:02:02 -070022 out.include.extend(opt.include);
David Tolnay7db73692019-10-20 14:51:12 -040023 for api in apis {
24 if let Api::Include(include) = api {
David Tolnay9c68b1a2020-03-06 11:12:55 -080025 out.include.insert(include.value());
David Tolnay7db73692019-10-20 14:51:12 -040026 }
27 }
28
29 write_includes(out, types);
David Tolnayf51447e2020-03-06 14:14:27 -080030 write_include_cxxbridge(out, apis, types);
David Tolnay7db73692019-10-20 14:51:12 -040031
David Tolnay7db73692019-10-20 14:51:12 -040032 out.next_section();
33 for name in &namespace {
34 writeln!(out, "namespace {} {{", name);
35 }
36
David Tolnay7db73692019-10-20 14:51:12 -040037 out.next_section();
38 for api in apis {
39 match api {
40 Api::Struct(strct) => write_struct_decl(out, &strct.ident),
David Tolnay8861bee2020-01-20 18:39:24 -080041 Api::CxxType(ety) => write_struct_using(out, &ety.ident),
42 Api::RustType(ety) => write_struct_decl(out, &ety.ident),
David Tolnay7db73692019-10-20 14:51:12 -040043 _ => {}
44 }
45 }
46
47 for api in apis {
48 if let Api::Struct(strct) = api {
49 out.next_section();
50 write_struct(out, strct);
51 }
52 }
53
54 if !header {
55 out.begin_block("extern \"C\"");
David Tolnayebef4a22020-03-17 15:33:47 -070056 write_exception_glue(out, apis);
David Tolnay7db73692019-10-20 14:51:12 -040057 for api in apis {
58 let (efn, write): (_, fn(_, _, _)) = match api {
59 Api::CxxFunction(efn) => (efn, write_cxx_function_shim),
60 Api::RustFunction(efn) => (efn, write_rust_function_decl),
61 _ => continue,
62 };
63 out.next_section();
64 write(out, efn, types);
65 }
David Tolnay9ad1fbc2020-03-01 14:01:24 -080066 out.end_block("extern \"C\"");
David Tolnay7db73692019-10-20 14:51:12 -040067 }
68
69 for api in apis {
70 if let Api::RustFunction(efn) = api {
71 out.next_section();
72 write_rust_function_shim(out, efn, types);
73 }
74 }
75
76 out.next_section();
77 for name in namespace.iter().rev() {
78 writeln!(out, "}} // namespace {}", name);
79 }
80
81 if !header {
82 out.next_section();
83 write_generic_instantiations(out, types);
84 }
85
David Tolnay9c68b1a2020-03-06 11:12:55 -080086 out.prepend(out.include.to_string());
87
David Tolnay7db73692019-10-20 14:51:12 -040088 out_file
89}
90
91fn write_includes(out: &mut OutFile, types: &Types) {
David Tolnay7db73692019-10-20 14:51:12 -040092 for ty in types {
93 match ty {
94 Type::Ident(ident) => match Atom::from(ident) {
David Tolnay30430f12020-03-19 20:49:00 -070095 Some(U8) | Some(U16) | Some(U32) | Some(U64) | Some(I8) | Some(I16) | Some(I32)
96 | Some(I64) => out.include.cstdint = true,
97 Some(Usize) => out.include.cstddef = true,
David Tolnay9c68b1a2020-03-06 11:12:55 -080098 Some(CxxString) => out.include.string = true,
David Tolnay30430f12020-03-19 20:49:00 -070099 Some(Bool) | Some(Isize) | Some(F32) | Some(F64) | Some(RustString) | None => {}
David Tolnay7db73692019-10-20 14:51:12 -0400100 },
David Tolnay9c68b1a2020-03-06 11:12:55 -0800101 Type::RustBox(_) => out.include.type_traits = true,
102 Type::UniquePtr(_) => out.include.memory = true,
David Tolnay7db73692019-10-20 14:51:12 -0400103 _ => {}
104 }
105 }
David Tolnay7db73692019-10-20 14:51:12 -0400106}
107
David Tolnayf51447e2020-03-06 14:14:27 -0800108fn write_include_cxxbridge(out: &mut OutFile, apis: &[Api], types: &Types) {
David Tolnayb7a7cb62020-03-17 21:18:40 -0700109 let mut needs_rust_string = false;
110 let mut needs_rust_str = false;
David Tolnay7db73692019-10-20 14:51:12 -0400111 let mut needs_rust_box = false;
David Tolnay75dca2e2020-03-25 20:17:52 -0700112 let mut needs_rust_fn = false;
David Tolnayb8a6fb22020-04-10 11:17:28 -0700113 let mut needs_rust_isize = false;
David Tolnay7db73692019-10-20 14:51:12 -0400114 for ty in types {
David Tolnayb7a7cb62020-03-17 21:18:40 -0700115 match ty {
116 Type::RustBox(_) => {
117 out.include.type_traits = true;
118 needs_rust_box = true;
119 }
120 Type::Str(_) => {
121 out.include.cstdint = true;
122 out.include.string = true;
123 needs_rust_str = true;
124 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700125 Type::Fn(_) => {
126 needs_rust_fn = true;
127 }
David Tolnayb8a6fb22020-04-10 11:17:28 -0700128 ty if ty == Isize => {
David Tolnay59b5ba12020-04-10 11:32:19 -0700129 out.include.base_tsd = true;
David Tolnayb8a6fb22020-04-10 11:17:28 -0700130 needs_rust_isize = true;
131 }
David Tolnayb7a7cb62020-03-17 21:18:40 -0700132 ty if ty == RustString => {
133 out.include.array = true;
134 out.include.cstdint = true;
135 out.include.string = true;
136 needs_rust_string = true;
137 }
138 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400139 }
140 }
141
David Tolnayb7a7cb62020-03-17 21:18:40 -0700142 let mut needs_rust_error = false;
143 let mut needs_unsafe_bitcopy = false;
David Tolnayf51447e2020-03-06 14:14:27 -0800144 let mut needs_manually_drop = false;
David Tolnay09011c32020-03-06 14:40:28 -0800145 let mut needs_maybe_uninit = false;
David Tolnay5d121442020-03-17 22:14:40 -0700146 let mut needs_trycatch = false;
David Tolnay09011c32020-03-06 14:40:28 -0800147 for api in apis {
David Tolnayb7a7cb62020-03-17 21:18:40 -0700148 match api {
149 Api::CxxFunction(efn) if !out.header => {
David Tolnay5d121442020-03-17 22:14:40 -0700150 if efn.throws {
151 needs_trycatch = true;
152 }
David Tolnayb7a7cb62020-03-17 21:18:40 -0700153 for arg in &efn.args {
154 if arg.ty == RustString {
155 needs_unsafe_bitcopy = true;
156 break;
157 }
David Tolnay09011c32020-03-06 14:40:28 -0800158 }
159 }
David Tolnayb7a7cb62020-03-17 21:18:40 -0700160 Api::RustFunction(efn) if !out.header => {
161 if efn.throws {
162 out.include.exception = true;
163 needs_rust_error = true;
164 }
165 for arg in &efn.args {
166 if arg.ty != RustString && types.needs_indirect_abi(&arg.ty) {
167 needs_manually_drop = true;
168 break;
169 }
170 }
171 if let Some(ret) = &efn.ret {
172 if types.needs_indirect_abi(ret) {
173 needs_maybe_uninit = true;
174 }
David Tolnayf51447e2020-03-06 14:14:27 -0800175 }
176 }
David Tolnayb7a7cb62020-03-17 21:18:40 -0700177 _ => {}
David Tolnayf51447e2020-03-06 14:14:27 -0800178 }
179 }
180
David Tolnay750755e2020-03-01 13:04:08 -0800181 out.begin_block("namespace rust");
David Tolnay8c730492020-03-13 01:29:06 -0700182 out.begin_block("inline namespace cxxbridge02");
David Tolnayf51447e2020-03-06 14:14:27 -0800183
David Tolnayb7a7cb62020-03-17 21:18:40 -0700184 if needs_rust_string
185 || needs_rust_str
186 || needs_rust_box
David Tolnay75dca2e2020-03-25 20:17:52 -0700187 || needs_rust_fn
David Tolnayb7a7cb62020-03-17 21:18:40 -0700188 || needs_rust_error
David Tolnayb8a6fb22020-04-10 11:17:28 -0700189 || needs_rust_isize
David Tolnayb7a7cb62020-03-17 21:18:40 -0700190 || needs_unsafe_bitcopy
191 || needs_manually_drop
192 || needs_maybe_uninit
David Tolnay5d121442020-03-17 22:14:40 -0700193 || needs_trycatch
David Tolnayb7a7cb62020-03-17 21:18:40 -0700194 {
David Tolnay736cbca2020-03-11 16:49:18 -0700195 writeln!(out, "// #include \"rust/cxx.h\"");
David Tolnayf51447e2020-03-06 14:14:27 -0800196 }
197
David Tolnayd1402742020-03-25 22:21:42 -0700198 if needs_rust_string {
199 out.next_section();
200 writeln!(out, "struct unsafe_bitcopy_t;");
201 }
202
David Tolnayb7a7cb62020-03-17 21:18:40 -0700203 write_header_section(out, needs_rust_string, "CXXBRIDGE02_RUST_STRING");
204 write_header_section(out, needs_rust_str, "CXXBRIDGE02_RUST_STR");
205 write_header_section(out, needs_rust_box, "CXXBRIDGE02_RUST_BOX");
David Tolnay75dca2e2020-03-25 20:17:52 -0700206 write_header_section(out, needs_rust_fn, "CXXBRIDGE02_RUST_FN");
David Tolnayb7a7cb62020-03-17 21:18:40 -0700207 write_header_section(out, needs_rust_error, "CXXBRIDGE02_RUST_ERROR");
David Tolnayb8a6fb22020-04-10 11:17:28 -0700208 write_header_section(out, needs_rust_isize, "CXXBRIDGE02_RUST_ISIZE");
David Tolnayb7a7cb62020-03-17 21:18:40 -0700209 write_header_section(out, needs_unsafe_bitcopy, "CXXBRIDGE02_RUST_BITCOPY");
David Tolnayf51447e2020-03-06 14:14:27 -0800210
211 if needs_manually_drop {
212 out.next_section();
David Tolnay4791f1c2020-03-17 21:53:16 -0700213 out.include.utility = true;
David Tolnayf51447e2020-03-06 14:14:27 -0800214 writeln!(out, "template <typename T>");
215 writeln!(out, "union ManuallyDrop {{");
216 writeln!(out, " T value;");
217 writeln!(
218 out,
219 " ManuallyDrop(T &&value) : value(::std::move(value)) {{}}",
220 );
221 writeln!(out, " ~ManuallyDrop() {{}}");
222 writeln!(out, "}};");
223 }
224
David Tolnay09011c32020-03-06 14:40:28 -0800225 if needs_maybe_uninit {
226 out.next_section();
227 writeln!(out, "template <typename T>");
228 writeln!(out, "union MaybeUninit {{");
229 writeln!(out, " T value;");
230 writeln!(out, " MaybeUninit() {{}}");
231 writeln!(out, " ~MaybeUninit() {{}}");
232 writeln!(out, "}};");
233 }
234
David Tolnay3e3e0af2020-03-17 22:42:49 -0700235 out.end_block("namespace cxxbridge02");
236
David Tolnay5d121442020-03-17 22:14:40 -0700237 if needs_trycatch {
David Tolnay3e3e0af2020-03-17 22:42:49 -0700238 out.begin_block("namespace behavior");
David Tolnay5d121442020-03-17 22:14:40 -0700239 out.include.exception = true;
David Tolnay04722332020-03-18 11:31:54 -0700240 out.include.type_traits = true;
241 out.include.utility = true;
242 writeln!(out, "class missing {{}};");
243 writeln!(out, "missing trycatch(...);");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700244 writeln!(out);
David Tolnay04722332020-03-18 11:31:54 -0700245 writeln!(out, "template <typename Try, typename Fail>");
246 writeln!(out, "static typename std::enable_if<");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700247 writeln!(
248 out,
David Tolnay04722332020-03-18 11:31:54 -0700249 " std::is_same<decltype(trycatch(std::declval<Try>(), std::declval<Fail>())),",
David Tolnay3e3e0af2020-03-17 22:42:49 -0700250 );
David Tolnay04722332020-03-18 11:31:54 -0700251 writeln!(out, " missing>::value>::type");
252 writeln!(out, "trycatch(Try &&func, Fail &&fail) noexcept try {{");
David Tolnay5d121442020-03-17 22:14:40 -0700253 writeln!(out, " func();");
254 writeln!(out, "}} catch (const ::std::exception &e) {{");
255 writeln!(out, " fail(e.what());");
256 writeln!(out, "}}");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700257 out.end_block("namespace behavior");
David Tolnay5d121442020-03-17 22:14:40 -0700258 }
259
David Tolnay9ad1fbc2020-03-01 14:01:24 -0800260 out.end_block("namespace rust");
David Tolnay7db73692019-10-20 14:51:12 -0400261}
262
David Tolnayb7a7cb62020-03-17 21:18:40 -0700263fn write_header_section(out: &mut OutFile, needed: bool, section: &str) {
264 if needed {
265 out.next_section();
266 for line in include::get(section).lines() {
267 if !line.trim_start().starts_with("//") {
268 writeln!(out, "{}", line);
269 }
270 }
271 }
272}
273
David Tolnay7db73692019-10-20 14:51:12 -0400274fn write_struct(out: &mut OutFile, strct: &Struct) {
275 for line in strct.doc.to_string().lines() {
276 writeln!(out, "//{}", line);
277 }
278 writeln!(out, "struct {} final {{", strct.ident);
279 for field in &strct.fields {
280 write!(out, " ");
281 write_type_space(out, &field.ty);
282 writeln!(out, "{};", field.ident);
283 }
284 writeln!(out, "}};");
285}
286
287fn write_struct_decl(out: &mut OutFile, ident: &Ident) {
288 writeln!(out, "struct {};", ident);
289}
290
David Tolnay8861bee2020-01-20 18:39:24 -0800291fn write_struct_using(out: &mut OutFile, ident: &Ident) {
292 writeln!(out, "using {} = {};", ident, ident);
293}
294
David Tolnayebef4a22020-03-17 15:33:47 -0700295fn write_exception_glue(out: &mut OutFile, apis: &[Api]) {
296 let mut has_cxx_throws = false;
297 for api in apis {
298 if let Api::CxxFunction(efn) = api {
299 if efn.throws {
300 has_cxx_throws = true;
301 break;
302 }
303 }
304 }
305
306 if has_cxx_throws {
307 out.next_section();
David Tolnaye68634c2020-03-18 12:03:40 -0700308 writeln!(
David Tolnayebef4a22020-03-17 15:33:47 -0700309 out,
310 "const char *cxxbridge02$exception(const char *, size_t);",
311 );
312 }
313}
314
David Tolnay7db73692019-10-20 14:51:12 -0400315fn write_cxx_function_shim(out: &mut OutFile, efn: &ExternFn, types: &Types) {
David Tolnayebef4a22020-03-17 15:33:47 -0700316 if efn.throws {
317 write!(out, "::rust::Str::Repr ");
318 } else {
David Tolnay99642622020-03-25 13:07:35 -0700319 write_extern_return_type_space(out, &efn.ret, types);
David Tolnayebef4a22020-03-17 15:33:47 -0700320 }
David Tolnayd815de02020-03-29 21:29:17 -0700321 write!(out, "{}cxxbridge02${}(", out.namespace, efn.ident);
David Tolnay7db73692019-10-20 14:51:12 -0400322 for (i, arg) in efn.args.iter().enumerate() {
323 if i > 0 {
324 write!(out, ", ");
325 }
David Tolnaya46a2372020-03-06 10:03:48 -0800326 if arg.ty == RustString {
327 write!(out, "const ");
328 }
David Tolnay7db73692019-10-20 14:51:12 -0400329 write_extern_arg(out, arg, types);
330 }
David Tolnay277e3cc2020-03-17 00:11:01 -0700331 let indirect_return = indirect_return(efn, types);
David Tolnay7db73692019-10-20 14:51:12 -0400332 if indirect_return {
333 if !efn.args.is_empty() {
334 write!(out, ", ");
335 }
David Tolnay99642622020-03-25 13:07:35 -0700336 write_indirect_return_type_space(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400337 write!(out, "*return$");
338 }
339 writeln!(out, ") noexcept {{");
340 write!(out, " ");
341 write_return_type(out, &efn.ret);
342 write!(out, "(*{}$)(", efn.ident);
343 for (i, arg) in efn.args.iter().enumerate() {
344 if i > 0 {
345 write!(out, ", ");
346 }
347 write_type(out, &arg.ty);
348 }
349 writeln!(out, ") = {};", efn.ident);
350 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700351 if efn.throws {
352 writeln!(out, "::rust::Str::Repr throw$;");
David Tolnay3e3e0af2020-03-17 22:42:49 -0700353 writeln!(out, " ::rust::behavior::trycatch(");
David Tolnay5d121442020-03-17 22:14:40 -0700354 writeln!(out, " [&] {{");
355 write!(out, " ");
David Tolnayebef4a22020-03-17 15:33:47 -0700356 }
David Tolnay7db73692019-10-20 14:51:12 -0400357 if indirect_return {
358 write!(out, "new (return$) ");
David Tolnay99642622020-03-25 13:07:35 -0700359 write_indirect_return_type(out, efn.ret.as_ref().unwrap());
David Tolnay7db73692019-10-20 14:51:12 -0400360 write!(out, "(");
David Tolnay99642622020-03-25 13:07:35 -0700361 } else if efn.ret.is_some() {
David Tolnay7db73692019-10-20 14:51:12 -0400362 write!(out, "return ");
David Tolnay99642622020-03-25 13:07:35 -0700363 }
364 match &efn.ret {
365 Some(Type::Ref(_)) => write!(out, "&"),
366 Some(Type::Str(_)) if !indirect_return => write!(out, "::rust::Str::Repr("),
367 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400368 }
369 write!(out, "{}$(", efn.ident);
370 for (i, arg) in efn.args.iter().enumerate() {
371 if i > 0 {
372 write!(out, ", ");
373 }
374 if let Type::RustBox(_) = &arg.ty {
375 write_type(out, &arg.ty);
376 write!(out, "::from_raw({})", arg.ident);
377 } else if let Type::UniquePtr(_) = &arg.ty {
378 write_type(out, &arg.ty);
379 write!(out, "({})", arg.ident);
David Tolnaya46a2372020-03-06 10:03:48 -0800380 } else if arg.ty == RustString {
David Tolnaycc3767f2020-03-06 10:41:51 -0800381 write!(
382 out,
383 "::rust::String(::rust::unsafe_bitcopy, *{})",
384 arg.ident,
385 );
David Tolnay7db73692019-10-20 14:51:12 -0400386 } else if types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700387 out.include.utility = true;
David Tolnay7e219b82020-03-01 13:14:51 -0800388 write!(out, "::std::move(*{})", arg.ident);
David Tolnay7db73692019-10-20 14:51:12 -0400389 } else {
390 write!(out, "{}", arg.ident);
391 }
392 }
393 write!(out, ")");
394 match &efn.ret {
395 Some(Type::RustBox(_)) => write!(out, ".into_raw()"),
396 Some(Type::UniquePtr(_)) => write!(out, ".release()"),
David Tolnay99642622020-03-25 13:07:35 -0700397 Some(Type::Str(_)) if !indirect_return => write!(out, ")"),
David Tolnay7db73692019-10-20 14:51:12 -0400398 _ => {}
399 }
400 if indirect_return {
401 write!(out, ")");
402 }
403 writeln!(out, ";");
David Tolnayebef4a22020-03-17 15:33:47 -0700404 if efn.throws {
405 out.include.cstring = true;
David Tolnay5d121442020-03-17 22:14:40 -0700406 writeln!(out, " throw$.ptr = nullptr;");
407 writeln!(out, " }},");
David Tolnay82c16172020-03-17 22:54:12 -0700408 writeln!(out, " [&](const char *catch$) noexcept {{");
David Tolnay5d121442020-03-17 22:14:40 -0700409 writeln!(out, " throw$.len = ::std::strlen(catch$);");
David Tolnayebef4a22020-03-17 15:33:47 -0700410 writeln!(
411 out,
David Tolnay5d121442020-03-17 22:14:40 -0700412 " throw$.ptr = cxxbridge02$exception(catch$, throw$.len);",
David Tolnayebef4a22020-03-17 15:33:47 -0700413 );
David Tolnay5d121442020-03-17 22:14:40 -0700414 writeln!(out, " }});");
David Tolnayebef4a22020-03-17 15:33:47 -0700415 writeln!(out, " return throw$;");
416 }
David Tolnay7db73692019-10-20 14:51:12 -0400417 writeln!(out, "}}");
David Tolnay75dca2e2020-03-25 20:17:52 -0700418 for arg in &efn.args {
419 if let Type::Fn(f) = &arg.ty {
420 let var = &arg.ident;
421 write_function_pointer_trampoline(out, efn, var, f, types);
422 }
423 }
424}
425
426fn write_function_pointer_trampoline(
427 out: &mut OutFile,
428 efn: &ExternFn,
429 var: &Ident,
430 f: &Signature,
431 types: &Types,
432) {
433 out.next_section();
434 let r_trampoline = format!("{}cxxbridge02${}${}$1", out.namespace, efn.ident, var);
435 let indirect_call = true;
436 write_rust_function_decl_impl(out, &r_trampoline, f, types, indirect_call);
437
438 out.next_section();
439 let c_trampoline = format!("{}cxxbridge02${}${}$0", out.namespace, efn.ident, var);
440 write_rust_function_shim_impl(out, &c_trampoline, f, types, &r_trampoline, indirect_call);
David Tolnay7db73692019-10-20 14:51:12 -0400441}
442
443fn write_rust_function_decl(out: &mut OutFile, efn: &ExternFn, types: &Types) {
David Tolnay75dca2e2020-03-25 20:17:52 -0700444 let link_name = format!("{}cxxbridge02${}", out.namespace, efn.ident);
445 let indirect_call = false;
446 write_rust_function_decl_impl(out, &link_name, efn, types, indirect_call);
447}
448
449fn write_rust_function_decl_impl(
450 out: &mut OutFile,
451 link_name: &str,
452 sig: &Signature,
453 types: &Types,
454 indirect_call: bool,
455) {
456 if sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700457 write!(out, "::rust::Str::Repr ");
458 } else {
David Tolnay75dca2e2020-03-25 20:17:52 -0700459 write_extern_return_type_space(out, &sig.ret, types);
David Tolnay1e548172020-03-16 13:37:09 -0700460 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700461 write!(out, "{}(", link_name);
462 let mut needs_comma = false;
463 for arg in &sig.args {
464 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400465 write!(out, ", ");
466 }
467 write_extern_arg(out, arg, types);
David Tolnay75dca2e2020-03-25 20:17:52 -0700468 needs_comma = true;
David Tolnay7db73692019-10-20 14:51:12 -0400469 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700470 if indirect_return(sig, types) {
471 if needs_comma {
David Tolnay7db73692019-10-20 14:51:12 -0400472 write!(out, ", ");
473 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700474 write_return_type(out, &sig.ret);
David Tolnay7db73692019-10-20 14:51:12 -0400475 write!(out, "*return$");
David Tolnay75dca2e2020-03-25 20:17:52 -0700476 needs_comma = true;
477 }
478 if indirect_call {
479 if needs_comma {
480 write!(out, ", ");
481 }
482 write!(out, "void *");
David Tolnay7db73692019-10-20 14:51:12 -0400483 }
484 writeln!(out, ") noexcept;");
485}
486
487fn write_rust_function_shim(out: &mut OutFile, efn: &ExternFn, types: &Types) {
David Tolnay7db73692019-10-20 14:51:12 -0400488 for line in efn.doc.to_string().lines() {
489 writeln!(out, "//{}", line);
490 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700491 let local_name = efn.ident.to_string();
492 let invoke = format!("{}cxxbridge02${}", out.namespace, efn.ident);
493 let indirect_call = false;
494 write_rust_function_shim_impl(out, &local_name, efn, types, &invoke, indirect_call);
495}
496
497fn write_rust_function_shim_impl(
498 out: &mut OutFile,
499 local_name: &str,
500 sig: &Signature,
501 types: &Types,
502 invoke: &str,
503 indirect_call: bool,
504) {
505 write_return_type(out, &sig.ret);
506 write!(out, "{}(", local_name);
507 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400508 if i > 0 {
509 write!(out, ", ");
510 }
511 write_type_space(out, &arg.ty);
512 write!(out, "{}", arg.ident);
513 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700514 if indirect_call {
515 if !sig.args.is_empty() {
516 write!(out, ", ");
517 }
518 write!(out, "void *extern$");
519 }
David Tolnay1e548172020-03-16 13:37:09 -0700520 write!(out, ")");
David Tolnay75dca2e2020-03-25 20:17:52 -0700521 if !sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700522 write!(out, " noexcept");
523 }
David Tolnay7db73692019-10-20 14:51:12 -0400524 if out.header {
525 writeln!(out, ";");
526 } else {
527 writeln!(out, " {{");
David Tolnay75dca2e2020-03-25 20:17:52 -0700528 for arg in &sig.args {
David Tolnayf51447e2020-03-06 14:14:27 -0800529 if arg.ty != RustString && types.needs_indirect_abi(&arg.ty) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700530 out.include.utility = true;
David Tolnayf51447e2020-03-06 14:14:27 -0800531 write!(out, " ::rust::ManuallyDrop<");
532 write_type(out, &arg.ty);
533 writeln!(out, "> {}$(::std::move({0}));", arg.ident);
534 }
535 }
David Tolnay7db73692019-10-20 14:51:12 -0400536 write!(out, " ");
David Tolnay75dca2e2020-03-25 20:17:52 -0700537 let indirect_return = indirect_return(sig, types);
David Tolnay7db73692019-10-20 14:51:12 -0400538 if indirect_return {
David Tolnay09011c32020-03-06 14:40:28 -0800539 write!(out, "::rust::MaybeUninit<");
David Tolnay75dca2e2020-03-25 20:17:52 -0700540 write_type(out, sig.ret.as_ref().unwrap());
David Tolnay09011c32020-03-06 14:40:28 -0800541 writeln!(out, "> return$;");
David Tolnay7db73692019-10-20 14:51:12 -0400542 write!(out, " ");
David Tolnay75dca2e2020-03-25 20:17:52 -0700543 } else if let Some(ret) = &sig.ret {
David Tolnay7db73692019-10-20 14:51:12 -0400544 write!(out, "return ");
David Tolnay5cd8d612020-03-06 15:56:30 -0800545 match ret {
546 Type::RustBox(_) => {
547 write_type(out, ret);
548 write!(out, "::from_raw(");
549 }
David Tolnay4b3a66e2020-03-06 16:14:00 -0800550 Type::UniquePtr(_) => {
551 write_type(out, ret);
552 write!(out, "(");
553 }
David Tolnay5cd8d612020-03-06 15:56:30 -0800554 Type::Ref(_) => write!(out, "*"),
555 _ => {}
David Tolnay4a441222020-01-25 16:24:27 -0800556 }
David Tolnay7db73692019-10-20 14:51:12 -0400557 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700558 if sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700559 write!(out, "::rust::Str::Repr error$ = ");
560 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700561 write!(out, "{}(", invoke);
562 for (i, arg) in sig.args.iter().enumerate() {
David Tolnay7db73692019-10-20 14:51:12 -0400563 if i > 0 {
564 write!(out, ", ");
565 }
David Tolnaybaae4432020-03-01 20:20:10 -0800566 match &arg.ty {
567 Type::Str(_) => write!(out, "::rust::Str::Repr("),
568 ty if types.needs_indirect_abi(ty) => write!(out, "&"),
569 _ => {}
David Tolnay7db73692019-10-20 14:51:12 -0400570 }
571 write!(out, "{}", arg.ident);
David Tolnayf51447e2020-03-06 14:14:27 -0800572 match &arg.ty {
David Tolnay17955e22020-01-20 17:58:24 -0800573 Type::RustBox(_) => write!(out, ".into_raw()"),
574 Type::UniquePtr(_) => write!(out, ".release()"),
David Tolnaybaae4432020-03-01 20:20:10 -0800575 Type::Str(_) => write!(out, ")"),
David Tolnayf51447e2020-03-06 14:14:27 -0800576 ty if ty != RustString && types.needs_indirect_abi(ty) => write!(out, "$.value"),
David Tolnay17955e22020-01-20 17:58:24 -0800577 _ => {}
578 }
David Tolnay7db73692019-10-20 14:51:12 -0400579 }
580 if indirect_return {
David Tolnay75dca2e2020-03-25 20:17:52 -0700581 if !sig.args.is_empty() {
David Tolnay7db73692019-10-20 14:51:12 -0400582 write!(out, ", ");
583 }
David Tolnay09011c32020-03-06 14:40:28 -0800584 write!(out, "&return$.value");
David Tolnay7db73692019-10-20 14:51:12 -0400585 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700586 if indirect_call {
587 if !sig.args.is_empty() || indirect_return {
588 write!(out, ", ");
589 }
590 write!(out, "extern$");
591 }
David Tolnay5cd8d612020-03-06 15:56:30 -0800592 write!(out, ")");
David Tolnay75dca2e2020-03-25 20:17:52 -0700593 if let Some(ret) = &sig.ret {
David Tolnay4b3a66e2020-03-06 16:14:00 -0800594 if let Type::RustBox(_) | Type::UniquePtr(_) = ret {
David Tolnay5cd8d612020-03-06 15:56:30 -0800595 write!(out, ")");
596 }
597 }
598 writeln!(out, ";");
David Tolnay75dca2e2020-03-25 20:17:52 -0700599 if sig.throws {
David Tolnay1e548172020-03-16 13:37:09 -0700600 writeln!(out, " if (error$.ptr) {{");
601 writeln!(out, " throw ::rust::Error(error$);");
602 writeln!(out, " }}");
603 }
David Tolnay7db73692019-10-20 14:51:12 -0400604 if indirect_return {
David Tolnay4791f1c2020-03-17 21:53:16 -0700605 out.include.utility = true;
David Tolnay09011c32020-03-06 14:40:28 -0800606 writeln!(out, " return ::std::move(return$.value);");
David Tolnay7db73692019-10-20 14:51:12 -0400607 }
608 writeln!(out, "}}");
609 }
610}
611
612fn write_return_type(out: &mut OutFile, ty: &Option<Type>) {
613 match ty {
614 None => write!(out, "void "),
615 Some(ty) => write_type_space(out, ty),
616 }
617}
618
David Tolnay75dca2e2020-03-25 20:17:52 -0700619fn indirect_return(sig: &Signature, types: &Types) -> bool {
620 sig.ret
David Tolnay277e3cc2020-03-17 00:11:01 -0700621 .as_ref()
David Tolnay75dca2e2020-03-25 20:17:52 -0700622 .map_or(false, |ret| sig.throws || types.needs_indirect_abi(ret))
David Tolnay277e3cc2020-03-17 00:11:01 -0700623}
624
David Tolnay99642622020-03-25 13:07:35 -0700625fn write_indirect_return_type(out: &mut OutFile, ty: &Type) {
626 match ty {
627 Type::RustBox(ty) | Type::UniquePtr(ty) => {
628 write_type_space(out, &ty.inner);
629 write!(out, "*");
630 }
631 Type::Ref(ty) => {
632 if ty.mutability.is_none() {
633 write!(out, "const ");
634 }
635 write_type(out, &ty.inner);
636 write!(out, " *");
637 }
638 Type::Str(_) => write!(out, "::rust::Str::Repr"),
639 _ => write_type(out, ty),
640 }
641}
642
643fn write_indirect_return_type_space(out: &mut OutFile, ty: &Type) {
644 write_indirect_return_type(out, ty);
645 match ty {
646 Type::RustBox(_) | Type::UniquePtr(_) | Type::Ref(_) => {}
647 Type::Str(_) => write!(out, " "),
648 _ => write_space_after_type(out, ty),
649 }
650}
651
652fn write_extern_return_type_space(out: &mut OutFile, ty: &Option<Type>, types: &Types) {
David Tolnay7db73692019-10-20 14:51:12 -0400653 match ty {
654 Some(Type::RustBox(ty)) | Some(Type::UniquePtr(ty)) => {
655 write_type_space(out, &ty.inner);
656 write!(out, "*");
657 }
David Tolnay4a441222020-01-25 16:24:27 -0800658 Some(Type::Ref(ty)) => {
659 if ty.mutability.is_none() {
660 write!(out, "const ");
661 }
662 write_type(out, &ty.inner);
663 write!(out, " *");
664 }
David Tolnay750755e2020-03-01 13:04:08 -0800665 Some(Type::Str(_)) => write!(out, "::rust::Str::Repr "),
David Tolnay7db73692019-10-20 14:51:12 -0400666 Some(ty) if types.needs_indirect_abi(ty) => write!(out, "void "),
667 _ => write_return_type(out, ty),
668 }
669}
670
671fn write_extern_arg(out: &mut OutFile, arg: &Var, types: &Types) {
672 match &arg.ty {
673 Type::RustBox(ty) | Type::UniquePtr(ty) => {
674 write_type_space(out, &ty.inner);
675 write!(out, "*");
676 }
David Tolnay750755e2020-03-01 13:04:08 -0800677 Type::Str(_) => write!(out, "::rust::Str::Repr "),
David Tolnay7db73692019-10-20 14:51:12 -0400678 _ => write_type_space(out, &arg.ty),
679 }
680 if types.needs_indirect_abi(&arg.ty) {
681 write!(out, "*");
682 }
683 write!(out, "{}", arg.ident);
684}
685
686fn write_type(out: &mut OutFile, ty: &Type) {
687 match ty {
688 Type::Ident(ident) => match Atom::from(ident) {
689 Some(Bool) => write!(out, "bool"),
690 Some(U8) => write!(out, "uint8_t"),
691 Some(U16) => write!(out, "uint16_t"),
692 Some(U32) => write!(out, "uint32_t"),
693 Some(U64) => write!(out, "uint64_t"),
694 Some(Usize) => write!(out, "size_t"),
695 Some(I8) => write!(out, "int8_t"),
696 Some(I16) => write!(out, "int16_t"),
697 Some(I32) => write!(out, "int32_t"),
698 Some(I64) => write!(out, "int64_t"),
David Tolnayb8a6fb22020-04-10 11:17:28 -0700699 Some(Isize) => write!(out, "::rust::isize"),
David Tolnay3383ae72020-03-13 01:12:26 -0700700 Some(F32) => write!(out, "float"),
701 Some(F64) => write!(out, "double"),
David Tolnay7e219b82020-03-01 13:14:51 -0800702 Some(CxxString) => write!(out, "::std::string"),
David Tolnay750755e2020-03-01 13:04:08 -0800703 Some(RustString) => write!(out, "::rust::String"),
David Tolnay7db73692019-10-20 14:51:12 -0400704 None => write!(out, "{}", ident),
705 },
706 Type::RustBox(ty) => {
David Tolnay750755e2020-03-01 13:04:08 -0800707 write!(out, "::rust::Box<");
David Tolnay7db73692019-10-20 14:51:12 -0400708 write_type(out, &ty.inner);
709 write!(out, ">");
710 }
711 Type::UniquePtr(ptr) => {
David Tolnay7e219b82020-03-01 13:14:51 -0800712 write!(out, "::std::unique_ptr<");
David Tolnay7db73692019-10-20 14:51:12 -0400713 write_type(out, &ptr.inner);
714 write!(out, ">");
715 }
716 Type::Ref(r) => {
717 if r.mutability.is_none() {
718 write!(out, "const ");
719 }
720 write_type(out, &r.inner);
721 write!(out, " &");
722 }
723 Type::Str(_) => {
David Tolnay750755e2020-03-01 13:04:08 -0800724 write!(out, "::rust::Str");
David Tolnay7db73692019-10-20 14:51:12 -0400725 }
David Tolnay75dca2e2020-03-25 20:17:52 -0700726 Type::Fn(f) => {
727 write!(out, "::rust::{}<", if f.throws { "TryFn" } else { "Fn" });
728 match &f.ret {
729 Some(ret) => write_type(out, ret),
730 None => write!(out, "void"),
731 }
732 write!(out, "(");
733 for (i, arg) in f.args.iter().enumerate() {
734 if i > 0 {
735 write!(out, ", ");
736 }
737 write_type(out, &arg.ty);
738 }
739 write!(out, ")>");
740 }
David Tolnay2fb14e92020-03-15 23:11:38 -0700741 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -0400742 }
743}
744
745fn write_type_space(out: &mut OutFile, ty: &Type) {
746 write_type(out, ty);
David Tolnay99642622020-03-25 13:07:35 -0700747 write_space_after_type(out, ty);
748}
749
750fn write_space_after_type(out: &mut OutFile, ty: &Type) {
David Tolnay7db73692019-10-20 14:51:12 -0400751 match ty {
David Tolnay75dca2e2020-03-25 20:17:52 -0700752 Type::Ident(_) | Type::RustBox(_) | Type::UniquePtr(_) | Type::Str(_) | Type::Fn(_) => {
753 write!(out, " ")
754 }
David Tolnay7db73692019-10-20 14:51:12 -0400755 Type::Ref(_) => {}
David Tolnay2fb14e92020-03-15 23:11:38 -0700756 Type::Void(_) => unreachable!(),
David Tolnay7db73692019-10-20 14:51:12 -0400757 }
758}
759
760fn write_generic_instantiations(out: &mut OutFile, types: &Types) {
761 fn allow_unique_ptr(ident: &Ident) -> bool {
762 Atom::from(ident).is_none()
763 }
764
765 out.begin_block("extern \"C\"");
766 for ty in types {
767 if let Type::RustBox(ty) = ty {
768 if let Type::Ident(inner) = &ty.inner {
769 out.next_section();
770 write_rust_box_extern(out, inner);
771 }
772 } else if let Type::UniquePtr(ptr) = ty {
773 if let Type::Ident(inner) = &ptr.inner {
774 if allow_unique_ptr(inner) {
775 out.next_section();
David Tolnay53838912020-04-09 20:56:44 -0700776 write_unique_ptr(out, inner, types);
David Tolnay7db73692019-10-20 14:51:12 -0400777 }
778 }
779 }
780 }
David Tolnay9ad1fbc2020-03-01 14:01:24 -0800781 out.end_block("extern \"C\"");
David Tolnay7db73692019-10-20 14:51:12 -0400782
David Tolnay750755e2020-03-01 13:04:08 -0800783 out.begin_block("namespace rust");
David Tolnay8c730492020-03-13 01:29:06 -0700784 out.begin_block("inline namespace cxxbridge02");
David Tolnay7db73692019-10-20 14:51:12 -0400785 for ty in types {
786 if let Type::RustBox(ty) = ty {
787 if let Type::Ident(inner) = &ty.inner {
788 write_rust_box_impl(out, inner);
789 }
790 }
791 }
David Tolnay8c730492020-03-13 01:29:06 -0700792 out.end_block("namespace cxxbridge02");
David Tolnay9ad1fbc2020-03-01 14:01:24 -0800793 out.end_block("namespace rust");
David Tolnay7db73692019-10-20 14:51:12 -0400794}
795
796fn write_rust_box_extern(out: &mut OutFile, ident: &Ident) {
797 let mut inner = String::new();
798 for name in &out.namespace {
799 inner += name;
800 inner += "::";
801 }
802 inner += &ident.to_string();
803 let instance = inner.replace("::", "$");
804
David Tolnay8c730492020-03-13 01:29:06 -0700805 writeln!(out, "#ifndef CXXBRIDGE02_RUST_BOX_{}", instance);
806 writeln!(out, "#define CXXBRIDGE02_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400807 writeln!(
808 out,
David Tolnay8c730492020-03-13 01:29:06 -0700809 "void cxxbridge02$box${}$uninit(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -0400810 instance, inner,
811 );
812 writeln!(
813 out,
David Tolnay8c730492020-03-13 01:29:06 -0700814 "void cxxbridge02$box${}$drop(::rust::Box<{}> *ptr) noexcept;",
David Tolnay7db73692019-10-20 14:51:12 -0400815 instance, inner,
816 );
David Tolnay8c730492020-03-13 01:29:06 -0700817 writeln!(out, "#endif // CXXBRIDGE02_RUST_BOX_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400818}
819
820fn write_rust_box_impl(out: &mut OutFile, ident: &Ident) {
821 let mut inner = String::new();
822 for name in &out.namespace {
823 inner += name;
824 inner += "::";
825 }
826 inner += &ident.to_string();
827 let instance = inner.replace("::", "$");
828
829 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -0800830 writeln!(out, "void Box<{}>::uninit() noexcept {{", inner);
David Tolnay737e02e2020-04-04 21:52:46 -0700831 writeln!(out, " cxxbridge02$box${}$uninit(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400832 writeln!(out, "}}");
833
834 writeln!(out, "template <>");
David Tolnay324437a2020-03-01 13:02:24 -0800835 writeln!(out, "void Box<{}>::drop() noexcept {{", inner);
David Tolnay737e02e2020-04-04 21:52:46 -0700836 writeln!(out, " cxxbridge02$box${}$drop(this);", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400837 writeln!(out, "}}");
David Tolnay7db73692019-10-20 14:51:12 -0400838}
839
David Tolnay53838912020-04-09 20:56:44 -0700840fn write_unique_ptr(out: &mut OutFile, ident: &Ident, types: &Types) {
David Tolnay4791f1c2020-03-17 21:53:16 -0700841 out.include.utility = true;
842
David Tolnay7db73692019-10-20 14:51:12 -0400843 let mut inner = String::new();
844 for name in &out.namespace {
845 inner += name;
846 inner += "::";
847 }
848 inner += &ident.to_string();
849 let instance = inner.replace("::", "$");
850
David Tolnay8c730492020-03-13 01:29:06 -0700851 writeln!(out, "#ifndef CXXBRIDGE02_UNIQUE_PTR_{}", instance);
852 writeln!(out, "#define CXXBRIDGE02_UNIQUE_PTR_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400853 writeln!(
854 out,
David Tolnay7e219b82020-03-01 13:14:51 -0800855 "static_assert(sizeof(::std::unique_ptr<{}>) == sizeof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -0400856 inner,
857 );
858 writeln!(
859 out,
David Tolnay7e219b82020-03-01 13:14:51 -0800860 "static_assert(alignof(::std::unique_ptr<{}>) == alignof(void *), \"\");",
David Tolnay7db73692019-10-20 14:51:12 -0400861 inner,
862 );
863 writeln!(
864 out,
David Tolnay8c730492020-03-13 01:29:06 -0700865 "void cxxbridge02$unique_ptr${}$null(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -0400866 instance, inner,
867 );
David Tolnay7e219b82020-03-01 13:14:51 -0800868 writeln!(out, " new (ptr) ::std::unique_ptr<{}>();", inner);
David Tolnay7db73692019-10-20 14:51:12 -0400869 writeln!(out, "}}");
David Tolnay53838912020-04-09 20:56:44 -0700870 if types.structs.contains_key(ident) {
871 writeln!(
872 out,
873 "void cxxbridge02$unique_ptr${}$new(::std::unique_ptr<{}> *ptr, {} *value) noexcept {{",
874 instance, inner, inner,
875 );
876 writeln!(
877 out,
878 " new (ptr) ::std::unique_ptr<{}>(new {}(::std::move(*value)));",
879 inner, inner,
880 );
881 writeln!(out, "}}");
882 }
David Tolnay7db73692019-10-20 14:51:12 -0400883 writeln!(
884 out,
David Tolnay8c730492020-03-13 01:29:06 -0700885 "void cxxbridge02$unique_ptr${}$raw(::std::unique_ptr<{}> *ptr, {} *raw) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -0400886 instance, inner, inner,
887 );
David Tolnay7e219b82020-03-01 13:14:51 -0800888 writeln!(out, " new (ptr) ::std::unique_ptr<{}>(raw);", inner);
David Tolnay7db73692019-10-20 14:51:12 -0400889 writeln!(out, "}}");
890 writeln!(
891 out,
David Tolnay8c730492020-03-13 01:29:06 -0700892 "const {} *cxxbridge02$unique_ptr${}$get(const ::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -0400893 inner, instance, inner,
894 );
895 writeln!(out, " return ptr.get();");
896 writeln!(out, "}}");
897 writeln!(
898 out,
David Tolnay8c730492020-03-13 01:29:06 -0700899 "{} *cxxbridge02$unique_ptr${}$release(::std::unique_ptr<{}>& ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -0400900 inner, instance, inner,
901 );
902 writeln!(out, " return ptr.release();");
903 writeln!(out, "}}");
904 writeln!(
905 out,
David Tolnay8c730492020-03-13 01:29:06 -0700906 "void cxxbridge02$unique_ptr${}$drop(::std::unique_ptr<{}> *ptr) noexcept {{",
David Tolnay7db73692019-10-20 14:51:12 -0400907 instance, inner,
908 );
909 writeln!(out, " ptr->~unique_ptr();");
910 writeln!(out, "}}");
David Tolnay8c730492020-03-13 01:29:06 -0700911 writeln!(out, "#endif // CXXBRIDGE02_UNIQUE_PTR_{}", instance);
David Tolnay7db73692019-10-20 14:51:12 -0400912}