gcc/testsuite/jit.dg/test-array-as-pointer.c

   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 }