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