tests/abitest.c

   1 #include <libtcc.h>
   2 #include <stdlib.h>
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <stdarg.h>
   6
   7 // MinGW has 80-bit rather than 64-bit long double which isn't compatible with TCC or MSVC
   8 #if defined(_WIN32) && defined(__GNUC__)
   9 #define LONG_DOUBLE double
  10 #define LONG_DOUBLE_LITERAL(x) x
  11 #else
  12 #define LONG_DOUBLE long double
  13 #define LONG_DOUBLE_LITERAL(x) x ## L
  14 #endif
  15
  16 static int g_argc;
  17 static char **g_argv;
  18
  19 static void set_options(TCCState *s, int argc, char **argv)
  20 {
  21     int i;
  22     for (i = 1; i < argc; ++i) {
  23         char *a = argv[i];
  24         if (a[0] == '-') {
  25             if (a[1] == 'B')
  26                 tcc_set_lib_path(s, a+2);
  27             else if (a[1] == 'I')
  28                 tcc_add_include_path(s, a+2);
  29             else if (a[1] == 'L')
  30                 tcc_add_library_path(s, a+2);
  31         }
  32     }
  33 }
  34
  35 typedef int (*callback_type) (void*);
  36
  37 /*
  38  * Compile source code and call a callback with a pointer to the symbol "f".
  39  */
  40 static int run_callback(const char *src, callback_type callback) {
  41   TCCState *s;
  42   int result;
  43   void *ptr;
  44
  45   s = tcc_new();
  46   if (!s)
  47     return -1;
  48
  49   set_options(s, g_argc, g_argv);
  50
  51   if (tcc_set_output_type(s, TCC_OUTPUT_MEMORY) == -1)
  52     return -1;
  53   if (tcc_compile_string(s, src) == -1)
  54     return -1;
  55   if (tcc_relocate(s, TCC_RELOCATE_AUTO) == -1)
  56     return -1;
  57
  58   ptr = tcc_get_symbol(s, "f");
  59   if (!ptr)
  60     return -1;
  61   result = callback(ptr);
  62
  63   tcc_delete(s);
  64
  65   return result;
  66 }
  67
  68 #define STR2(x) #x
  69 #define STR(x) STR2(x)
  70
  71 #define RET_PRIMITIVE_TEST(name, type, val) \
  72   static int ret_ ## name ## _test_callback(void *ptr) { \
  73     type (*callback) (type) = (type(*)(type))ptr; \
  74     type x = val; \
  75     type y = callback(x); \
  76     return (y == x+x) ? 0 : -1; \
  77   } \
  78   \
  79   static int ret_ ## name ## _test(void) { \
  80     const char *src = STR(type) " f(" STR(type) " x) {return x+x;}"; \
  81     return run_callback(src, ret_ ## name ## _test_callback); \
  82   }
  83
  84 RET_PRIMITIVE_TEST(int, int, 70000)
  85 RET_PRIMITIVE_TEST(longlong, long long, 4333369356528LL)
  86 RET_PRIMITIVE_TEST(float, float, 63.0)
  87 RET_PRIMITIVE_TEST(double, double, 14789798.0)
  88 RET_PRIMITIVE_TEST(longdouble, LONG_DOUBLE, LONG_DOUBLE_LITERAL(378943892.0))
  89
  90 /*
  91  * ret_2float_test:
  92  *
  93  * On x86-64, a struct with 2 floats should be packed into a single
  94  * SSE register (VT_DOUBLE is used for this purpose).
  95  */
  96 typedef struct ret_2float_test_type_s {float x, y;} ret_2float_test_type;
  97 typedef ret_2float_test_type (*ret_2float_test_function_type) (ret_2float_test_type);
  98
  99 static int ret_2float_test_callback(void *ptr) {
 100   ret_2float_test_function_type f = (ret_2float_test_function_type)ptr;
 101   ret_2float_test_type a = {10, 35};
 102   ret_2float_test_type r;
 103   r = f(a);
 104   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 105 }
 106
 107 static int ret_2float_test(void) {
 108   const char *src =
 109   "typedef struct ret_2float_test_type_s {float x, y;} ret_2float_test_type;"
 110   "ret_2float_test_type f(ret_2float_test_type a) {\n"
 111   "  ret_2float_test_type r = {a.x*5, a.y*3};\n"
 112   "  return r;\n"
 113   "}\n";
 114
 115   return run_callback(src, ret_2float_test_callback);
 116 }
 117
 118 /*
 119  * ret_2double_test:
 120  *
 121  * On x86-64, a struct with 2 doubles should be passed in two SSE
 122  * registers.
 123  */
 124 typedef struct ret_2double_test_type_s {double x, y;} ret_2double_test_type;
 125 typedef ret_2double_test_type (*ret_2double_test_function_type) (ret_2double_test_type);
 126
 127 static int ret_2double_test_callback(void *ptr) {
 128   ret_2double_test_function_type f = (ret_2double_test_function_type)ptr;
 129   ret_2double_test_type a = {10, 35};
 130   ret_2double_test_type r;
 131   r = f(a);
 132   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 133 }
 134
 135 static int ret_2double_test(void) {
 136   const char *src =
 137   "typedef struct ret_2double_test_type_s {double x, y;} ret_2double_test_type;"
 138   "ret_2double_test_type f(ret_2double_test_type a) {\n"
 139   "  ret_2double_test_type r = {a.x*5, a.y*3};\n"
 140   "  return r;\n"
 141   "}\n";
 142
 143   return run_callback(src, ret_2double_test_callback);
 144 }
 145
 146 /*
 147  * ret_8plus2double_test:
 148  *
 149  * This catches a corner case in the x86_64 ABI code: the first 7
 150  * arguments fit into registers, the 8th doesn't, but the 9th argument
 151  * fits into the 8th XMM register.
 152  *
 153  * Note that the purpose of the 10th argument is to avoid a situation
 154  * in which gcc would accidentally put the double at the right
 155  * address, thus causing a success message even though TCC actually
 156  * generated incorrect code.
 157  */
 158 typedef ret_2double_test_type (*ret_8plus2double_test_function_type) (double, double, double, double, double, double, double, ret_2double_test_type, double, double);
 159
 160 static int ret_8plus2double_test_callback(void *ptr) {
 161   ret_8plus2double_test_function_type f = (ret_8plus2double_test_function_type)ptr;
 162   ret_2double_test_type a = {10, 35};
 163   ret_2double_test_type r;
 164   r = f(0, 0, 0, 0, 0, 0, 0, a, 37, 38);
 165   return ((r.x == 37) && (r.y == 37)) ? 0 : -1;
 166 }
 167
 168 static int ret_8plus2double_test(void) {
 169   const char *src =
 170   "typedef struct ret_2double_test_type_s {double x, y;} ret_2double_test_type;"
 171   "ret_2double_test_type f(double x1, double x2, double x3, double x4, double x5, double x6, double x7, ret_2double_test_type a, double x8, double x9) {\n"
 172   "  ret_2double_test_type r = { x8, x8 };\n"
 173   "  return r;\n"
 174   "}\n";
 175
 176   return run_callback(src, ret_8plus2double_test_callback);
 177 }
 178
 179 /*
 180  * ret_mixed_test:
 181  *
 182  * On x86-64, a struct with a double and a 64-bit integer should be
 183  * passed in one SSE register and one integer register.
 184  */
 185 typedef struct ret_mixed_test_type_s {double x; long long y;} ret_mixed_test_type;
 186 typedef ret_mixed_test_type (*ret_mixed_test_function_type) (ret_mixed_test_type);
 187
 188 static int ret_mixed_test_callback(void *ptr) {
 189   ret_mixed_test_function_type f = (ret_mixed_test_function_type)ptr;
 190   ret_mixed_test_type a = {10, 35};
 191   ret_mixed_test_type r;
 192   r = f(a);
 193   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 194 }
 195
 196 static int ret_mixed_test(void) {
 197   const char *src =
 198   "typedef struct ret_mixed_test_type_s {double x; long long y;} ret_mixed_test_type;"
 199   "ret_mixed_test_type f(ret_mixed_test_type a) {\n"
 200   "  ret_mixed_test_type r = {a.x*5, a.y*3};\n"
 201   "  return r;\n"
 202   "}\n";
 203
 204   return run_callback(src, ret_mixed_test_callback);
 205 }
 206
 207 /*
 208  * ret_mixed2_test:
 209  *
 210  * On x86-64, a struct with two floats and two 32-bit integers should
 211  * be passed in one SSE register and one integer register.
 212  */
 213 typedef struct ret_mixed2_test_type_s {float x,x2; int y,y2;} ret_mixed2_test_type;
 214 typedef ret_mixed2_test_type (*ret_mixed2_test_function_type) (ret_mixed2_test_type);
 215
 216 static int ret_mixed2_test_callback(void *ptr) {
 217   ret_mixed2_test_function_type f = (ret_mixed2_test_function_type)ptr;
 218   ret_mixed2_test_type a = {10, 5, 35, 7 };
 219   ret_mixed2_test_type r;
 220   r = f(a);
 221   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 222 }
 223
 224 static int ret_mixed2_test(void) {
 225   const char *src =
 226   "typedef struct ret_mixed2_test_type_s {float x, x2; int y,y2;} ret_mixed2_test_type;"
 227   "ret_mixed2_test_type f(ret_mixed2_test_type a) {\n"
 228   "  ret_mixed2_test_type r = {a.x*5, 0, a.y*3, 0};\n"
 229   "  return r;\n"
 230   "}\n";
 231
 232   return run_callback(src, ret_mixed2_test_callback);
 233 }
 234
 235 /*
 236  * ret_mixed3_test:
 237  *
 238  * On x86-64, this struct should be passed in two integer registers.
 239  */
 240 typedef struct ret_mixed3_test_type_s {float x; int y; float x2; int y2;} ret_mixed3_test_type;
 241 typedef ret_mixed3_test_type (*ret_mixed3_test_function_type) (ret_mixed3_test_type);
 242
 243 static int ret_mixed3_test_callback(void *ptr) {
 244   ret_mixed3_test_function_type f = (ret_mixed3_test_function_type)ptr;
 245   ret_mixed3_test_type a = {10, 5, 35, 7 };
 246   ret_mixed3_test_type r;
 247   r = f(a);
 248   return ((r.x == a.x*5) && (r.y2 == a.y*3)) ? 0 : -1;
 249 }
 250
 251 static int ret_mixed3_test(void) {
 252   const char *src =
 253   "typedef struct ret_mixed3_test_type_s {float x; int y; float x2; int y2;} ret_mixed3_test_type;"
 254   "ret_mixed3_test_type f(ret_mixed3_test_type a) {\n"
 255   "  ret_mixed3_test_type r = {a.x*5, 0, 0, a.y*3};\n"
 256   "  return r;\n"
 257   "}\n";
 258
 259   return run_callback(src, ret_mixed3_test_callback);
 260 }
 261
 262 /*
 263  * reg_pack_test: return a small struct which should be packed into
 264  * registers (Win32) during return.
 265  */
 266 typedef struct reg_pack_test_type_s {int x, y;} reg_pack_test_type;
 267 typedef reg_pack_test_type (*reg_pack_test_function_type) (reg_pack_test_type);
 268
 269 static int reg_pack_test_callback(void *ptr) {
 270   reg_pack_test_function_type f = (reg_pack_test_function_type)ptr;
 271   reg_pack_test_type a = {10, 35};
 272   reg_pack_test_type r;
 273   r = f(a);
 274   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 275 }
 276
 277 static int reg_pack_test(void) {
 278   const char *src =
 279   "typedef struct reg_pack_test_type_s {int x, y;} reg_pack_test_type;"
 280   "reg_pack_test_type f(reg_pack_test_type a) {\n"
 281   "  reg_pack_test_type r = {a.x*5, a.y*3};\n"
 282   "  return r;\n"
 283   "}\n";
 284
 285   return run_callback(src, reg_pack_test_callback);
 286 }
 287
 288 /*
 289  * reg_pack_longlong_test: return a small struct which should be packed into
 290  * registers (x86-64) during return.
 291  */
 292 typedef struct reg_pack_longlong_test_type_s {long long x, y;} reg_pack_longlong_test_type;
 293 typedef reg_pack_longlong_test_type (*reg_pack_longlong_test_function_type) (reg_pack_longlong_test_type);
 294
 295 static int reg_pack_longlong_test_callback(void *ptr) {
 296   reg_pack_longlong_test_function_type f = (reg_pack_longlong_test_function_type)ptr;
 297   reg_pack_longlong_test_type a = {10, 35};
 298   reg_pack_longlong_test_type r;
 299   r = f(a);
 300   return ((r.x == a.x*5) && (r.y == a.y*3)) ? 0 : -1;
 301 }
 302
 303 static int reg_pack_longlong_test(void) {
 304   const char *src =
 305   "typedef struct reg_pack_longlong_test_type_s {long long x, y;} reg_pack_longlong_test_type;"
 306   "reg_pack_longlong_test_type f(reg_pack_longlong_test_type a) {\n"
 307   "  reg_pack_longlong_test_type r = {a.x*5, a.y*3};\n"
 308   "  return r;\n"
 309   "}\n";
 310
 311   return run_callback(src, reg_pack_longlong_test_callback);
 312 }
 313
 314 /*
 315  * ret_6plus2longlong_test:
 316  *
 317  * This catches a corner case in the x86_64 ABI code: the first 5
 318  * arguments fit into registers, the 6th doesn't, but the 7th argument
 319  * fits into the 6th argument integer register, %r9.
 320  *
 321  * Note that the purpose of the 10th argument is to avoid a situation
 322  * in which gcc would accidentally put the longlong at the right
 323  * address, thus causing a success message even though TCC actually
 324  * generated incorrect code.
 325  */
 326 typedef reg_pack_longlong_test_type (*ret_6plus2longlong_test_function_type) (long long, long long, long long, long long, long long, reg_pack_longlong_test_type, long long, long long);
 327
 328 static int ret_6plus2longlong_test_callback(void *ptr) {
 329   ret_6plus2longlong_test_function_type f = (ret_6plus2longlong_test_function_type)ptr;
 330   reg_pack_longlong_test_type a = {10, 35};
 331   reg_pack_longlong_test_type r;
 332   r = f(0, 0, 0, 0, 0, a, 37, 38);
 333   return ((r.x == 37) && (r.y == 37)) ? 0 : -1;
 334 }
 335
 336 static int ret_6plus2longlong_test(void) {
 337   const char *src =
 338   "typedef struct reg_pack_longlong_test_type_s {long long x, y;} reg_pack_longlong_test_type;"
 339   "reg_pack_longlong_test_type f(long long x1, long long x2, long long x3, long long x4, long long x5, reg_pack_longlong_test_type a, long long x8, long long x9) {\n"
 340   "  reg_pack_longlong_test_type r = { x8, x8 };\n"
 341   "  return r;\n"
 342   "}\n";
 343
 344   return run_callback(src, ret_6plus2longlong_test_callback);
 345 }
 346
 347 /*
 348  * sret_test: Create a struct large enough to be returned via sret
 349  * (hidden pointer as first function argument)
 350  */
 351 typedef struct sret_test_type_s {long long a, b, c;} sret_test_type;
 352 typedef sret_test_type (*sret_test_function_type) (sret_test_type);
 353
 354 static int sret_test_callback(void *ptr) {
 355   sret_test_function_type f = (sret_test_function_type)(ptr);
 356   sret_test_type x = {5436LL, 658277698LL, 43878957LL};
 357   sret_test_type r = f(x);
 358   return ((r.a==x.a*35)&&(r.b==x.b*19)&&(r.c==x.c*21)) ? 0 : -1;
 359 }
 360
 361 static int sret_test(void) {
 362   const char *src =
 363   "typedef struct sret_test_type_s {long long a, b, c;} sret_test_type;\n"
 364   "sret_test_type f(sret_test_type x) {\n"
 365   "  sret_test_type r = {x.a*35, x.b*19, x.c*21};\n"
 366   "  return r;\n"
 367   "}\n";
 368
 369   return run_callback(src, sret_test_callback);
 370 }
 371
 372 /*
 373  * one_member_union_test:
 374  *
 375  * In the x86-64 ABI a union should always be passed on the stack. However
 376  * it appears that a single member union is treated by GCC as its member.
 377  */
 378 typedef union one_member_union_test_type_u {int x;} one_member_union_test_type;
 379 typedef one_member_union_test_type (*one_member_union_test_function_type) (one_member_union_test_type);
 380
 381 static int one_member_union_test_callback(void *ptr) {
 382   one_member_union_test_function_type f = (one_member_union_test_function_type)ptr;
 383   one_member_union_test_type a, b;
 384   a.x = 34;
 385   b = f(a);
 386   return (b.x == a.x*2) ? 0 : -1;
 387 }
 388
 389 static int one_member_union_test(void) {
 390   const char *src =
 391   "typedef union one_member_union_test_type_u {int x;} one_member_union_test_type;\n"
 392   "one_member_union_test_type f(one_member_union_test_type a) {\n"
 393   "  one_member_union_test_type b;\n"
 394   "  b.x = a.x * 2;\n"
 395   "  return b;\n"
 396   "}\n";
 397   return run_callback(src, one_member_union_test_callback);
 398 }
 399
 400 /*
 401  * two_member_union_test:
 402  *
 403  * In the x86-64 ABI a union should always be passed on the stack.
 404  */
 405 typedef union two_member_union_test_type_u {int x; long y;} two_member_union_test_type;
 406 typedef two_member_union_test_type (*two_member_union_test_function_type) (two_member_union_test_type);
 407
 408 static int two_member_union_test_callback(void *ptr) {
 409   two_member_union_test_function_type f = (two_member_union_test_function_type)ptr;
 410   two_member_union_test_type a, b;
 411   a.x = 34;
 412   b = f(a);
 413   return (b.x == a.x*2) ? 0 : -1;
 414 }
 415
 416 static int two_member_union_test(void) {
 417   const char *src =
 418   "typedef union two_member_union_test_type_u {int x; long y;} two_member_union_test_type;\n"
 419   "two_member_union_test_type f(two_member_union_test_type a) {\n"
 420   "  two_member_union_test_type b;\n"
 421   "  b.x = a.x * 2;\n"
 422   "  return b;\n"
 423   "}\n";
 424   return run_callback(src, two_member_union_test_callback);
 425 }
 426
 427 /*
 428  * Win64 calling convetntion test.
 429  */
 430
 431 typedef struct many_struct_test_type_s {long long a, b, c;} many_struct_test_type;
 432 typedef many_struct_test_type (*many_struct_test_function_type) (many_struct_test_type,many_struct_test_type,many_struct_test_type,many_struct_test_type,many_struct_test_type,many_struct_test_type);
 433
 434 static int many_struct_test_callback(void *ptr) {
 435   many_struct_test_function_type f = (many_struct_test_function_type)ptr;
 436   many_struct_test_type v = {1, 2, 3};
 437   many_struct_test_type r = f(v,v,v,v,v,v);
 438   return ((r.a == 6) && (r.b == 12) && (r.c == 18))?0:-1;
 439 }
 440
 441 static int many_struct_test(void) {
 442   const char *src =
 443   "typedef struct many_struct_test_type_s {long long a, b, c;} many_struct_test_type;\n"
 444   "many_struct_test_type f(many_struct_test_type x1, many_struct_test_type x2, many_struct_test_type x3, many_struct_test_type x4, many_struct_test_type x5, many_struct_test_type x6) {\n"
 445   "  many_struct_test_type y;\n"
 446   "  y.a = x1.a + x2.a + x3.a + x4.a + x5.a + x6.a;\n"
 447   "  y.b = x1.b + x2.b + x3.b + x4.b + x5.b + x6.b;\n"
 448   "  y.c = x1.c + x2.c + x3.c + x4.c + x5.c + x6.c;\n"
 449   "  return y;\n"
 450   "}\n";
 451   return run_callback(src, many_struct_test_callback);
 452 }
 453
 454 /*
 455  * Win64 calling convention test.
 456  */
 457
 458 typedef struct many_struct_test_2_type_s {int a, b;} many_struct_test_2_type;
 459 typedef many_struct_test_2_type (*many_struct_test_2_function_type) (many_struct_test_2_type,many_struct_test_2_type,many_struct_test_2_type,many_struct_test_2_type,many_struct_test_2_type,many_struct_test_2_type);
 460
 461 static int many_struct_test_2_callback(void *ptr) {
 462   many_struct_test_2_function_type f = (many_struct_test_2_function_type)ptr;
 463   many_struct_test_2_type v = {1,2};
 464   many_struct_test_2_type r = f(v,v,v,v,v,v);
 465   return ((r.a == 6) && (r.b == 12))?0:-1;
 466 }
 467
 468 static int many_struct_test_2(void) {
 469   const char *src =
 470   "typedef struct many_struct_test_2_type_s {int a, b;} many_struct_test_2_type;\n"
 471   "many_struct_test_2_type f(many_struct_test_2_type x1, many_struct_test_2_type x2, many_struct_test_2_type x3, many_struct_test_2_type x4, many_struct_test_2_type x5, many_struct_test_2_type x6) {\n"
 472   "  many_struct_test_2_type y;\n"
 473   "  y.a = x1.a + x2.a + x3.a + x4.a + x5.a + x6.a;\n"
 474   "  y.b = x1.b + x2.b + x3.b + x4.b + x5.b + x6.b;\n"
 475   "  return y;\n"
 476   "}\n";
 477   return run_callback(src, many_struct_test_2_callback);
 478 }
 479
 480 /*
 481  * Win64 calling convention test.
 482  */
 483
 484 typedef struct many_struct_test_3_type_s {int a, b;} many_struct_test_3_type;
 485 typedef many_struct_test_3_type (*many_struct_test_3_function_type) (many_struct_test_3_type,many_struct_test_3_type,many_struct_test_3_type,many_struct_test_3_type,many_struct_test_3_type,many_struct_test_3_type, ...);
 486 typedef struct many_struct_test_3_struct_type { many_struct_test_3_function_type f; many_struct_test_3_function_type *f2; } many_struct_test_3_struct_type;
 487
 488 static void many_struct_test_3_dummy(double d, ...)
 489 {
 490   volatile double x = d;
 491 }
 492
 493 static int many_struct_test_3_callback(void *ptr) {
 494   many_struct_test_3_struct_type s = { ptr, };
 495   many_struct_test_3_struct_type *s2 = &s;
 496   s2->f2 = &s2->f;
 497   many_struct_test_3_dummy(1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, &s2);
 498   many_struct_test_3_function_type f = *(s2->f2);
 499   many_struct_test_3_type v = {1,2};
 500   many_struct_test_3_type r = (*((s2->f2=&f)+0))(v,v,v,v,v,v,1.0);
 501   return ((r.a == 6) && (r.b == 12))?0:-1;
 502 }
 503
 504 static int many_struct_test_3(void) {
 505   const char *src =
 506   "typedef struct many_struct_test_3_type_s {int a, b;} many_struct_test_3_type;\n"
 507   "many_struct_test_3_type f(many_struct_test_3_type x1, many_struct_test_3_type x2, many_struct_test_3_type x3, many_struct_test_3_type x4, many_struct_test_3_type x5, many_struct_test_3_type x6, ...) {\n"
 508   "  many_struct_test_3_type y;\n"
 509   "  y.a = x1.a + x2.a + x3.a + x4.a + x5.a + x6.a;\n"
 510   "  y.b = x1.b + x2.b + x3.b + x4.b + x5.b + x6.b;\n"
 511   "  return y;\n"
 512   "}\n";
 513   return run_callback(src, many_struct_test_3_callback);
 514 }
 515
 516 /*
 517  * stdarg_test: Test variable argument list ABI
 518  */
 519
 520 typedef struct {long long a, b, c;} stdarg_test_struct_type;
 521 typedef void (*stdarg_test_function_type) (int,int,int,...);
 522
 523 static int stdarg_test_callback(void *ptr) {
 524   stdarg_test_function_type f = (stdarg_test_function_type)ptr;
 525   int x;
 526   double y;
 527   stdarg_test_struct_type z = {1, 2, 3}, w;
 528   f(10, 10, 5,
 529     1, 2, 3, 4, 5, 6, 7, 8, 9, 10, &x,
 530     1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, &y,
 531     z, z, z, z, z, &w);
 532   return ((x == 55) && (y == 55) && (w.a == 5) && (w.b == 10) && (w.c == 15)) ? 0 : -1;
 533 }
 534
 535 static int stdarg_test(void) {
 536   const char *src =
 537   "#include <stdarg.h>\n"
 538   "typedef struct {long long a, b, c;} stdarg_test_struct_type;\n"
 539   "void f(int n_int, int n_float, int n_struct, ...) {\n"
 540   "  int i, ti = 0;\n"
 541   "  double td = 0.0;\n"
 542   "  stdarg_test_struct_type ts = {0,0,0}, tmp;\n"
 543   "  va_list ap;\n"
 544   "  va_start(ap, n_struct);\n"
 545   "  for (i = 0, ti = 0; i < n_int; ++i)\n"
 546   "    ti += va_arg(ap, int);\n"
 547   "  *va_arg(ap, int*) = ti;\n"
 548   "  for (i = 0, td = 0; i < n_float; ++i)\n"
 549   "    td += va_arg(ap, double);\n"
 550   "  *va_arg(ap, double*) = td;\n"
 551   "  for (i = 0; i < n_struct; ++i) {\n"
 552   "    tmp = va_arg(ap, stdarg_test_struct_type);\n"
 553   "    ts.a += tmp.a; ts.b += tmp.b; ts.c += tmp.c;"
 554   "  }\n"
 555   "  *va_arg(ap, stdarg_test_struct_type*) = ts;\n"
 556   "  va_end(ap);"
 557   "}\n";
 558   return run_callback(src, stdarg_test_callback);
 559 }
 560
 561 /*
 562  * Test Win32 stdarg handling, since the calling convention will pass a pointer
 563  * to the struct and the stdarg pointer must point to that pointer initially.
 564  */
 565
 566 typedef struct {long long a, b, c;} stdarg_struct_test_struct_type;
 567 typedef int (*stdarg_struct_test_function_type) (stdarg_struct_test_struct_type a, ...);
 568
 569 static int stdarg_struct_test_callback(void *ptr) {
 570   stdarg_struct_test_function_type f = (stdarg_struct_test_function_type)ptr;
 571   stdarg_struct_test_struct_type v = {10, 35, 99};
 572   int x = f(v, 234);
 573   return (x == 378) ? 0 : -1;
 574 }
 575
 576 static int stdarg_struct_test(void) {
 577   const char *src =
 578   "#include <stdarg.h>\n"
 579   "typedef struct {long long a, b, c;} stdarg_struct_test_struct_type;\n"
 580   "int f(stdarg_struct_test_struct_type a, ...) {\n"
 581   "  va_list ap;\n"
 582   "  va_start(ap, a);\n"
 583   "  int z = va_arg(ap, int);\n"
 584   "  va_end(ap);\n"
 585   "  return z + a.a + a.b + a.c;\n"
 586   "}\n";
 587   return run_callback(src, stdarg_struct_test_callback);
 588 }
 589
 590 /* Test that x86-64 arranges the stack correctly for arguments with alignment >8 bytes */
 591
 592 typedef LONG_DOUBLE (*arg_align_test_callback_type) (LONG_DOUBLE,int,LONG_DOUBLE,int,LONG_DOUBLE);
 593
 594 static int arg_align_test_callback(void *ptr) {
 595   arg_align_test_callback_type f = (arg_align_test_callback_type)ptr;
 596   long double x = f(12, 0, 25, 0, 37);
 597   return (x == 74) ? 0 : -1;
 598 }
 599
 600 static int arg_align_test(void) {
 601   const char *src =
 602   "long double f(long double a, int b, long double c, int d, long double e) {\n"
 603   "  return a + c + e;\n"
 604   "}\n";
 605   return run_callback(src, arg_align_test_callback);
 606 }
 607
 608 #define RUN_TEST(t) \
 609   if (!testname || (strcmp(#t, testname) == 0)) { \
 610     fputs(#t "... ", stdout); \
 611     fflush(stdout); \
 612     if (t() == 0) { \
 613       fputs("success\n", stdout); \
 614     } else { \
 615       fputs("failure\n", stdout); \
 616       retval = EXIT_FAILURE; \
 617     } \
 618   }
 619
 620 int main(int argc, char **argv) {
 621   int i;
 622   const char *testname = NULL;
 623   int retval = EXIT_SUCCESS;
 624
 625   /* if tcclib.h and libtcc1.a are not installed, where can we find them */
 626   for (i = 1; i < argc; ++i) {
 627     if (!memcmp(argv[i], "run_test=", 9))
 628       testname = argv[i] + 9;
 629   }
 630
 631   g_argv = argv, g_argc = argc;
 632
 633   RUN_TEST(ret_int_test);
 634   RUN_TEST(ret_longlong_test);
 635   RUN_TEST(ret_float_test);
 636   RUN_TEST(ret_double_test);
 637   RUN_TEST(ret_longdouble_test);
 638   RUN_TEST(ret_2float_test);
 639   RUN_TEST(ret_2double_test);
 640   RUN_TEST(ret_8plus2double_test);
 641   RUN_TEST(ret_6plus2longlong_test);
 642 #if !defined __x86_64__ || defined _WIN32
 643   /* currently broken on x86_64 linux */
 644   RUN_TEST(ret_mixed_test);
 645   RUN_TEST(ret_mixed2_test);
 646 #endif
 647   RUN_TEST(ret_mixed3_test);
 648   RUN_TEST(reg_pack_test);
 649   RUN_TEST(reg_pack_longlong_test);
 650   RUN_TEST(sret_test);
 651   RUN_TEST(one_member_union_test);
 652   RUN_TEST(two_member_union_test);
 653   RUN_TEST(many_struct_test);
 654   RUN_TEST(many_struct_test_2);
 655   RUN_TEST(many_struct_test_3);
 656   RUN_TEST(stdarg_test);
 657   RUN_TEST(stdarg_struct_test);
 658   RUN_TEST(arg_align_test);
 659   return retval;
 660 }