gcc/testsuite/jit.dg/test-accessing-struct.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3
   4 #include "libgccjit.h"
   5
   6 #include "harness.h"
   7
   8 struct foo
   9 {
  10   int x;
  11   int y;
  12   int z;
  13 };
  14
  15 void
  16 create_code (gcc_jit_context *ctxt, void *user_data)
  17 {
  18   /* Let's try to inject the equivalent of:
  19      void
  20      test_access (struct foo *f)
  21      {
  22         f->z = f->x * f->y;
  23      }
  24   */
  25   gcc_jit_type *void_type =
  26     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_VOID);
  27   gcc_jit_type *int_type =
  28     gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
  29   gcc_jit_field *x =
  30     gcc_jit_context_new_field (ctxt,
  31                                NULL,
  32                                int_type,
  33                                "x");
  34   gcc_jit_field *y =
  35     gcc_jit_context_new_field (ctxt,
  36                                NULL,
  37                                int_type,
  38                                "y");
  39   gcc_jit_field *z =
  40     gcc_jit_context_new_field (ctxt,
  41                                NULL,
  42                                int_type,
  43                                "z");
  44   gcc_jit_field *fields[] = {x, y, z};
  45   gcc_jit_struct *struct_type =
  46     gcc_jit_context_new_struct_type (ctxt, NULL, "foo", 3, fields);
  47   gcc_jit_type *ptr_type =
  48     gcc_jit_type_get_pointer (gcc_jit_struct_as_type (struct_type));
  49
  50   /* Build the test function.  */
  51   gcc_jit_param *param_f =
  52     gcc_jit_context_new_param (ctxt, NULL, ptr_type, "f");
  53   gcc_jit_function *test_fn =
  54     gcc_jit_context_new_function (ctxt, NULL,
  55                                   GCC_JIT_FUNCTION_EXPORTED,
  56                                   void_type,
  57                                   "test_access",
  58                                   1, &param_f,
  59                                   0);
  60
  61   /* f->x * f->y */
  62   gcc_jit_rvalue *sum =
  63     gcc_jit_context_new_binary_op (
  64       ctxt, NULL,
  65       GCC_JIT_BINARY_OP_MULT,
  66       int_type,
  67       gcc_jit_lvalue_as_rvalue (
  68         gcc_jit_rvalue_dereference_field (
  69           gcc_jit_param_as_rvalue (param_f),
  70           NULL,
  71           x)),
  72       gcc_jit_lvalue_as_rvalue (
  73         gcc_jit_rvalue_dereference_field (
  74         gcc_jit_param_as_rvalue (param_f),
  75         NULL,
  76         y)));
  77
  78   /* f->z = ... */
  79   gcc_jit_block *block = gcc_jit_function_new_block (test_fn, NULL);
  80   gcc_jit_block_add_assignment (
  81     block,
  82     NULL,
  83     gcc_jit_rvalue_dereference_field (
  84       gcc_jit_param_as_rvalue (param_f),
  85       NULL,
  86       z),
  87     sum);
  88   gcc_jit_block_end_with_void_return (block, NULL);
  89 }
  90
  91 void
  92 verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
  93 {
  94   typedef void (*fn_type) (struct foo *);
  95   CHECK_NON_NULL (result);
  96
  97   fn_type test_access =
  98     (fn_type)gcc_jit_result_get_code (result, "test_access");
  99   CHECK_NON_NULL (test_access);
 100
 101   struct foo tmp;
 102   tmp.x = 5;
 103   tmp.y = 7;
 104   tmp.z = 0;
 105
 106   /* Call the JIT-generated function.  */
 107   test_access (&tmp);
 108
 109   /* Verify that the code correctly modified the field "z".  */
 110   CHECK_VALUE (tmp.z, 35);
 111 }
 112