136 lines
3.7 KiB
C
136 lines
3.7 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "libgccjit.h"
|
|
|
|
#include "harness.h"
|
|
|
|
struct bar
|
|
{
|
|
int x;
|
|
int y;
|
|
};
|
|
|
|
void
|
|
create_code (gcc_jit_context *ctxt, void *user_data)
|
|
{
|
|
/* Let's try to inject the equivalent of:
|
|
|
|
int
|
|
test_reading (const struct bar *f)
|
|
{
|
|
return f->x * f->y;
|
|
}
|
|
|
|
int
|
|
test_writing ()
|
|
{
|
|
struct bar tmp;
|
|
tmp.x = 5;
|
|
tmp.y = 7;
|
|
return test_reading (&tmp);
|
|
}
|
|
*/
|
|
gcc_jit_type *int_type =
|
|
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
|
|
gcc_jit_field *x =
|
|
gcc_jit_context_new_field (ctxt,
|
|
NULL,
|
|
int_type,
|
|
"x");
|
|
gcc_jit_field *y =
|
|
gcc_jit_context_new_field (ctxt,
|
|
NULL,
|
|
int_type,
|
|
"y");
|
|
gcc_jit_field *fields[] = {x, y};
|
|
gcc_jit_type *struct_type =
|
|
gcc_jit_struct_as_type (
|
|
gcc_jit_context_new_struct_type (ctxt, NULL, "bar", 2, fields));
|
|
gcc_jit_type *const_struct_type = gcc_jit_type_get_const (struct_type);
|
|
gcc_jit_type *ptr_type = gcc_jit_type_get_pointer (const_struct_type);
|
|
|
|
/* Build "test_reading". */
|
|
gcc_jit_param *param_f =
|
|
gcc_jit_context_new_param (ctxt, NULL, ptr_type, "f");
|
|
gcc_jit_function *fn_test_reading =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_EXPORTED,
|
|
int_type,
|
|
"test_reading",
|
|
1, ¶m_f,
|
|
0);
|
|
|
|
/* return f->x * f->y; */
|
|
gcc_jit_block *reading_block = gcc_jit_function_new_block (fn_test_reading, NULL);
|
|
gcc_jit_block_end_with_return (
|
|
reading_block,
|
|
NULL,
|
|
gcc_jit_context_new_binary_op (
|
|
ctxt, NULL,
|
|
GCC_JIT_BINARY_OP_MULT,
|
|
int_type,
|
|
gcc_jit_lvalue_as_rvalue (
|
|
gcc_jit_rvalue_dereference_field (
|
|
gcc_jit_param_as_rvalue (param_f),
|
|
NULL,
|
|
x)),
|
|
gcc_jit_lvalue_as_rvalue (
|
|
gcc_jit_rvalue_dereference_field (
|
|
gcc_jit_param_as_rvalue (param_f),
|
|
NULL,
|
|
y))));
|
|
|
|
/* Build "test_writing". */
|
|
gcc_jit_function *fn_test_writing =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_EXPORTED,
|
|
int_type,
|
|
"test_writing",
|
|
0, NULL,
|
|
0);
|
|
|
|
/* struct bar tmp; */
|
|
gcc_jit_lvalue *local_tmp =
|
|
gcc_jit_function_new_local (fn_test_writing, NULL,
|
|
struct_type,
|
|
"tmp");
|
|
/* tmp.x = 5; */
|
|
gcc_jit_block *writing_block = gcc_jit_function_new_block (fn_test_writing, NULL);
|
|
gcc_jit_block_add_assignment (
|
|
writing_block, NULL,
|
|
gcc_jit_lvalue_access_field (local_tmp, NULL, x),
|
|
gcc_jit_context_new_rvalue_from_int (ctxt, int_type, 5));
|
|
|
|
/* tmp.y = 7; */
|
|
gcc_jit_block_add_assignment (
|
|
writing_block, NULL,
|
|
gcc_jit_lvalue_access_field (local_tmp, NULL, y),
|
|
gcc_jit_context_new_rvalue_from_int (ctxt, int_type, 7));
|
|
|
|
/* return test_reading (&tmp); */
|
|
gcc_jit_rvalue *arg = gcc_jit_lvalue_get_address (local_tmp, NULL);
|
|
gcc_jit_block_end_with_return (
|
|
writing_block,
|
|
NULL,
|
|
gcc_jit_context_new_call (
|
|
ctxt, NULL,
|
|
fn_test_reading,
|
|
1, &arg));
|
|
}
|
|
|
|
void
|
|
verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
|
|
{
|
|
typedef int (*fn_type) (void);
|
|
CHECK_NON_NULL (result);
|
|
|
|
fn_type test_writing =
|
|
(fn_type)gcc_jit_result_get_code (result, "test_writing");
|
|
CHECK_NON_NULL (test_writing);
|
|
|
|
/* Verify that the code correctly returns the product of the fields. */
|
|
CHECK_VALUE (test_writing (), 35);
|
|
}
|
|
|