1 from bindingstypes import ConvInfo
4 def first_to_lower(string: str) -> str:
6 return first.lower() + string[1:]
10 def __init__(self, DEBUG):
12 self.c_type_map = dict(
13 uint8_t = ['number', 'Uint8Array'],
14 uint16_t = ['number', 'Uint16Array'],
15 uint32_t = ['number', 'Uint32Array'],
19 self.common_base = """
20 export default class CommonBase {
22 ptrs_to: object[] = []; // new LinkedList(); TODO: build linked list implementation
23 protected constructor(ptr: number) { this.ptr = ptr; }
24 public _test_only_get_ptr(): number { return this.ptr; }
25 protected finalize() {
26 // TODO: finalize myself
31 self.c_file_pfx = """#include <rust_types.h>
32 #include <stdatomic.h>
33 #include <lightning.h>
35 // These should be provided...somehow...
36 void *memset(void *s, int c, size_t n);
37 void *memcpy(void *dest, const void *src, size_t n);
38 int memcmp(const void *s1, const void *s2, size_t n);
40 void __attribute__((noreturn)) abort(void);
41 void assert(scalar expression);
45 self.c_file_pfx = self.c_file_pfx + """
46 void *malloc(size_t size);
49 #define MALLOC(a, _) malloc(a)
50 #define FREE(p) if ((long)(p) > 1024) { free(p); }
51 #define DO_ASSERT(a) (void)(a)
55 self.c_file_pfx = self.c_file_pfx + """
56 // Always run a, then assert it is true:
57 #define DO_ASSERT(a) do { bool _assert_val = (a); assert(_assert_val); } while(0)
58 // Assert a is true or do nothing
59 #define CHECK(a) DO_ASSERT(a)
61 // Running a leak check across all the allocations and frees of the JDK is a mess,
62 // so instead we implement our own naive leak checker here, relying on the -wrap
63 // linker option to wrap malloc/calloc/realloc/free, tracking everyhing allocated
64 // and free'd in Rust or C across the generated bindings shared library.
67 typedef struct allocation {
68 struct allocation* next;
70 const char* struct_name;
72 static allocation* allocation_ll = NULL;
74 void* __real_malloc(size_t len);
75 void* __real_calloc(size_t nmemb, size_t len);
76 static void new_allocation(void* res, const char* struct_name) {
77 allocation* new_alloc = __real_malloc(sizeof(allocation));
79 new_alloc->struct_name = struct_name;
80 new_alloc->next = allocation_ll;
81 allocation_ll = new_alloc;
83 static void* MALLOC(size_t len, const char* struct_name) {
84 void* res = __real_malloc(len);
85 new_allocation(res, struct_name);
88 void __real_free(void* ptr);
89 static void alloc_freed(void* ptr) {
91 allocation* it = allocation_ll;
92 while (it->ptr != ptr) {
93 p = it; it = it->next;
95 //XXX: fprintf(stderr, "Tried to free unknown pointer %p\\n", ptr);
96 return; // addrsan should catch malloc-unknown and print more info than we have
99 if (p) { p->next = it->next; } else { allocation_ll = it->next; }
100 DO_ASSERT(it->ptr == ptr);
103 static void FREE(void* ptr) {
104 if ((long)ptr < 1024) return; // Rust loves to create pointers to the NULL page for dummys
109 void* __wrap_malloc(size_t len) {
110 void* res = __real_malloc(len);
111 new_allocation(res, "malloc call");
114 void* __wrap_calloc(size_t nmemb, size_t len) {
115 void* res = __real_calloc(nmemb, len);
116 new_allocation(res, "calloc call");
119 void __wrap_free(void* ptr) {
120 if (ptr == NULL) return;
125 void* __real_realloc(void* ptr, size_t newlen);
126 void* __wrap_realloc(void* ptr, size_t len) {
127 if (ptr != NULL) alloc_freed(ptr);
128 void* res = __real_realloc(ptr, len);
129 new_allocation(res, "realloc call");
132 void __wrap_reallocarray(void* ptr, size_t new_sz) {
133 // Rust doesn't seem to use reallocarray currently
137 void __attribute__((destructor)) check_leaks() {
138 for (allocation* a = allocation_ll; a != NULL; a = a->next) {
139 //XXX: fprintf(stderr, "%s %p remains\\n", a->struct_name, a->ptr);
141 DO_ASSERT(allocation_ll == NULL);
144 self.c_file_pfx = self.c_file_pfx + """
145 // We assume that CVec_u8Z and u8slice are the same size and layout (and thus pointers to the two can be mixed)
146 _Static_assert(sizeof(LDKCVec_u8Z) == sizeof(LDKu8slice), "Vec<u8> and [u8] need to have been mapped identically");
147 _Static_assert(offsetof(LDKCVec_u8Z, data) == offsetof(LDKu8slice, data), "Vec<u8> and [u8] need to have been mapped identically");
148 _Static_assert(offsetof(LDKCVec_u8Z, datalen) == offsetof(LDKu8slice, datalen), "Vec<u8> and [u8] need to have been mapped identically");
150 _Static_assert(sizeof(void*) == 4, "Pointers mut be 32 bits");
152 typedef struct int64_tArray {uint32_t len;int64_t *ptr;} int64_tArray;
153 typedef struct uint32_tArray {uint32_t len;int32_t *ptr;} uint32_tArray;
154 typedef struct int8_tArray {uint32_t len;int8_t *ptr;} int8_tArray;
156 typedef bool jboolean;
160 self.hu_struct_file_prefix = f"""
161 import CommonBase from './CommonBase';
162 import * as bindings from '../bindings' // TODO: figure out location
165 self.c_fn_ty_pfx = ""
166 self.c_fn_name_pfx = ""
167 self.c_fn_args_pfx = "void* ctx_TODO"
168 self.file_ext = ".ts"
169 self.ptr_c_ty = "uint32_t"
170 self.ptr_native_ty = "uint32_t"
171 self.result_c_ty = "uint32_t"
172 self.ptr_arr = "uint32_tArray"
173 self.get_native_arr_len_call = ("", ".len")
174 self.get_native_arr_ptr_call = ("", ".ptr")
176 def release_native_arr_ptr_call(self, arr_var, arr_ptr_var):
178 def create_native_arr_call(self, arr_len, ty_info):
179 if ty_info.c_ty == "int8_tArray":
180 return "{ .len = " + arr_len + ", .ptr = MALLOC(" + arr_len + ", \"Native " + ty_info.c_ty + " Bytes\") }"
181 elif ty_info.c_ty == "int64_tArray":
182 return "{ .len = " + arr_len + ", .ptr = MALLOC(" + arr_len + " * sizeof(int64_t), \"Native " + ty_info.c_ty + " Bytes\") }"
183 elif ty_info.c_ty == "uint32_tArray":
184 return "{ .len = " + arr_len + ", .ptr = MALLOC(" + arr_len + " * sizeof(int32_t), \"Native " + ty_info.c_ty + " Bytes\") }"
186 print("Need to create arr!", ty_info.c_ty)
188 def set_native_arr_contents(self, arr_name, arr_len, ty_info):
189 if ty_info.c_ty == "int8_tArray":
190 return ("memcpy(" + arr_name + ".ptr, ", ", " + arr_len + ")")
193 def get_native_arr_contents(self, arr_name, dest_name, arr_len, ty_info, copy):
194 if ty_info.c_ty == "int8_tArray":
196 return "memcpy(" + dest_name + ", " + arr_name + ".ptr, " + arr_len + ")"
198 return arr_name + ".ptr"
200 return "(" + ty_info.subty.c_ty + "*) " + arr_name + ".ptr"
201 def get_native_arr_elem(self, arr_name, idxc, ty_info):
202 assert False # Only called if above is None
203 def cleanup_native_arr_ref_contents(self, arr_name, dest_name, arr_len, ty_info):
204 if ty_info.c_ty == "int8_tArray":
209 def init_str(self, c_array_class_caches):
212 def native_c_unitary_enum_map(self, struct_name, variants):
213 out_c = "static inline " + struct_name + " " + struct_name + "_from_js(int32_t ord) {\n"
214 out_c = out_c + "\tswitch (ord) {\n"
217 out_typescript_enum_fields = ""
220 out_c = out_c + "\t\tcase %d: return %s;\n" % (ord_v, var)
222 out_typescript_enum_fields += f"{var},\n\t\t\t\t"
223 out_c = out_c + "\t}\n"
224 out_c = out_c + "\tabort();\n"
225 out_c = out_c + "}\n"
227 out_c = out_c + "static inline int32_t " + struct_name + "_to_js(" + struct_name + " val) {\n"
228 out_c = out_c + "\tswitch (val) {\n"
231 out_c = out_c + "\t\tcase " + var + ": return %d;\n" % ord_v
233 out_c = out_c + "\t\tdefault: abort();\n"
234 out_c = out_c + "\t}\n"
235 out_c = out_c + "}\n"
237 out_typescript_enum = f"""
238 export enum {struct_name} {{
239 {out_typescript_enum_fields}
243 return (out_c, out_typescript_enum, "")
245 def c_unitary_enum_to_native_call(self, ty_info):
246 return (ty_info.rust_obj + "_to_js(", ")")
247 def native_unitary_enum_to_c_call(self, ty_info):
248 return (ty_info.rust_obj + "_from_js(", ")")
250 def c_complex_enum_pass_ty(self, struct_name):
253 def c_constr_native_complex_enum(self, struct_name, variant, c_params):
254 ret = "0 /* " + struct_name + " - " + variant + " */"
255 for param in c_params:
256 ret = ret + "; (void) " + param
259 def native_c_map_trait(self, struct_name, field_var_conversions, field_function_lines):
260 out_java = "out_java:native_c_map_trait"
261 out_java_trait = "out_java_trait:native_c_map_trait"
262 out_c = "out_c:native_c_map_trait"
268 constructor_arguments = ""
269 super_instantiator = ""
270 pointer_to_adder = ""
271 impl_constructor_arguments = ""
272 for var in field_var_conversions:
273 if isinstance(var, ConvInfo):
274 constructor_arguments += f", {first_to_lower(var.arg_name)}?: {var.java_hu_ty}"
275 impl_constructor_arguments += f", {var.arg_name}: {var.java_hu_ty}"
276 if var.from_hu_conv is not None:
277 super_instantiator += ", " + var.from_hu_conv[0]
278 if var.from_hu_conv[1] != "":
279 pointer_to_adder += var.from_hu_conv[1] + ";\n"
281 super_instantiator += ", " + first_to_lower(var.arg_name)
283 constructor_arguments += f", {first_to_lower(var[1])}?: bindings.{var[0]}"
284 super_instantiator += ", " + first_to_lower(var[1])
285 pointer_to_adder += "this.ptrs_to.push(" + first_to_lower(var[1]) + ");\n"
286 impl_constructor_arguments += f", {first_to_lower(var[1])}_impl: {var[0].replace('LDK', '')}.{var[0].replace('LDK', '')}Interface"
288 # BUILD INTERFACE METHODS
289 out_java_interface = ""
290 out_interface_implementation_overrides = ""
292 for fn_line in field_function_lines:
293 java_method_descriptor = ""
294 if fn_line.fn_name != "free" and fn_line.fn_name != "clone":
295 out_java_interface += fn_line.fn_name + "("
296 out_interface_implementation_overrides += f"{fn_line.fn_name} ("
298 for idx, arg_conv_info in enumerate(fn_line.args_ty):
300 out_java_interface += ", "
301 out_interface_implementation_overrides += ", "
302 out_java_interface += f"{arg_conv_info.arg_name}: {arg_conv_info.java_hu_ty}"
303 out_interface_implementation_overrides += f"{arg_conv_info.arg_name}: {arg_conv_info.java_ty}"
304 java_method_descriptor += arg_conv_info.java_fn_ty_arg
305 out_java_interface += f"): {fn_line.ret_ty_info.java_hu_ty};\n\t\t\t\t"
306 java_method_descriptor += ")" + fn_line.ret_ty_info.java_fn_ty_arg
307 java_methods.append((fn_line.fn_name, java_method_descriptor))
309 out_interface_implementation_overrides += f"): {fn_line.ret_ty_info.java_ty} {{\n"
311 interface_method_override_inset = "\t\t\t\t\t\t"
312 interface_implementation_inset = "\t\t\t\t\t\t\t"
313 for arg_info in fn_line.args_ty:
314 if arg_info.to_hu_conv is not None:
315 out_interface_implementation_overrides += interface_implementation_inset + arg_info.to_hu_conv.replace("\n", "\n\t\t\t\t") + "\n"
317 if fn_line.ret_ty_info.java_ty != "void":
318 out_interface_implementation_overrides += interface_implementation_inset + fn_line.ret_ty_info.java_hu_ty + " ret = arg." + fn_line.fn_name + "("
320 out_interface_implementation_overrides += f"{interface_implementation_inset}arg." + fn_line.fn_name + "("
322 for idx, arg_info in enumerate(fn_line.args_ty):
324 out_interface_implementation_overrides += ", "
325 if arg_info.to_hu_conv_name is not None:
326 out_interface_implementation_overrides += arg_info.to_hu_conv_name
328 out_interface_implementation_overrides += arg_info.arg_name
330 out_interface_implementation_overrides += ");\n"
331 if fn_line.ret_ty_info.java_ty != "void":
332 if fn_line.ret_ty_info.from_hu_conv is not None:
333 out_interface_implementation_overrides = out_interface_implementation_overrides + "\t\t\t\t" + f"result: {fn_line.ret_ty_info.java_ty} = " + fn_line.ret_ty_info.from_hu_conv[0] + ";\n"
334 if fn_line.ret_ty_info.from_hu_conv[1] != "":
335 out_interface_implementation_overrides = out_interface_implementation_overrides + "\t\t\t\t" + fn_line.ret_ty_info.from_hu_conv[1].replace("this", "impl_holder.held") + ";\n"
336 #if fn_line.ret_ty_info.rust_obj in result_types:
337 # XXX: We need to handle this in conversion logic so that its cross-language!
338 # Avoid double-free by breaking the result - we should learn to clone these and then we can be safe instead
339 # out_interface_implementation_overrides = out_interface_implementation_overrides + "\t\t\t\tret.ptr = 0;\n"
340 out_interface_implementation_overrides = out_interface_implementation_overrides + "\t\t\t\treturn result;\n"
342 out_interface_implementation_overrides = out_interface_implementation_overrides + "\t\t\t\treturn ret;\n"
343 out_interface_implementation_overrides += f"{interface_method_override_inset}}},\n\n{interface_method_override_inset}"
347 out_java_trait = f"""
348 {self.hu_struct_file_prefix}
350 export class {struct_name.replace("LDK","")} extends CommonBase {{
352 bindings_instance?: bindings.{struct_name};
354 constructor(ptr?: number, arg?: bindings.{struct_name}{constructor_arguments}) {{
355 if (Number.isFinite(ptr)) {{
357 this.bindings_instance = null;
359 // TODO: private constructor instantiation
360 super(bindings.{struct_name}_new(arg{super_instantiator}));
361 this.ptrs_to.push(arg);
366 protected finalize() {{
367 if (this.ptr != 0) {{
368 bindings.{struct_name.replace("LDK","")}_free(this.ptr);
373 static new_impl(arg: {struct_name.replace("LDK", "")}Interface{impl_constructor_arguments}): {struct_name.replace("LDK", "")} {{
374 const impl_holder: {struct_name}Holder = new {struct_name}Holder();
375 let structImplementation = <bindings.{struct_name}>{{
377 // todo: in-line interface filling
379 {out_interface_implementation_overrides}
381 impl_holder.held = new {struct_name.replace("LDK", "")} (null, structImplementation);
386 export interface {struct_name.replace("LDK", "")}Interface {{
390 class {struct_name}Holder {{
391 held: {struct_name.replace("LDK", "")};
398 java_trait_constr = ") {\n\t\tfinal " + struct_name + "Holder impl_holder = new " + struct_name + "Holder();\n"
399 java_trait_constr = java_trait_constr + "\t\timpl_holder.held = new " + struct_name.replace("LDK", "") + "(new bindings." + struct_name + "() {\n"
400 out_java = out_java + "\tpublic interface " + struct_name + " {\n"
402 for fn_line in field_function_lines:
403 java_meth_descr = "("
404 if fn_line.fn_name != "free" and fn_line.fn_name != "clone":
405 out_java = out_java + "\t\t " + fn_line.ret_ty_info.java_ty + " " + fn_line.fn_name + "("
406 java_trait_constr = java_trait_constr + "\t\t\t@Override public " + fn_line.ret_ty_info.java_ty + " " + fn_line.fn_name + "("
408 for idx, arg_conv_info in enumerate(fn_line.args_ty):
410 out_java = out_java + ", "
411 java_trait_constr = java_trait_constr + ", "
412 # out_java_trait = out_java_trait + ", "
413 out_java = out_java + arg_conv_info.java_ty + " " + arg_conv_info.arg_name
414 # out_java_trait = out_java_trait + arg_conv_info.java_hu_ty + " " + arg_conv_info.arg_name
415 java_trait_constr = java_trait_constr + arg_conv_info.java_ty + " " + arg_conv_info.arg_name
416 java_meth_descr = java_meth_descr + arg_conv_info.java_fn_ty_arg
417 java_meth_descr = java_meth_descr + ")" + fn_line.ret_ty_info.java_fn_ty_arg
418 java_meths.append((fn_line.fn_name, java_meth_descr))
420 out_java = out_java + ");\n"
421 # out_java_trait = out_java_trait + ");\n"
422 java_trait_constr = java_trait_constr + ") {\n"
424 for arg_info in fn_line.args_ty:
425 if arg_info.to_hu_conv is not None:
426 java_trait_constr = java_trait_constr + "\t\t\t\t" + arg_info.to_hu_conv.replace("\n", "\n\t\t\t\t") + "\n"
428 if fn_line.ret_ty_info.java_ty != "void":
429 java_trait_constr = java_trait_constr + "\t\t\t\t" + fn_line.ret_ty_info.java_hu_ty + " ret = arg." + fn_line.fn_name + "("
431 java_trait_constr = java_trait_constr + "\t\t\t\targ." + fn_line.fn_name + "("
433 for idx, arg_info in enumerate(fn_line.args_ty):
435 java_trait_constr = java_trait_constr + ", "
436 if arg_info.to_hu_conv_name is not None:
437 java_trait_constr = java_trait_constr + arg_info.to_hu_conv_name
439 java_trait_constr = java_trait_constr + arg_info.arg_name
441 java_trait_constr = java_trait_constr + ");\n"
442 if fn_line.ret_ty_info.java_ty != "void":
443 if fn_line.ret_ty_info.from_hu_conv is not None:
444 java_trait_constr = java_trait_constr + "\t\t\t\t" + fn_line.ret_ty_info.java_ty + " result = " + fn_line.ret_ty_info.from_hu_conv[0] + ";\n"
445 if fn_line.ret_ty_info.from_hu_conv[1] != "":
446 java_trait_constr = java_trait_constr + "\t\t\t\t" + fn_line.ret_ty_info.from_hu_conv[1].replace("this", "impl_holder.held") + ";\n"
447 #if fn_line.ret_ty_info.rust_obj in result_types:
448 # XXX: We need to handle this in conversion logic so that its cross-language!
449 # Avoid double-free by breaking the result - we should learn to clone these and then we can be safe instead
450 # java_trait_constr = java_trait_constr + "\t\t\t\tret.ptr = 0;\n"
451 java_trait_constr = java_trait_constr + "\t\t\t\treturn result;\n"
453 java_trait_constr = java_trait_constr + "\t\t\t\treturn ret;\n"
454 java_trait_constr = java_trait_constr + "\t\t\t}\n"
455 java_trait_constr = java_trait_constr + "\t\t}"
456 for var in field_var_conversions:
457 if isinstance(var, ConvInfo):
458 java_trait_constr = java_trait_constr + ", " + var.arg_name
460 java_trait_constr = java_trait_constr + ", " + var[1] + ".new_impl(" + var[1] + "_impl).bindings_instance"
461 out_java_trait += java_trait_constr + ");\n\t\treturn impl_holder.held;\n\t}\n"
463 out_java = out_java + "\t}\n"
465 out_java = out_java + "\tpublic static native long " + struct_name + "_new(" + struct_name + " impl"
466 for var in field_var_conversions:
467 if isinstance(var, ConvInfo):
468 out_java = out_java + ", " + var.java_ty + " " + var.arg_name
470 out_java = out_java + ", " + var[0] + " " + var[1]
471 out_java = out_java + ");\n"
472 out_java = out_java + "\tpublic static native " + struct_name + " " + struct_name + "_get_obj_from_jcalls(long val);\n"
474 # Now that we've written out our java code (and created java_meths), generate C
475 out_c = "typedef struct " + struct_name + "_JCalls {\n"
476 out_c = out_c + "\tatomic_size_t refcnt;\n"
477 out_c = out_c + "\tJavaVM *vm;\n"
478 out_c = out_c + "\tjweak o;\n"
479 for var in field_var_conversions:
480 if isinstance(var, ConvInfo):
481 # We're a regular ol' field
485 out_c = out_c + "\t" + var[0] + "_JCalls* " + var[1] + ";\n"
486 for fn in field_function_lines:
487 if fn.fn_name != "free" and fn.fn_name != "clone":
488 out_c = out_c + "\tjmethodID " + fn.fn_name + "_meth;\n"
489 out_c = out_c + "} " + struct_name + "_JCalls;\n"
491 for fn_line in field_function_lines:
492 if fn_line.fn_name == "free":
493 out_c = out_c + "static void " + struct_name + "_JCalls_free(void* this_arg) {\n"
494 out_c = out_c + "\t" + struct_name + "_JCalls *j_calls = (" + struct_name + "_JCalls*) this_arg;\n"
495 out_c = out_c + "\tif (atomic_fetch_sub_explicit(&j_calls->refcnt, 1, memory_order_acquire) == 1) {\n"
496 out_c = out_c + "\t\tJNIEnv *env;\n"
497 out_c = out_c + "\t\tDO_ASSERT((*j_calls->vm)->GetEnv(j_calls->vm, (void**)&env, JNI_VERSION_1_8) == JNI_OK);\n"
498 out_c = out_c + "\t\t(*env)->DeleteWeakGlobalRef(env, j_calls->o);\n"
499 out_c = out_c + "\t\tFREE(j_calls);\n"
500 out_c = out_c + "\t}\n}\n"
502 for idx, fn_line in enumerate(field_function_lines):
503 if fn_line.fn_name != "free" and fn_line.fn_name != "clone":
504 assert fn_line.ret_ty_info.ty_info.get_full_rust_ty()[1] == ""
505 out_c = out_c + fn_line.ret_ty_info.ty_info.get_full_rust_ty()[0] + " " + fn_line.fn_name + "_jcall("
506 if fn_line.self_is_const:
507 out_c = out_c + "const void* this_arg"
509 out_c = out_c + "void* this_arg"
511 for idx, arg in enumerate(fn_line.args_ty):
512 out_c = out_c + ", " + arg.ty_info.get_full_rust_ty()[0] + " " + arg.arg_name + arg.ty_info.get_full_rust_ty()[1]
514 out_c = out_c + ") {\n"
515 out_c = out_c + "\t" + struct_name + "_JCalls *j_calls = (" + struct_name + "_JCalls*) this_arg;\n"
516 out_c = out_c + "\tJNIEnv *env;\n"
517 out_c = out_c + "\tDO_ASSERT((*j_calls->vm)->GetEnv(j_calls->vm, (void**)&env, JNI_VERSION_1_8) == JNI_OK);\n"
519 for arg_info in fn_line.args_ty:
520 if arg_info.ret_conv is not None:
521 out_c = out_c + "\t" + arg_info.ret_conv[0].replace('\n', '\n\t')
522 out_c = out_c + arg_info.arg_name
523 out_c = out_c + arg_info.ret_conv[1].replace('\n', '\n\t') + "\n"
525 out_c = out_c + "\tjobject obj = (*env)->NewLocalRef(env, j_calls->o);\n\tCHECK(obj != NULL);\n"
526 if fn_line.ret_ty_info.c_ty.endswith("Array"):
527 out_c = out_c + "\t" + fn_line.ret_ty_info.c_ty + " arg = (*env)->CallObjectMethod(env, obj, j_calls->" + fn_line.fn_name + "_meth"
528 elif not fn_line.ret_ty_info.passed_as_ptr:
529 out_c = out_c + "\treturn (*env)->Call" + fn_line.ret_ty_info.java_ty.title() + "Method(env, obj, j_calls->" + fn_line.fn_name + "_meth"
531 out_c = out_c + "\t" + fn_line.ret_ty_info.rust_obj + "* ret = (" + fn_line.ret_ty_info.rust_obj + "*)(*env)->CallLongMethod(env, obj, j_calls->" + fn_line.fn_name + "_meth"
533 for idx, arg_info in enumerate(fn_line.args_ty):
534 if arg_info.ret_conv is not None:
535 out_c = out_c + ", " + arg_info.ret_conv_name
537 out_c = out_c + ", " + arg_info.arg_name
538 out_c = out_c + ");\n"
539 if fn_line.ret_ty_info.arg_conv is not None:
540 out_c = out_c + "\t" + fn_line.ret_ty_info.arg_conv.replace("\n", "\n\t") + "\n\treturn " + fn_line.ret_ty_info.arg_conv_name + ";\n"
542 out_c = out_c + "}\n"
544 # Write out a clone function whether we need one or not, as we use them in moving to rust
545 out_c = out_c + "static void* " + struct_name + "_JCalls_clone(const void* this_arg) {\n"
546 out_c = out_c + "\t" + struct_name + "_JCalls *j_calls = (" + struct_name + "_JCalls*) this_arg;\n"
547 out_c = out_c + "\tatomic_fetch_add_explicit(&j_calls->refcnt, 1, memory_order_release);\n"
548 for var in field_var_conversions:
549 if not isinstance(var, ConvInfo):
550 out_c = out_c + "\tatomic_fetch_add_explicit(&j_calls->" + var[1] + "->refcnt, 1, memory_order_release);\n"
551 out_c = out_c + "\treturn (void*) this_arg;\n"
552 out_c = out_c + "}\n"
554 out_c = out_c + "static inline " + struct_name + " " + struct_name + "_init (" + self.c_fn_args_pfx + ", jobject o"
555 for var in field_var_conversions:
556 if isinstance(var, ConvInfo):
557 out_c = out_c + ", " + var.c_ty + " " + var.arg_name
559 out_c = out_c + ", jobject " + var[1]
560 out_c = out_c + ") {\n"
562 out_c = out_c + "\tjclass c = (*env)->GetObjectClass(env, o);\n"
563 out_c = out_c + "\tCHECK(c != NULL);\n"
564 out_c = out_c + "\t" + struct_name + "_JCalls *calls = MALLOC(sizeof(" + struct_name + "_JCalls), \"" + struct_name + "_JCalls\");\n"
565 out_c = out_c + "\tatomic_init(&calls->refcnt, 1);\n"
566 out_c = out_c + "\tDO_ASSERT((*env)->GetJavaVM(env, &calls->vm) == 0);\n"
567 out_c = out_c + "\tcalls->o = (*env)->NewWeakGlobalRef(env, o);\n"
569 for (fn_name, java_meth_descr) in java_meths:
570 if fn_name != "free" and fn_name != "clone":
571 out_c = out_c + "\tcalls->" + fn_name + "_meth = (*env)->GetMethodID(env, c, \"" + fn_name + "\", \"" + java_meth_descr + "\");\n"
572 out_c = out_c + "\tCHECK(calls->" + fn_name + "_meth != NULL);\n"
574 for var in field_var_conversions:
575 if isinstance(var, ConvInfo) and var.arg_conv is not None:
576 out_c = out_c + "\n\t" + var.arg_conv.replace("\n", "\n\t") +"\n"
577 out_c = out_c + "\n\t" + struct_name + " ret = {\n"
578 out_c = out_c + "\t\t.this_arg = (void*) calls,\n"
579 for fn_line in field_function_lines:
580 if fn_line.fn_name != "free" and fn_line.fn_name != "clone":
581 out_c = out_c + "\t\t." + fn_line.fn_name + " = " + fn_line.fn_name + "_jcall,\n"
582 elif fn_line.fn_name == "free":
583 out_c = out_c + "\t\t.free = " + struct_name + "_JCalls_free,\n"
585 out_c = out_c + "\t\t.clone = " + struct_name + "_JCalls_clone,\n"
586 for var in field_var_conversions:
587 if isinstance(var, ConvInfo):
588 if var.arg_conv_name is not None:
589 out_c = out_c + "\t\t." + var.arg_name + " = " + var.arg_conv_name + ",\n"
590 out_c = out_c + "\t\t.set_" + var.arg_name + " = NULL,\n"
592 out_c = out_c + "\t\t." + var.var_name + " = " + var.var_name + ",\n"
593 out_c = out_c + "\t\t.set_" + var.var_name + " = NULL,\n"
595 out_c = out_c + "\t\t." + var[1] + " = " + var[0] + "_init(env, clz, " + var[1] + "),\n"
596 out_c = out_c + "\t};\n"
597 for var in field_var_conversions:
598 if not isinstance(var, ConvInfo):
599 out_c = out_c + "\tcalls->" + var[1] + " = ret." + var[1] + ".this_arg;\n"
600 out_c = out_c + "\treturn ret;\n"
601 out_c = out_c + "}\n"
603 out_c = out_c + self.c_fn_ty_pfx + "long " + self.c_fn_name_pfx + struct_name.replace("_", "_1") + "_1new (" + self.c_fn_args_pfx + ", jobject o"
604 for var in field_var_conversions:
605 if isinstance(var, ConvInfo):
606 out_c = out_c + ", " + var.c_ty + " " + var.arg_name
608 out_c = out_c + ", jobject " + var[1]
609 out_c = out_c + ") {\n"
610 out_c = out_c + "\t" + struct_name + " *res_ptr = MALLOC(sizeof(" + struct_name + "), \"" + struct_name + "\");\n"
611 out_c = out_c + "\t*res_ptr = " + struct_name + "_init(env, clz, o"
612 for var in field_var_conversions:
613 if isinstance(var, ConvInfo):
614 out_c = out_c + ", " + var.arg_name
616 out_c = out_c + ", " + var[1]
617 out_c = out_c + ");\n"
618 out_c = out_c + "\treturn (long)res_ptr;\n"
619 out_c = out_c + "}\n"
621 out_c = out_c + self.c_fn_ty_pfx + "jobject " + self.c_fn_name_pfx + struct_name.replace("_", "_1") + "_1get_1obj_1from_1jcalls (" + self.c_fn_args_pfx + ", " + self.ptr_c_ty + " val) {\n"
622 out_c = out_c + "\tjobject ret = (*env)->NewLocalRef(env, ((" + struct_name + "_JCalls*)val)->o);\n"
623 out_c = out_c + "\tCHECK(ret != NULL);\n"
624 out_c = out_c + "\treturn ret;\n"
625 out_c = out_c + "}\n"
628 return (out_java, out_java_trait, out_c)
630 def map_complex_enum(self, struct_name, variant_list, camel_to_snake):
631 java_hu_type = struct_name.replace("LDK", "")
637 out_java_enum += (self.hu_struct_file_prefix)
638 out_java_enum += ("export default class " + java_hu_type + " extends CommonBase {\n")
639 out_java_enum += ("\tprotected constructor(_dummy: object, ptr: number) { super(ptr); }\n")
640 out_java_enum += ("\tprotected finalize() {\n")
641 out_java_enum += ("\t\tsuper.finalize();\n")
642 out_java_enum += ("\t\tif (this.ptr != 0) { bindings." + java_hu_type + "_free(this.ptr); }\n")
643 out_java_enum += ("\t}\n")
644 out_java_enum += f"\tstatic constr_from_ptr(ptr: number): {java_hu_type} {{\n"
645 out_java_enum += (f"\t\tconst raw_val: bindings.{struct_name} = bindings." + struct_name + "_ref_from_ptr(ptr);\n")
646 java_hu_subclasses = ""
648 out_java += ("\tpublic static class " + struct_name + " {\n")
649 out_java += ("\t\tprivate " + struct_name + "() {}\n")
650 for var in variant_list:
651 out_java += ("\t\texport class " + var.var_name + " extends " + struct_name + " {\n")
652 java_hu_subclasses = java_hu_subclasses + "export class " + var.var_name + " extends " + java_hu_type + " {\n"
653 out_java_enum += ("\t\tif (raw_val instanceof bindings." + struct_name + "." + var.var_name + ") {\n")
654 out_java_enum += ("\t\t\treturn new " + var.var_name + "(this.ptr, raw_val);\n")
655 init_meth_params = ""
658 for idx, field_ty in enumerate(var.fields):
659 out_java += ("\t\t\tpublic " + field_ty.java_ty + " " + field_ty.arg_name + ";\n")
660 java_hu_subclasses = java_hu_subclasses + "\tpublic " + field_ty.arg_name + f": {field_ty.java_hu_ty};\n"
661 if field_ty.to_hu_conv is not None:
662 hu_conv_body = hu_conv_body + "\t\tconst " + field_ty.arg_name + f": {field_ty.java_ty} = obj." + field_ty.arg_name + ";\n"
663 hu_conv_body = hu_conv_body + "\t\t" + field_ty.to_hu_conv.replace("\n", "\n\t\t\t") + "\n"
664 hu_conv_body = hu_conv_body + "\t\tthis." + field_ty.arg_name + " = " + field_ty.to_hu_conv_name + ";\n"
666 hu_conv_body = hu_conv_body + "\t\tthis." + field_ty.arg_name + " = obj." + field_ty.arg_name + ";\n"
668 init_meth_params = init_meth_params + ", "
669 init_meth_params = init_meth_params + field_ty.java_ty + " " + field_ty.arg_name
670 init_meth_body = init_meth_body + "this." + field_ty.arg_name + " = " + field_ty.arg_name + "; "
671 out_java += ("\t\t\t" + var.var_name + "(" + init_meth_params + ") { ")
672 out_java += (init_meth_body)
674 out_java += ("\t\t}\n")
675 out_java_enum += ("\t\t}\n")
676 java_hu_subclasses = java_hu_subclasses + "\tprivate constructor(ptr: number, obj: bindings." + struct_name + "." + var.var_name + ") {\n\t\tsuper(null, ptr);\n"
677 java_hu_subclasses = java_hu_subclasses + hu_conv_body
678 java_hu_subclasses = java_hu_subclasses + "\t}\n}\n"
679 out_java += ("\t\tstatic native void init();\n")
680 out_java += ("\t}\n")
681 out_java_enum += ("\t\tthrow new Error('oops, this should be unreachable'); // Unreachable without extending the (internal) bindings interface\n\t}\n\n")
682 out_java += ("\tstatic { " + struct_name + ".init(); }\n")
683 out_java += ("\tpublic static native " + struct_name + " " + struct_name + "_ref_from_ptr(long ptr);\n");
685 out_c += (self.c_fn_ty_pfx + self.c_complex_enum_pass_ty(struct_name) + " " + self.c_fn_name_pfx + struct_name.replace("_", "_1") + "_1ref_1from_1ptr (" + self.c_fn_args_pfx + ", " + self.ptr_c_ty + " ptr) {\n")
686 out_c += ("\t" + struct_name + " *obj = (" + struct_name + "*)ptr;\n")
687 out_c += ("\tswitch(obj->tag) {\n")
688 for var in variant_list:
689 out_c += ("\t\tcase " + struct_name + "_" + var.var_name + ": {\n")
691 for idx, field_map in enumerate(var.fields):
692 if field_map.ret_conv is not None:
693 out_c += ("\t\t\t" + field_map.ret_conv[0].replace("\n", "\n\t\t\t"))
694 out_c += ("obj->" + camel_to_snake(var.var_name) + "." + field_map.arg_name)
695 out_c += (field_map.ret_conv[1].replace("\n", "\n\t\t\t") + "\n")
696 c_params.append(field_map.ret_conv_name)
698 c_params.append("obj->" + camel_to_snake(var.var_name) + "." + field_map.arg_name)
699 out_c += ("\t\t\treturn " + self.c_constr_native_complex_enum(struct_name, var.var_name, c_params) + ";\n")
701 out_c += ("\t\tdefault: abort();\n")
702 out_c += ("\t}\n}\n")
703 out_java_enum += ("}\n")
704 out_java_enum += (java_hu_subclasses)
705 return (out_java, out_java_enum, out_c)