gcc/testsuite/jit.dg/test-returning-function-ptr.c

#include <stdlib.h>
#include <stdio.h>

#include "libgccjit.h"

#include "harness.h"

#ifdef __cplusplus
extern "C" {
#endif

  extern void
  internally_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 internally_called_function (int i, int j, int k);

     static void
     internal_test_caller (int a)
     {
        internally_called_function (a * 3, a * 4, a * 5);
     }

     typedef void (*fn_ptr_type) (int);

     fn_ptr_type
     get_test_caller (void)
     {
       // Verify that we can assign function pointers to variables
       fn_ptr_type p;
       p = internal_test_caller;
       return p;
     }
  */
  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,
				  "internally_called_function",
				  3, params,
				  0);

  /* Build the test_caller fn.  */
  gcc_jit_param *param_a =
    gcc_jit_context_new_param (ctxt, NULL, int_type, "a");
  gcc_jit_function *test_caller =
    gcc_jit_context_new_function (ctxt, NULL,
				  GCC_JIT_FUNCTION_EXPORTED,
				  void_type,
				  "internal_test_caller",
				  1, &param_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_caller, 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);

  gcc_jit_type *fn_ptr_type
    = gcc_jit_context_new_function_ptr_type (ctxt, NULL,
					     void_type,
					     1, &int_type,
					     0);

  /* Build the get_test_caller fn.  */
  gcc_jit_function *get_test_caller =
    gcc_jit_context_new_function (ctxt, NULL,
				  GCC_JIT_FUNCTION_EXPORTED,
				  fn_ptr_type,
				  "get_test_caller",
				  0, NULL,
				  0);
  block = gcc_jit_function_new_block (get_test_caller, NULL);

  /* fn_ptr_type p; */
  gcc_jit_lvalue *local_p
    = gcc_jit_function_new_local (get_test_caller, NULL,
				  fn_ptr_type, "p");

  /* p = internal_test_caller; */
  gcc_jit_block_add_assignment (block, NULL,
				local_p,
				gcc_jit_function_get_address (test_caller,
							      NULL));

  /* return p; */
  gcc_jit_block_end_with_return (block, NULL,
				 gcc_jit_lvalue_as_rvalue (local_p));
}

static int called_with[3];

extern void
internally_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 (*test_caller_type) (int);
  typedef test_caller_type (*get_test_caller_type) (void);
  CHECK_NON_NULL (result);

  get_test_caller_type get_test_caller =
    (get_test_caller_type)gcc_jit_result_get_code (result, "get_test_caller");
  CHECK_NON_NULL (get_test_caller);

  test_caller_type test_caller = (test_caller_type)get_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 "internally_called_function".  */
  CHECK_VALUE (called_with[0], 15);
  CHECK_VALUE (called_with[1], 20);
  CHECK_VALUE (called_with[2], 25);
}