Showing the changes to tests/libtcc_test.c (1 of 16 files). Show all
Unified · Split
| @@ -0,0 +1,84 @@ | ||
| 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 | +#include <string.h> | |
| 9 | + | |
| 10 | +#include "libtcc.h" | |
| 11 | + | |
| 12 | +/* this function is called by the generated code */ | |
| 13 | +int add(int a, int b) | |
| 14 | +{ | |
| 15 | + return a + b; | |
| 16 | +} | |
| 17 | + | |
| 18 | +char my_program[] = | |
| 19 | +"int fib(int n)\n" | |
| 20 | +"{\n" | |
| 21 | +" if (n <= 2)\n" | |
| 22 | +" return 1;\n" | |
| 23 | +" else\n" | |
| 24 | +" return fib(n-1) + fib(n-2);\n" | |
| 25 | +"}\n" | |
| 26 | +"\n" | |
| 27 | +"int foo(int n)\n" | |
| 28 | +"{\n" | |
| 29 | +" printf(\"Hello World!\\n\");\n" | |
| 30 | +" printf(\"fib(%d) = %d\\n\", n, fib(n));\n" | |
| 31 | +" printf(\"add(%d, %d) = %d\\n\", n, 2 * n, add(n, 2 * n));\n" | |
| 32 | +" return 0;\n" | |
| 33 | +"}\n"; | |
| 34 | + | |
| 35 | +int main(int argc, char **argv) | |
| 36 | +{ | |
| 37 | + TCCState *s; | |
| 38 | + int (*func)(int); | |
| 39 | + void *mem; | |
| 40 | + int size; | |
| 41 | + | |
| 42 | + s = tcc_new(); | |
| 43 | + if (!s) { | |
| 44 | + fprintf(stderr, "Could not create tcc state\n"); | |
| 45 | + exit(1); | |
| 46 | + } | |
| 47 | + | |
| 48 | + /* if tcclib.h and libtcc1.a are not installed, where can we find them */ | |
| 49 | + if (argc == 2 && !memcmp(argv[1], "lib_path=",9)) | |
| 50 | + tcc_set_lib_path(s, argv[1]+9); | |
| 51 | + | |
| 52 | + /* MUST BE CALLED before any compilation */ | |
| 53 | + tcc_set_output_type(s, TCC_OUTPUT_MEMORY); | |
| 54 | + | |
| 55 | + if (tcc_compile_string(s, my_program) == -1) | |
| 56 | + return 1; | |
| 57 | + | |
| 58 | + /* as a test, we add a symbol that the compiled program can use. | |
| 59 | + You may also open a dll with tcc_add_dll() and use symbols from that */ | |
| 60 | + tcc_add_symbol(s, "add", add); | |
| 61 | + | |
| 62 | + /* get needed size of the code */ | |
| 63 | + size = tcc_relocate(s, NULL); | |
| 64 | + if (size == -1) | |
| 65 | + return 1; | |
| 66 | + | |
| 67 | + /* allocate memory and copy the code into it */ | |
| 68 | + mem = malloc(size); | |
| 69 | + tcc_relocate(s, mem); | |
| 70 | + | |
| 71 | + /* get entry symbol */ | |
| 72 | + func = tcc_get_symbol(s, "foo"); | |
| 73 | + if (!func) | |
| 74 | + return 1; | |
| 75 | + | |
| 76 | + /* delete the state */ | |
| 77 | + tcc_delete(s); | |
| 78 | + | |
| 79 | + /* run the code */ | |
| 80 | + func(32); | |
| 81 | + | |
| 82 | + free(mem); | |
| 83 | + return 0; | |
| 84 | +} | |