diff options
-rwxr-xr-x | main.py | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -575,6 +575,7 @@ class CodeGen(object): void_source.write("#include <stdlib.h>\n") void_source.write('#include "aggregate.h"\n') if self.argparser.args.calloc: void_source.write(text.ft_calloc_def) + if self.argparser.args.luaalloc: void_source.write(text.ft_luanewuserdata_def) #void_source.write("void** void_train;\n") #void_source.write("uint64_t current_void_size = 0U;\n") #void_source.write("uint64_t current_void_count = 0U;\n") @@ -801,6 +802,7 @@ class CodeGen(object): struct_source_c.write('#include "stdio.h"\n') struct_source.write('#include <unistd.h>\n') if self.argparser.args.calloc: struct_source_c.write(text.ft_calloc_def) + elif self.argparser.args.luaalloc: struct_source_c.write(text.ft_luanewuserdata_def) struct_source.write(text.header_inttype) struct_source_c.write(text.c_read_leb_u_def + "\n") struct_source_c.write(text.c_read_leb_s_def + "\n") |