1#include <stdio.h>
2#include <stdlib.h>
3#include <stdarg.h>
4
5volatile int should_optimize;
6
7int
8__attribute__((noinline))
9__printf_chk (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 = vprintf (fmt, ap);
20  va_end (ap);
21  return ret;
22}
23
24int
25main (void)
26{
27#define test(ret, opt, args...) \
28  should_optimize = opt;		\
29  __printf_chk (1, args); 		\
30  if (!should_optimize)			\
31    abort ();				\
32  should_optimize = 0;			\
33  if (__printf_chk (1, args) != ret)	\
34    abort ();				\
35  if (!should_optimize)			\
36    abort ();
37  test (5, 0, "hello");
38  test (6, 1, "hello\n");
39  test (1, 1, "a");
40  test (0, 1, "");
41  test (5, 0, "%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, 1, "%s\n", "hello\n");
47  test (2, 0, "%d\n", 0);
48  return 0;
49}
50