1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <stdarg.h>
4
5 volatile int should_optimize;
6
7 int
8 __attribute__((noinline))
__fprintf_chk(FILE * f,int flag,const char * fmt,...)9 __fprintf_chk (FILE *f, int flag, const char *fmt, ...)
10 {
11 va_list ap;
12 int ret;
13 #ifdef __OPTIMIZE__
14 if (should_optimize)
15 abort ();
16 #endif
17 should_optimize = 1;
18 va_start (ap, fmt);
19 ret = vfprintf (f, fmt, ap);
20 va_end (ap);
21 return ret;
22 }
23
24 int
main(void)25 main (void)
26 {
27 #define test(ret, opt, args...) \
28 should_optimize = opt; \
29 __fprintf_chk (stdout, 1, args); \
30 if (!should_optimize) \
31 abort (); \
32 should_optimize = 0; \
33 if (__fprintf_chk (stdout, 1, args) != ret) \
34 abort (); \
35 if (!should_optimize) \
36 abort ();
37 test (5, 1, "hello");
38 test (6, 1, "hello\n");
39 test (1, 1, "a");
40 test (0, 1, "");
41 test (5, 1, "%s", "hello");
42 test (6, 1, "%s", "hello\n");
43 test (1, 1, "%s", "a");
44 test (0, 1, "%s", "");
45 test (1, 1, "%c", 'x');
46 test (7, 0, "%s\n", "hello\n");
47 test (2, 0, "%d\n", 0);
48 return 0;
49 }
50