mirror of
https://github.com/autc04/Retro68.git
synced 2024-06-28 11:30:19 +00:00
119 lines
3.0 KiB
C
119 lines
3.0 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "libgccjit.h"
|
|
|
|
#include "harness.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
extern void
|
|
called_function (int i, int j, int k);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
void
|
|
create_code (gcc_jit_context *ctxt, void *user_data)
|
|
{
|
|
/* Let's try to inject the equivalent of:
|
|
extern void called_function (int i, int j, int k);
|
|
|
|
void
|
|
test_caller (int a)
|
|
{
|
|
called_function (a * 3, a * 4, a * 5);
|
|
}
|
|
*/
|
|
int i;
|
|
gcc_jit_type *void_type =
|
|
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_VOID);
|
|
gcc_jit_type *int_type =
|
|
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
|
|
|
|
/* Declare the imported function. */
|
|
gcc_jit_param *params[3];
|
|
params[0] =
|
|
gcc_jit_context_new_param (ctxt, NULL, int_type, "i");
|
|
params[1] =
|
|
gcc_jit_context_new_param (ctxt, NULL, int_type, "j");
|
|
params[2] =
|
|
gcc_jit_context_new_param (ctxt, NULL, int_type, "k");
|
|
gcc_jit_function *called_fn =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_IMPORTED,
|
|
void_type,
|
|
"called_function",
|
|
3, params,
|
|
0);
|
|
|
|
/* Build the test_fn. */
|
|
gcc_jit_param *param_a =
|
|
gcc_jit_context_new_param (ctxt, NULL, int_type, "a");
|
|
gcc_jit_function *test_fn =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_EXPORTED,
|
|
void_type,
|
|
"test_caller",
|
|
1, ¶m_a,
|
|
0);
|
|
/* "a * 3, a * 4, a * 5" */
|
|
gcc_jit_rvalue *args[3];
|
|
for (i = 0; i < 3; i++)
|
|
args[i] =
|
|
gcc_jit_context_new_binary_op (
|
|
ctxt, NULL,
|
|
GCC_JIT_BINARY_OP_MULT,
|
|
int_type,
|
|
gcc_jit_param_as_rvalue (param_a),
|
|
gcc_jit_context_new_rvalue_from_int (
|
|
ctxt,
|
|
int_type,
|
|
(i + 3) ));
|
|
gcc_jit_block *block = gcc_jit_function_new_block (test_fn, NULL);
|
|
gcc_jit_block_add_eval (
|
|
block, NULL,
|
|
gcc_jit_context_new_call (ctxt,
|
|
NULL,
|
|
called_fn,
|
|
3, args));
|
|
gcc_jit_block_end_with_void_return (block, NULL);
|
|
}
|
|
|
|
static int called_with[3];
|
|
|
|
extern void
|
|
called_function (int i, int j, int k)
|
|
{
|
|
called_with[0] = i;
|
|
called_with[1] = j;
|
|
called_with[2] = k;
|
|
}
|
|
|
|
void
|
|
verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
|
|
{
|
|
typedef void (*fn_type) (int);
|
|
CHECK_NON_NULL (result);
|
|
|
|
fn_type test_caller =
|
|
(fn_type)gcc_jit_result_get_code (result, "test_caller");
|
|
CHECK_NON_NULL (test_caller);
|
|
|
|
called_with[0] = 0;
|
|
called_with[1] = 0;
|
|
called_with[2] = 0;
|
|
|
|
/* Call the JIT-generated function. */
|
|
test_caller (5);
|
|
|
|
/* Verify that it correctly called "called_function". */
|
|
CHECK_VALUE (called_with[0], 15);
|
|
CHECK_VALUE (called_with[1], 20);
|
|
CHECK_VALUE (called_with[2], 25);
|
|
}
|
|
|