libtcc_test.c

   1 /*
   2  * Simple Test program for libtcc
   3  *
   4  * libtcc can be useful to use tcc as a "backend" for a code generator.
   5  */
   6 #include <stdlib.h>
   7 #include <stdio.h>
   8
   9 #include "libtcc.h"
  10
  11 /* this function is called by the generated code */
  12 int add(int a, int b)
  13 {
  14     return a + b;
  15 }
  16
  17 char my_program[] =
  18 "int fib(int n)\n"
  19 "{\n"
  20 "    if (n <= 2)\n"
  21 "        return 1;\n"
  22 "    else\n"
  23 "        return fib(n-1) + fib(n-2);\n"
  24 "}\n"
  25 "\n"
  26 "int foo(int n)\n"
  27 "{\n"
  28 "    printf(\"Hello World!\\n\");\n"
  29 "    printf(\"fib(%d) = %d\\n\", n, fib(n));\n"
  30 "    printf(\"add(%d, %d) = %d\\n\", n, 2 * n, add(n, 2 * n));\n"
  31 "    return 0;\n"
  32 "}\n";
  33
  34 int main(int argc, char **argv)
  35 {
  36     TCCState *s;
  37     int (*func)(int);
  38     void *mem;
  39     int size;
  40
  41     s = tcc_new();
  42     if (!s) {
  43         fprintf(stderr, "Could not create tcc state\n");
  44         exit(1);
  45     }
  46
  47     /* MUST BE CALLED before any compilation or file loading */
  48     tcc_set_output_type(s, TCC_OUTPUT_MEMORY);
  49
  50     tcc_compile_string(s, my_program);
  51
  52     /* as a test, we add a symbol that the compiled program can use.
  53        You may also open a dll with tcc_add_dll() and use symbols from that */
  54     tcc_add_symbol(s, "add", add);
  55
  56     /* get needed size of the code */
  57     size = tcc_relocate(s, NULL);
  58     if (size == -1)
  59         return 1;
  60
  61     /* allocate memory and copy the code into it */
  62     mem = malloc(size);
  63     tcc_relocate(s, mem);
  64
  65     /* get entry symbol */
  66     func = tcc_get_symbol(s, "foo");
  67     if (!func)
  68         return 1;
  69
  70     /* delete the state */
  71     tcc_delete(s);
  72
  73     /* run the code */
  74     func(32);
  75
  76     free(mem);
  77     return 0;
  78 }