Merge pull request #17 from TheBlueMatt/2021-04-upstream-confirm
[ldk-c-bindings] / c-bindings-gen / src / blocks.rs
index dd57a82b5f2d557a188c8a15068d112a5f809afa..2c51572a9378150fdc861c12c5c9e3edec11c7aa 100644 (file)
@@ -1,9 +1,19 @@
+// This file is Copyright its original authors, visible in version control
+// history.
+//
+// This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE>
+// or the MIT license <LICENSE-MIT>, at your option.
+// You may not use this file except in accordance with one or both of these
+// licenses.
+
 //! Printing logic for basic blocks of Rust-mapped code - parts of functions and declarations but
 //! not the full mapping logic.
 
 use std::fs::File;
 use std::io::Write;
-use proc_macro2::{TokenTree, Span};
+
+use proc_macro2::TokenTree;
+use quote::format_ident;
 
 use crate::types::*;
 
@@ -35,14 +45,19 @@ pub fn write_cpp_wrapper(cpp_header_file: &mut File, ty: &str, has_destructor: b
 /// Writes out a C-callable concrete Result<A, B> struct and utility methods
 pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str, ok_type: &str, err_type: &str, clonable: bool) {
        writeln!(w, "#[repr(C)]").unwrap();
+       writeln!(w, "/// The contents of {}", mangled_container).unwrap();
        writeln!(w, "pub union {}Ptr {{", mangled_container).unwrap();
        if ok_type != "()" {
+               writeln!(w, "\t/// A pointer to the contents in the success state.").unwrap();
+               writeln!(w, "\t/// Reading from this pointer when `result_ok` is not set is undefined.").unwrap();
                writeln!(w, "\tpub result: *mut {},", ok_type).unwrap();
        } else {
                writeln!(w, "\t/// Note that this value is always NULL, as there are no contents in the OK variant").unwrap();
                writeln!(w, "\tpub result: *mut std::ffi::c_void,").unwrap();
        }
        if err_type != "()" {
+               writeln!(w, "\t/// A pointer to the contents in the error state.").unwrap();
+               writeln!(w, "\t/// Reading from this pointer when `result_ok` is set is undefined.").unwrap();
                writeln!(w, "\tpub err: *mut {},", err_type).unwrap();
        } else {
                writeln!(w, "\t/// Note that this value is always NULL, as there are no contents in the Err variant").unwrap();
@@ -50,15 +65,23 @@ pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
        }
        writeln!(w, "}}").unwrap();
        writeln!(w, "#[repr(C)]").unwrap();
+       writeln!(w, "/// A {} represents the result of a fallible operation,", mangled_container).unwrap();
+       writeln!(w, "/// containing a {} on success and a {} on failure.", ok_type, err_type).unwrap();
+       writeln!(w, "/// `result_ok` indicates the overall state, and the contents are provided via `contents`.").unwrap();
        writeln!(w, "pub struct {} {{", mangled_container).unwrap();
+       writeln!(w, "\t/// The contents of this {}, accessible via either", mangled_container).unwrap();
+       writeln!(w, "\t/// `err` or `result` depending on the state of `result_ok`.").unwrap();
        writeln!(w, "\tpub contents: {}Ptr,", mangled_container).unwrap();
+       writeln!(w, "\t/// Whether this {} represents a success state.", mangled_container).unwrap();
        writeln!(w, "\tpub result_ok: bool,").unwrap();
        writeln!(w, "}}").unwrap();
 
        writeln!(w, "#[no_mangle]").unwrap();
        if ok_type != "()" {
+               writeln!(w, "/// Creates a new {} in the success state.", mangled_container).unwrap();
                writeln!(w, "pub extern \"C\" fn {}_ok(o: {}) -> {} {{", mangled_container, ok_type, mangled_container).unwrap();
        } else {
+               writeln!(w, "/// Creates a new {} in the success state.", mangled_container).unwrap();
                writeln!(w, "pub extern \"C\" fn {}_ok() -> {} {{", mangled_container, mangled_container).unwrap();
        }
        writeln!(w, "\t{} {{", mangled_container).unwrap();
@@ -75,8 +98,10 @@ pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
 
        writeln!(w, "#[no_mangle]").unwrap();
        if err_type != "()" {
+               writeln!(w, "/// Creates a new {} in the error state.", mangled_container).unwrap();
                writeln!(w, "pub extern \"C\" fn {}_err(e: {}) -> {} {{", mangled_container, err_type, mangled_container).unwrap();
        } else {
+               writeln!(w, "/// Creates a new {} in the error state.", mangled_container).unwrap();
                writeln!(w, "pub extern \"C\" fn {}_err() -> {} {{", mangled_container, mangled_container).unwrap();
        }
        writeln!(w, "\t{} {{", mangled_container).unwrap();
@@ -92,6 +117,7 @@ pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
        writeln!(w, "}}").unwrap();
 
        writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Frees any resources used by the {}.", mangled_container).unwrap();
        writeln!(w, "pub extern \"C\" fn {}_free(_res: {}) {{ }}", mangled_container, mangled_container).unwrap();
        writeln!(w, "impl Drop for {} {{", mangled_container).unwrap();
        writeln!(w, "\tfn drop(&mut self) {{").unwrap();
@@ -168,6 +194,8 @@ pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
                writeln!(w, "\t}}").unwrap();
                writeln!(w, "}}").unwrap();
                writeln!(w, "#[no_mangle]").unwrap();
+               writeln!(w, "/// Creates a new {} which has the same data as `orig`", mangled_container).unwrap();
+               writeln!(w, "/// but with all dynamically-allocated buffers duplicated in new buffers.").unwrap();
                writeln!(w, "pub extern \"C\" fn {}_clone(orig: &{}) -> {} {{ orig.clone() }}", mangled_container, mangled_container, mangled_container).unwrap();
        }
 }
@@ -175,8 +203,13 @@ pub fn write_result_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
 /// Writes out a C-callable concrete Vec<A> struct and utility methods
 pub fn write_vec_block<W: std::io::Write>(w: &mut W, mangled_container: &str, inner_type: &str, clonable: bool) {
        writeln!(w, "#[repr(C)]").unwrap();
+       writeln!(w, "/// A dynamically-allocated array of {}s of arbitrary size.", inner_type).unwrap();
+       writeln!(w, "/// This corresponds to std::vector in C++").unwrap();
        writeln!(w, "pub struct {} {{", mangled_container).unwrap();
+       writeln!(w, "\t/// The elements in the array.").unwrap();
+       writeln!(w, "\t/// If datalen is non-0 this must be a valid, non-NULL pointer allocated by malloc().").unwrap();
        writeln!(w, "\tpub data: *mut {},", inner_type).unwrap();
+       writeln!(w, "\t/// The number of elements pointed to by `data`.").unwrap();
        writeln!(w, "\tpub datalen: usize").unwrap();
        writeln!(w, "}}").unwrap();
 
@@ -202,6 +235,7 @@ pub fn write_vec_block<W: std::io::Write>(w: &mut W, mangled_container: &str, in
        writeln!(w, "}}").unwrap();
 
        writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Frees the buffer pointed to by `data` if `datalen` is non-0.").unwrap();
        writeln!(w, "pub extern \"C\" fn {}_free(_res: {}) {{ }}", mangled_container, mangled_container).unwrap();
        writeln!(w, "impl Drop for {} {{", mangled_container).unwrap();
        writeln!(w, "\tfn drop(&mut self) {{").unwrap();
@@ -224,8 +258,10 @@ pub fn write_vec_block<W: std::io::Write>(w: &mut W, mangled_container: &str, in
 /// Writes out a C-callable concrete (A, B, ...) struct and utility methods
 pub fn write_tuple_block<W: std::io::Write>(w: &mut W, mangled_container: &str, types: &[String], clonable: bool) {
        writeln!(w, "#[repr(C)]").unwrap();
+       writeln!(w, "/// A tuple of {} elements. See the individual fields for the types contained.", types.len()).unwrap();
        writeln!(w, "pub struct {} {{", mangled_container).unwrap();
        for (idx, ty) in types.iter().enumerate() {
+               writeln!(w, "\t/// The element at position {}", idx).unwrap();
                writeln!(w, "\tpub {}: {},", ('a' as u8 + idx as u8) as char, ty).unwrap();
        }
        writeln!(w, "}}").unwrap();
@@ -267,9 +303,12 @@ pub fn write_tuple_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
                writeln!(w, "\t}}").unwrap();
                writeln!(w, "}}").unwrap();
                writeln!(w, "#[no_mangle]").unwrap();
+               writeln!(w, "/// Creates a new tuple which has the same data as `orig`").unwrap();
+               writeln!(w, "/// but with all dynamically-allocated buffers duplicated in new buffers.").unwrap();
                writeln!(w, "pub extern \"C\" fn {}_clone(orig: &{}) -> {} {{ orig.clone() }}", mangled_container, mangled_container, mangled_container).unwrap();
        }
 
+       writeln!(w, "/// Creates a new {} from the contained elements.", mangled_container).unwrap();
        write!(w, "#[no_mangle]\npub extern \"C\" fn {}_new(", mangled_container).unwrap();
        for (idx, gen) in types.iter().enumerate() {
                write!(w, "{}{}: ", if idx != 0 { ", " } else { "" }, ('a' as u8 + idx as u8) as char).unwrap();
@@ -284,7 +323,54 @@ pub fn write_tuple_block<W: std::io::Write>(w: &mut W, mangled_container: &str,
        writeln!(w, "}}\n}}\n").unwrap();
 
        writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Frees any resources used by the {}.", mangled_container).unwrap();
+       writeln!(w, "pub extern \"C\" fn {}_free(_res: {}) {{ }}", mangled_container, mangled_container).unwrap();
+}
+
+/// Writes out a C-callable concrete Option<A> struct and utility methods
+pub fn write_option_block<W: std::io::Write>(w: &mut W, mangled_container: &str, inner_type: &str, clonable: bool) {
+       writeln!(w, "#[repr(C)]").unwrap();
+       if clonable {
+               writeln!(w, "#[derive(Clone)]").unwrap();
+       }
+       writeln!(w, "/// An enum which can either contain a {} or not", inner_type).unwrap();
+       writeln!(w, "pub enum {} {{", mangled_container).unwrap();
+       writeln!(w, "\t/// When we're in this state, this {} contains a {}", mangled_container, inner_type).unwrap();
+       writeln!(w, "\tSome({}),", inner_type).unwrap();
+       writeln!(w, "\t/// When we're in this state, this {} contains nothing", mangled_container).unwrap();
+       writeln!(w, "\tNone").unwrap();
+       writeln!(w, "}}").unwrap();
+
+       writeln!(w, "impl {} {{", mangled_container).unwrap();
+       writeln!(w, "\t#[allow(unused)] pub(crate) fn is_some(&self) -> bool {{").unwrap();
+       writeln!(w, "\t\tif let Self::Some(_) = self {{ true }} else {{ false }}").unwrap();
+       writeln!(w, "\t}}").unwrap();
+       writeln!(w, "\t#[allow(unused)] pub(crate) fn take(mut self) -> {} {{", inner_type).unwrap();
+       writeln!(w, "\t\tif let Self::Some(v) = self {{ v }} else {{ unreachable!() }}").unwrap();
+       writeln!(w, "\t}}").unwrap();
+       writeln!(w, "}}").unwrap();
+
+       writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Constructs a new {} containing a {}", mangled_container, inner_type).unwrap();
+       writeln!(w, "pub extern \"C\" fn {}_some(o: {}) -> {} {{", mangled_container, inner_type, mangled_container).unwrap();
+       writeln!(w, "\t{}::Some(o)", mangled_container).unwrap();
+       writeln!(w, "}}").unwrap();
+
+       writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Constructs a new {} containing nothing", mangled_container).unwrap();
+       writeln!(w, "pub extern \"C\" fn {}_none() -> {} {{", mangled_container, mangled_container).unwrap();
+       writeln!(w, "\t{}::None", mangled_container).unwrap();
+       writeln!(w, "}}").unwrap();
+
+       writeln!(w, "#[no_mangle]").unwrap();
+       writeln!(w, "/// Frees any resources associated with the {}, if we are in the Some state", inner_type).unwrap();
        writeln!(w, "pub extern \"C\" fn {}_free(_res: {}) {{ }}", mangled_container, mangled_container).unwrap();
+       if clonable {
+               writeln!(w, "#[no_mangle]").unwrap();
+               writeln!(w, "/// Creates a new {} which has the same data as `orig`", mangled_container).unwrap();
+               writeln!(w, "/// but with all dynamically-allocated buffers duplicated in new buffers.").unwrap();
+               writeln!(w, "pub extern \"C\" fn {}_clone(orig: &{}) -> {} {{ orig.clone() }}", mangled_container, mangled_container, mangled_container).unwrap();
+       }
 }
 
 /// Prints the docs from a given attribute list unless its tagged no export
@@ -402,12 +488,7 @@ pub fn write_method_params<W: std::io::Write>(w: &mut W, sig: &syn::Signature, t
                                        return;
                                }
                        }
-                       if let syn::Type::Reference(r) = &**rtype {
-                               // We can't return a reference, cause we allocate things on the stack.
-                               types.write_c_type(w, &*r.elem, generics, true);
-                       } else {
-                               types.write_c_type(w, &*rtype, generics, true);
-                       }
+                       types.write_c_type(w, &*rtype, generics, true);
                },
                _ => {},
        }
@@ -418,7 +499,7 @@ pub fn write_method_params<W: std::io::Write>(w: &mut W, sig: &syn::Signature, t
 /// mut ret = " assuming the next print will be the unmapped Rust function to call followed by the
 /// parameters we mapped to/from C here.
 pub fn write_method_var_decl_body<W: std::io::Write>(w: &mut W, sig: &syn::Signature, extra_indent: &str, types: &TypeResolver, generics: Option<&GenericTypes>, to_c: bool) {
-       let mut num_unused = 0;
+       let mut num_unused = 0u32;
        for inp in sig.inputs.iter() {
                match inp {
                        syn::FnArg::Receiver(_) => {},
@@ -447,7 +528,7 @@ pub fn write_method_var_decl_body<W: std::io::Write>(w: &mut W, sig: &syn::Signa
                                        },
                                        syn::Pat::Wild(w) => {
                                                if !w.attrs.is_empty() { unimplemented!(); }
-                                               write_new_var!(syn::Ident::new(&format!("unused_{}", num_unused), Span::call_site()), *arg.ty);
+                                               write_new_var!(format_ident!("unused_{}", num_unused), *arg.ty);
                                                num_unused += 1;
                                        },
                                        _ => unimplemented!(),
@@ -540,7 +621,7 @@ pub fn write_method_call_params<W: std::io::Write>(w: &mut W, sig: &syn::Signatu
                                // If we're returning "Self" (and not "Self::X"), just do it manually
                                write!(w, "{} {{ inner: Box::into_raw(Box::new(ret)), is_owned: true }}", this_type).unwrap();
                        } else if to_c {
-                               let new_var = types.write_from_c_conversion_new_var(w, &syn::Ident::new("ret", Span::call_site()), rtype, generics);
+                               let new_var = types.write_from_c_conversion_new_var(w, &format_ident!("ret"), rtype, generics);
                                if new_var {
                                        write!(w, "\n\t{}", extra_indent).unwrap();
                                }
@@ -548,13 +629,12 @@ pub fn write_method_call_params<W: std::io::Write>(w: &mut W, sig: &syn::Signatu
                                write!(w, "ret").unwrap();
                                types.write_from_c_conversion_suffix(w, &*rtype, generics);
                        } else {
-                               let ret_returned = if let syn::Type::Reference(_) = &**rtype { true } else { false };
-                               let new_var = types.write_to_c_conversion_new_var(w, &syn::Ident::new("ret", Span::call_site()), &rtype, generics, true);
+                               let new_var = types.write_to_c_conversion_new_var(w, &format_ident!("ret"), &rtype, generics, true);
                                if new_var {
                                        write!(w, "\n\t{}", extra_indent).unwrap();
                                }
                                types.write_to_c_conversion_inline_prefix(w, &rtype, generics, true);
-                               write!(w, "{}ret", if ret_returned && !new_var { "*" } else { "" }).unwrap();
+                               write!(w, "ret").unwrap();
                                types.write_to_c_conversion_inline_suffix(w, &rtype, generics, true);
                        }
                }