vararg

Author: tgymnich

src/device/intrinsics/output.jl

@@ -49,9 +49,7 @@ end
 
         # create functions
         param_types = LLVMType[convert(LLVMType, typ) for typ in arg_types]
-        llvm_f, llvm_ft = create_function(T_void, param_types)
-        push!(function_attributes(llvm_f), EnumAttribute("alwaysinline", 0))
-
+        llvm_f, llvm_ft = create_function(T_void, LLVMType[]; vararg=true)
         mod = LLVM.parent(llvm_f)
 
         # generate IR
@@ -60,42 +58,54 @@ end
             position!(builder, entry)
 
             str = globalstring_ptr!(builder, String(fmt), addrspace=2)
-            argsize = 0
-
-            # construct and fill args buffer
-            if isempty(argspec)
-                buffer = LLVM.PointerNull(T_pint8)
-            else
-                argtypes = LLVM.StructType("os_log_args")
-                elements!(argtypes, param_types)
-
-                args = alloca!(builder, argtypes)
-                for (i, param) in enumerate(parameters(llvm_f))
-                    p = struct_gep!(builder, argtypes, args, i-1)
-                    store!(builder, param, p)
-                end
 
-                dl = datalayout(mod) 
-                argsize = sizeof(dl, argtypes)
-                buffer = bitcast!(builder, args, T_pint8)
-            end
+            # compute argsize
+            argtypes = LLVM.StructType("os_log_args")
+            elements!(argtypes, param_types)
+            dl = datalayout(mod) 
+            argsize = sizeof(dl, argtypes)
+
+            # argbuffer
+            alloc = alloca!(builder, T_pint8)
+            buffer = bitcast!(builder, alloc, T_pint8)
+
+            va_start_fty = LLVM.FunctionType(T_void, [T_pint8])
+            va_start = LLVM.Function(mod, "llvm.va_start", va_start_fty)
+            call!(builder, va_start_fty, va_start, [buffer])
 
             # invoke @air.os_log and return
             subsystem_str = globalstring_ptr!(builder, MTLLOG_SUBSYSTEM, addrspace=2)
-            # subsystem_str = LLVM.PointerNull(T_pint8a2)
-            # intptr = LLVM.ConstantInt(T_int64, Int64(-1))
-            # category_str = const_inttoptr(intptr, T_pint8a2)
             category_str = globalstring_ptr!(builder, MTLLOG_CATEGRORY, addrspace=2)
             log_type = LLVM.ConstantInt(T_int32, __METAL_OS_LOG_TYPE_DEBUG__)
-            arg_size = LLVM.ConstantInt(T_int64, Int64(argsize))
-
             os_log_fty = LLVM.FunctionType(T_void, [T_pint8a2, T_pint8a2, T_int32, T_pint8a2, T_pint8, T_int64])
             os_log = LLVM.Function(mod, "air.os_log", os_log_fty)
-            call!(builder, os_log_fty, os_log, [subsystem_str, category_str, log_type, str, buffer, arg_size])
+            
+            arg_size = LLVM.ConstantInt(T_int64, Int64(argsize))
+
+            arg_ptr = load!(builder, T_pint8, alloc)
+            
+            call!(builder, os_log_fty, os_log, [subsystem_str, category_str, log_type, str, arg_ptr, arg_size])
+
+            va_end_fty = LLVM.FunctionType(T_void, [T_pint8])
+            va_end = LLVM.Function(mod, "llvm.va_end", va_end_fty)
+            call!(builder, va_end_fty, va_end, [buffer])
+
+            ret!(builder)
+        end
+
+
+        wrapper_f, wrapper_ft = create_function(T_void, param_types)
+
+        @dispose builder=IRBuilder() begin
+            entry = BasicBlock(wrapper_f, "entry")
+            position!(builder, entry)
+
+            call!(builder, llvm_ft, llvm_f, collect(parameters(wrapper_f)))
+
             ret!(builder)
         end
 
-        call_function(llvm_f, Nothing, Tuple{arg_types...}, arg_exprs...)
+        call_function(wrapper_f, Nothing, Tuple{arg_types...}, arg_exprs...)
     end
 end