1 #include <stdlib.h>
2 #include <stdio.h>
3
4 #include "libgccjit.h"
5
6 #include "harness.h"
7
8 void
create_code(gcc_jit_context * ctxt,void * user_data)9 create_code (gcc_jit_context *ctxt, void *user_data)
10 {
11 /* Let's try to inject the equivalent of:
12
13 void
14 test_calling_function_ptr (void (*fn_ptr) (int, int, int) fn_ptr,
15 int a)
16 {
17 fn_ptr (a * 3, a * 4, a * 5);
18 }
19 */
20
21 int i;
22 gcc_jit_type *void_type =
23 gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_VOID);
24 gcc_jit_type *int_type =
25 gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
26
27 /* Build the function ptr type. */
28 gcc_jit_type *param_types[3];
29 param_types[0] = int_type;
30 param_types[1] = int_type;
31 param_types[2] = int_type;
32
33 gcc_jit_type *fn_ptr_type =
34 gcc_jit_context_new_function_ptr_type (ctxt, NULL,
35 void_type,
36 3, param_types, 0);
37
38 /* Ensure that function ptr types have sane debug strings. */
39
40 CHECK_STRING_VALUE (
41 gcc_jit_object_get_debug_string (gcc_jit_type_as_object (fn_ptr_type)),
42 "void (*) (int, int, int)");
43
44 /* Build the test_fn. */
45 gcc_jit_param *param_fn_ptr =
46 gcc_jit_context_new_param (ctxt, NULL, fn_ptr_type, "fn_ptr");
47 gcc_jit_param *param_a =
48 gcc_jit_context_new_param (ctxt, NULL, int_type, "a");
49
50 gcc_jit_param *params[2];
51 params[0] = param_fn_ptr;
52 params[1] = param_a;
53
54 gcc_jit_function *test_fn =
55 gcc_jit_context_new_function (ctxt, NULL,
56 GCC_JIT_FUNCTION_EXPORTED,
57 void_type,
58 "test_calling_function_ptr",
59 2, params,
60 0);
61 /* "a * 3, a * 4, a * 5" */
62 gcc_jit_rvalue *args[3];
63 for (i = 0; i < 3; i++)
64 args[i] =
65 gcc_jit_context_new_binary_op (
66 ctxt, NULL,
67 GCC_JIT_BINARY_OP_MULT,
68 int_type,
69 gcc_jit_param_as_rvalue (param_a),
70 gcc_jit_context_new_rvalue_from_int (
71 ctxt,
72 int_type,
73 (i + 3) ));
74 gcc_jit_block *block = gcc_jit_function_new_block (test_fn, NULL);
75 gcc_jit_block_add_eval (
76 block, NULL,
77 gcc_jit_context_new_call_through_ptr (
78 ctxt,
79 NULL,
80 gcc_jit_param_as_rvalue (param_fn_ptr),
81 3, args));
82 gcc_jit_block_end_with_void_return (block, NULL);
83 }
84
85 static int called_through_ptr_with[3];
86
87 static void
function_called_through_fn_ptr(int i,int j,int k)88 function_called_through_fn_ptr (int i, int j, int k)
89 {
90 called_through_ptr_with[0] = i;
91 called_through_ptr_with[1] = j;
92 called_through_ptr_with[2] = k;
93 }
94
95 void
verify_code(gcc_jit_context * ctxt,gcc_jit_result * result)96 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
97 {
98 typedef void (*fn_type) (void (*fn_ptr) (int, int, int),
99 int);
100 CHECK_NON_NULL (result);
101
102 fn_type test_caller =
103 (fn_type)gcc_jit_result_get_code (result, "test_calling_function_ptr");
104 CHECK_NON_NULL (test_caller);
105
106 called_through_ptr_with[0] = 0;
107 called_through_ptr_with[1] = 0;
108 called_through_ptr_with[2] = 0;
109
110 /* Call the JIT-generated function. */
111 test_caller (function_called_through_fn_ptr, 5);
112
113 /* Verify that it correctly called "function_called_through_fn_ptr". */
114 CHECK_VALUE (called_through_ptr_with[0], 15);
115 CHECK_VALUE (called_through_ptr_with[1], 20);
116 CHECK_VALUE (called_through_ptr_with[2], 25);
117 }
118
119