|
111
|
1 #include <stdlib.h>
|
|
|
2 #include <stdio.h>
|
|
|
3
|
|
|
4 #include "libgccjit.h"
|
|
|
5
|
|
|
6 #include "harness.h"
|
|
|
7
|
|
|
8 #ifdef __cplusplus
|
|
|
9 extern "C" {
|
|
|
10 #endif
|
|
|
11
|
|
|
12 extern int imported_global;
|
|
|
13
|
|
|
14 #ifdef __cplusplus
|
|
|
15 }
|
|
|
16 #endif
|
|
|
17
|
|
|
18 void
|
|
|
19 create_code (gcc_jit_context *ctxt, void *user_data)
|
|
|
20 {
|
|
|
21 /* Let's try to inject the equivalent of:
|
|
|
22
|
|
|
23 int exported_global;
|
|
|
24 extern int imported_global;
|
|
|
25 static int internal_global;
|
|
|
26
|
|
|
27 int
|
|
|
28 test_using_global (void)
|
|
|
29 {
|
|
|
30 exported_global += 1;
|
|
|
31 imported_global += 1;
|
|
|
32 internal_global += 1;
|
|
|
33 return internal_global;
|
|
|
34 }
|
|
|
35 */
|
|
|
36 gcc_jit_type *int_type =
|
|
|
37 gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
|
|
|
38
|
|
|
39 gcc_jit_lvalue *exported_global =
|
|
|
40 gcc_jit_context_new_global (ctxt,
|
|
|
41 NULL,
|
|
|
42 GCC_JIT_GLOBAL_EXPORTED,
|
|
|
43 int_type,
|
|
|
44 "exported_global");
|
|
|
45 gcc_jit_lvalue *imported_global =
|
|
|
46 gcc_jit_context_new_global (ctxt,
|
|
|
47 NULL,
|
|
|
48 GCC_JIT_GLOBAL_IMPORTED,
|
|
|
49 int_type,
|
|
|
50 "imported_global");
|
|
|
51 gcc_jit_lvalue *internal_global =
|
|
|
52 gcc_jit_context_new_global (ctxt,
|
|
|
53 NULL,
|
|
|
54 GCC_JIT_GLOBAL_INTERNAL,
|
|
|
55 int_type,
|
|
|
56 "internal_global");
|
|
|
57
|
|
|
58 /* Build the test_fn. */
|
|
|
59 gcc_jit_function *test_fn =
|
|
|
60 gcc_jit_context_new_function (ctxt, NULL,
|
|
|
61 GCC_JIT_FUNCTION_EXPORTED,
|
|
|
62 int_type,
|
|
|
63 "test_using_global",
|
|
|
64 0, NULL,
|
|
|
65 0);
|
|
|
66 gcc_jit_block *block = gcc_jit_function_new_block (test_fn, NULL);
|
|
|
67
|
|
|
68 gcc_jit_block_add_assignment_op (
|
|
|
69 block, NULL,
|
|
|
70 exported_global,
|
|
|
71 GCC_JIT_BINARY_OP_PLUS,
|
|
|
72 gcc_jit_context_one (ctxt, int_type));
|
|
|
73 gcc_jit_block_add_assignment_op (
|
|
|
74 block, NULL,
|
|
|
75 imported_global,
|
|
|
76 GCC_JIT_BINARY_OP_PLUS,
|
|
|
77 gcc_jit_context_one (ctxt, int_type));
|
|
|
78 gcc_jit_block_add_assignment_op (
|
|
|
79 block, NULL,
|
|
|
80 internal_global,
|
|
|
81 GCC_JIT_BINARY_OP_PLUS,
|
|
|
82 gcc_jit_context_one (ctxt, int_type));
|
|
|
83 gcc_jit_block_end_with_return (block,
|
|
|
84 NULL,
|
|
|
85 gcc_jit_lvalue_as_rvalue (internal_global));
|
|
|
86 }
|
|
|
87
|
|
|
88 int imported_global;
|
|
|
89
|
|
|
90 void
|
|
|
91 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
|
|
|
92 {
|
|
|
93 typedef int (*fn_type) (void);
|
|
|
94 CHECK_NON_NULL (result);
|
|
|
95
|
|
|
96 fn_type test_using_global =
|
|
|
97 (fn_type)gcc_jit_result_get_code (result, "test_using_global");
|
|
|
98 CHECK_NON_NULL (test_using_global);
|
|
|
99
|
|
|
100 /* The exported global should be visible. */
|
|
|
101 int *exported_global = (int *)gcc_jit_result_get_global (result, "exported_global");
|
|
|
102 CHECK_NON_NULL (exported_global);
|
|
|
103 /* ...and should be zero-initialized. */
|
|
|
104 CHECK_VALUE (*exported_global, 0);
|
|
|
105
|
|
|
106 /* Set some nonzero values. */
|
|
|
107 *exported_global = 11;
|
|
|
108 imported_global = 42;
|
|
|
109
|
|
|
110 /* The internal global shouldn't be visible. */
|
|
|
111 int *internal_global = (int *)gcc_jit_result_get_global (result, "internal_global");
|
|
|
112 CHECK_VALUE (internal_global, NULL);
|
|
|
113
|
|
|
114 /* Call the JIT-generated function. */
|
|
|
115 int call_count = test_using_global ();
|
|
|
116
|
|
|
117 /* Verify that it correctly modified imported_global and exported_global. */
|
|
|
118 CHECK_VALUE (*exported_global, 12);
|
|
|
119 CHECK_VALUE (imported_global, 43);
|
|
|
120 CHECK_VALUE (call_count, 1);
|
|
|
121
|
|
|
122 /* Try calling it again. */
|
|
|
123 call_count = test_using_global ();
|
|
|
124
|
|
|
125 /* Verify the new values. */
|
|
|
126 CHECK_VALUE (*exported_global, 13);
|
|
|
127 CHECK_VALUE (imported_global, 44);
|
|
|
128 CHECK_VALUE (call_count, 2);
|
|
|
129 }
|
|
|
130
|
