233 lines
6.7 KiB
C
233 lines
6.7 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#include "libgccjit.h"
|
|
|
|
#include "harness.h"
|
|
|
|
struct s2
|
|
{
|
|
char x __attribute__ ((aligned (2)));
|
|
char y __attribute__ ((aligned (2)));
|
|
};
|
|
|
|
struct s4
|
|
{
|
|
char x __attribute__ ((aligned (4)));
|
|
char y __attribute__ ((aligned (4)));
|
|
};
|
|
|
|
struct s8
|
|
{
|
|
char x __attribute__ ((aligned (8)));
|
|
char y __attribute__ ((aligned (8)));
|
|
};
|
|
|
|
struct s16
|
|
{
|
|
char x __attribute__ ((aligned (16)));
|
|
char y __attribute__ ((aligned (16)));
|
|
};
|
|
|
|
struct s32
|
|
{
|
|
char x __attribute__ ((aligned (32)));
|
|
char y __attribute__ ((aligned (32)));
|
|
};
|
|
|
|
struct s64
|
|
{
|
|
char x __attribute__ ((aligned (64)));
|
|
char y __attribute__ ((aligned (64)));
|
|
};
|
|
|
|
struct s128
|
|
{
|
|
char x __attribute__ ((aligned (128)));
|
|
char y __attribute__ ((aligned (128)));
|
|
};
|
|
|
|
static void
|
|
create_aligned_code (gcc_jit_context *ctxt, const char *struct_name,
|
|
unsigned int alignment, const char *reader_fn_name,
|
|
const char *writer_fn_name)
|
|
{
|
|
/* Let's try to inject the equivalent of:
|
|
|
|
char
|
|
READER_FN_NAME (const struct STRUCT_NAME *f)
|
|
{
|
|
return f->x * f->y;
|
|
}
|
|
|
|
char
|
|
WRITER_FN_NAME (struct STRUCT_NAME *g)
|
|
{
|
|
g->x = 5;
|
|
g->y = 7;
|
|
return READER_FN_NAME (g);
|
|
}
|
|
*/
|
|
gcc_jit_type *char_type =
|
|
gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_CHAR);
|
|
gcc_jit_type *aligned_char_type =
|
|
gcc_jit_type_get_aligned (char_type, alignment);
|
|
gcc_jit_field *x =
|
|
gcc_jit_context_new_field (ctxt,
|
|
NULL,
|
|
aligned_char_type,
|
|
"x");
|
|
gcc_jit_field *y =
|
|
gcc_jit_context_new_field (ctxt,
|
|
NULL,
|
|
aligned_char_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, struct_name, 2, fields));
|
|
gcc_jit_type *const_struct_type = gcc_jit_type_get_const (struct_type);
|
|
gcc_jit_type *const_ptr_type = gcc_jit_type_get_pointer (const_struct_type);
|
|
|
|
/* Build the reader fn. */
|
|
gcc_jit_param *param_f =
|
|
gcc_jit_context_new_param (ctxt, NULL, const_ptr_type, "f");
|
|
gcc_jit_function *fn_test_reading =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_EXPORTED,
|
|
char_type,
|
|
reader_fn_name,
|
|
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,
|
|
char_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 the writer fn. */
|
|
gcc_jit_type *ptr_type = gcc_jit_type_get_pointer (struct_type);
|
|
gcc_jit_param *param_g =
|
|
gcc_jit_context_new_param (ctxt, NULL, ptr_type, "g");
|
|
gcc_jit_function *fn_test_writing =
|
|
gcc_jit_context_new_function (ctxt, NULL,
|
|
GCC_JIT_FUNCTION_EXPORTED,
|
|
char_type,
|
|
writer_fn_name,
|
|
1, ¶m_g,
|
|
0);
|
|
|
|
/* g->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_rvalue_dereference_field (gcc_jit_param_as_rvalue (param_g),
|
|
NULL, x),
|
|
gcc_jit_context_new_rvalue_from_int (ctxt, char_type, 5));
|
|
|
|
/* g->y = 7; */
|
|
gcc_jit_block_add_assignment (
|
|
writing_block, NULL,
|
|
gcc_jit_rvalue_dereference_field (gcc_jit_param_as_rvalue (param_g),
|
|
NULL, y),
|
|
gcc_jit_context_new_rvalue_from_int (ctxt, char_type, 7));
|
|
|
|
/* return READER_FN_NAME (g); */
|
|
gcc_jit_rvalue *arg = gcc_jit_param_as_rvalue (param_g);
|
|
gcc_jit_block_end_with_return (
|
|
writing_block,
|
|
NULL,
|
|
gcc_jit_context_new_call (
|
|
ctxt, NULL,
|
|
fn_test_reading,
|
|
1, &arg));
|
|
}
|
|
|
|
/* Implement a verifier function for a given struct. */
|
|
|
|
#define IMPL_VERIFY_ALIGNED_CODE(TYPENAME) \
|
|
static void \
|
|
verify_aligned_code_ ##TYPENAME (gcc_jit_context *ctxt, \
|
|
gcc_jit_result *result, \
|
|
const char *writer_fn_name) \
|
|
{ \
|
|
typedef char (*fn_type) (struct TYPENAME *); \
|
|
CHECK_NON_NULL (result); \
|
|
\
|
|
struct TYPENAME tmp; \
|
|
memset (&tmp, 0xac, sizeof (tmp)); \
|
|
\
|
|
fn_type test_writing = \
|
|
(fn_type)gcc_jit_result_get_code (result, writer_fn_name); \
|
|
CHECK_NON_NULL (test_writing); \
|
|
\
|
|
/* Verify that the code correctly returns the product of the fields. */ \
|
|
CHECK_VALUE (test_writing (&tmp), 35); \
|
|
\
|
|
/* Verify the we can read the values of the fields, and thus that the \
|
|
struct layout agrees with that of the C frontend. */ \
|
|
CHECK_VALUE (tmp.x, 5); \
|
|
CHECK_VALUE (tmp.y, 7); \
|
|
}
|
|
|
|
void
|
|
create_code (gcc_jit_context *ctxt, void *user_data)
|
|
{
|
|
create_aligned_code (ctxt, "s2", 2, "test_aligned_reading_s2",
|
|
"test_aligned_writing_s2");
|
|
create_aligned_code (ctxt, "s4", 4, "test_aligned_reading_s4",
|
|
"test_aligned_writing_s4");
|
|
create_aligned_code (ctxt, "s8", 8, "test_aligned_reading_s8",
|
|
"test_aligned_writing_s8");
|
|
create_aligned_code (ctxt, "s16", 16, "test_aligned_reading_s16",
|
|
"test_aligned_writing_s16");
|
|
create_aligned_code (ctxt, "s32", 32, "test_aligned_reading_s32",
|
|
"test_aligned_writing_s32");
|
|
create_aligned_code (ctxt, "s64", 64, "test_aligned_reading_s64",
|
|
"test_aligned_writing_s64");
|
|
create_aligned_code (ctxt, "s128", 128, "test_aligned_reading_s128",
|
|
"test_aligned_writing_s128");
|
|
}
|
|
|
|
IMPL_VERIFY_ALIGNED_CODE(s2)
|
|
IMPL_VERIFY_ALIGNED_CODE(s4)
|
|
IMPL_VERIFY_ALIGNED_CODE(s8)
|
|
IMPL_VERIFY_ALIGNED_CODE(s16)
|
|
IMPL_VERIFY_ALIGNED_CODE(s32)
|
|
IMPL_VERIFY_ALIGNED_CODE(s64)
|
|
IMPL_VERIFY_ALIGNED_CODE(s128)
|
|
|
|
void
|
|
verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
|
|
{
|
|
verify_aligned_code_s2 (ctxt, result,
|
|
"test_aligned_writing_s2");
|
|
verify_aligned_code_s4 (ctxt, result,
|
|
"test_aligned_writing_s4");
|
|
verify_aligned_code_s8 (ctxt, result,
|
|
"test_aligned_writing_s8");
|
|
verify_aligned_code_s16 (ctxt, result,
|
|
"test_aligned_writing_s16");
|
|
verify_aligned_code_s32 (ctxt, result,
|
|
"test_aligned_writing_s32");
|
|
verify_aligned_code_s64 (ctxt, result,
|
|
"test_aligned_writing_s64");
|
|
verify_aligned_code_s128 (ctxt, result,
|
|
"test_aligned_writing_s128");
|
|
}
|