1  #include <stdlib.h>
       2  #include <stdio.h>
       3  #include <stddef.h>
       4  
       5  #include "libgccjit.h"
       6  
       7  #include "harness.h"
       8  
       9  #define BUFFER_SIZE (1024)
      10  
      11  char test_buffer[1024];
      12  
      13  void
      14  create_code (gcc_jit_context *ctxt, void *user_data)
      15  {
      16    /* Let's try to inject the equivalent of:
      17          void test_of_array_as_pointer (const char *name)
      18          {
      19              snprintf (test_buffer, sizeof (test_buffer),
      20  	              "hello %s", name);
      21          }
      22      */
      23    gcc_jit_type *void_type =
      24      gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_VOID);
      25    gcc_jit_type *const_char_ptr_type =
      26      gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_CONST_CHAR_PTR);
      27    gcc_jit_type *char_type =
      28      gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_CHAR);
      29    gcc_jit_type *char_ptr_type =
      30      gcc_jit_type_get_pointer (char_type);
      31    gcc_jit_type *int_type =
      32      gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_INT);
      33    gcc_jit_type *size_t_type =
      34      gcc_jit_context_get_type (ctxt, GCC_JIT_TYPE_SIZE_T);
      35    gcc_jit_type *buf_type =
      36      gcc_jit_context_new_array_type (ctxt, NULL, char_type, BUFFER_SIZE);
      37  
      38    /* extern int snprintf(char *str, size_t size, const char *format, ...); */
      39    gcc_jit_param *param_s =
      40      gcc_jit_context_new_param (ctxt, NULL, char_ptr_type, "s");
      41    gcc_jit_param *param_n =
      42      gcc_jit_context_new_param (ctxt, NULL, size_t_type, "n");
      43    gcc_jit_param *param_format =
      44      gcc_jit_context_new_param (ctxt, NULL, const_char_ptr_type, "format");
      45    gcc_jit_param *snprintf_params[3] = {param_s, param_n, param_format};
      46    gcc_jit_function *snprintf =
      47      gcc_jit_context_new_function (ctxt, NULL,
      48  				  GCC_JIT_FUNCTION_IMPORTED,
      49  				  int_type,
      50  				  "snprintf",
      51  				  3, snprintf_params,
      52  				  1);
      53  
      54    gcc_jit_param *param_name =
      55      gcc_jit_context_new_param (ctxt, NULL, const_char_ptr_type, "name");
      56    gcc_jit_function *test_fn =
      57      gcc_jit_context_new_function (ctxt, NULL,
      58  				  GCC_JIT_FUNCTION_EXPORTED,
      59  				  void_type,
      60  				  "test_of_array_as_pointer",
      61  				  1, &param_name,
      62  				  0);
      63  
      64    gcc_jit_lvalue *buffer =
      65      gcc_jit_context_new_global (ctxt, NULL, GCC_JIT_GLOBAL_IMPORTED,
      66  				buf_type, "test_buffer");
      67  
      68    gcc_jit_block *block = gcc_jit_function_new_block(test_fn, "entry");
      69  
      70    /* snprintf(buffer, sizeof(buffer), "hello %s", name); */
      71    gcc_jit_rvalue *args[4];
      72    args[0] = gcc_jit_context_new_cast (
      73      ctxt, NULL,
      74      /* Here's the difference with test-error-array-as-pointer.c: */
      75      gcc_jit_lvalue_get_address (buffer,
      76  				NULL),
      77      char_ptr_type);
      78    args[1] = gcc_jit_context_new_rvalue_from_int (ctxt,
      79  						 size_t_type,
      80  						 BUFFER_SIZE);
      81    args[2] = gcc_jit_context_new_string_literal (ctxt, "hello %s");
      82    args[3] = gcc_jit_param_as_rvalue (param_name);
      83  
      84    gcc_jit_block_add_eval (
      85      block, NULL,
      86      gcc_jit_context_new_call (ctxt, NULL, snprintf, 4, args));
      87    gcc_jit_block_end_with_void_return (block, NULL);
      88  }
      89  
      90  void
      91  verify_code (gcc_jit_context *ctxt, gcc_jit_result *result)
      92  {
      93    CHECK_NON_NULL (result);
      94  
      95    typedef void (*fn_type) (const char *);
      96    fn_type test_of_array_as_pointer =
      97      (fn_type)gcc_jit_result_get_code (result, "test_of_array_as_pointer");
      98    CHECK_NON_NULL (test_of_array_as_pointer);
      99  
     100    test_of_array_as_pointer ("world");
     101    CHECK_STRING_VALUE (test_buffer, "hello world");
     102  }