1#!/bin/sh
2
3cat << _EOF > $2
4#include <sys/types.h>
5#include <sys/time.h>
6#include <stdio.h>
7#include <stdarg.h>
8#include <stdint.h>
9#include <stdlib.h>
10#include <string.h>
11#include <sha2.h>
12
13#include <atf-c.h>
14
15/* Avoid SSP re-definitions */
16#undef snprintf
17#undef vsnprintf
18#undef sprintf
19#undef vsprintf
20#undef vasprintf
21
22#define KPRINTF_BUFSIZE 1024
23#undef putchar
24#define putchar xputchar
25
26#define kmem_alloc(n, f)      malloc(n)
27
28#define kprintf_lock()                  __nothing
29#define kprintf_unlock()      __nothing
30
31/* Arbitrary */
32#define TOCONS      1
33#define TOLOG       2
34
35#define kprintf_internal(f, i1, i2, i3, ...) \
36          printf(f, __VA_ARGS__)
37
38static int putchar(char c, int foo, void *b)
39{
40          return fputc(c, stderr);
41}
42
43#define TOBUFONLY 1
44static const char HEXDIGITS[] = "0123456789ABCDEF";
45static const char hexdigits[] = "0123456789abcdef";
46
47typedef int device_t;
48
49#if 0
50static SHA512_CTX kprnd_sha;
51#endif
52
53#define timespec timeval
54#define nanotime(ts) gettimeofday(ts, NULL)
55
56#define device_xname(a) ""
57int kprintf(const char *, int, void *, char *, va_list) __printflike(1, 0);
58void device_printf(device_t, const char *, ...) __printflike(2, 3);
59
60static void
61empty(void)
62{
63}
64
65static void (*v_flush)(void) = empty;
66
67ATF_TC(snprintf_print);
68ATF_TC_HEAD(snprintf_print, tc)
69{
70          atf_tc_set_md_var(tc, "descr", "checks snprintf print");
71}
72 
73ATF_TC_BODY(snprintf_print, tc)
74{
75          char buf[13];
76          int i;
77
78          memset(buf, 'x', sizeof(buf));
79          i = snprintf(buf, sizeof(buf), "number %d", 10);
80          ATF_CHECK_EQ(i, 9);
81          ATF_CHECK_STREQ(buf, "number 10");
82}
83
84ATF_TC(snprintf_print_overflow);
85ATF_TC_HEAD(snprintf_print_overflow, tc)
86{
87          atf_tc_set_md_var(tc, "descr", "checks snprintf print with overflow");
88}
89 
90ATF_TC_BODY(snprintf_print_overflow, tc)
91{
92          char buf[10];
93          int i;
94
95          memset(buf, 'x', sizeof(buf));
96          i = snprintf(buf, sizeof(buf), "fjsdfsdjfsdf %d\n", 10);
97          ATF_CHECK_EQ(i, 16);
98          ATF_CHECK_STREQ(buf, "fjsdfsdjf");
99}
100
101ATF_TC(snprintf_count);
102ATF_TC_HEAD(snprintf_count, tc)
103{
104          atf_tc_set_md_var(tc, "descr", "checks snprintf count");
105}
106 
107ATF_TC_BODY(snprintf_count, tc)
108{
109          int i;
110          
111          i = snprintf(NULL, 20, "number %d", 10);
112          ATF_CHECK_EQ(i, 9);
113}
114
115ATF_TC(snprintf_count_overflow);
116ATF_TC_HEAD(snprintf_count_overflow, tc)
117{
118          atf_tc_set_md_var(tc, "descr", "checks snprintf count with overflow");
119}
120 
121ATF_TC_BODY(snprintf_count_overflow, tc)
122{
123          int i;
124
125          i = snprintf(NULL, 10, "fjsdfsdjfsdf %d\n", 10);
126          ATF_CHECK_EQ(i, 16);
127}
128
129ATF_TC(vasprintf_print);
130ATF_TC_HEAD(vasprintf_print, tc)
131{
132          atf_tc_set_md_var(tc, "descr", "checks vasprintf works");
133}
134
135static int __printflike(2, 3)
136vasp_helper(char **buf, const char *fmt, ...)
137{
138          va_list ap;
139          int ret;
140
141          va_start(ap, fmt);
142          ret = vasprintf(buf, fmt, ap);
143          va_end(ap);
144          return ret;
145}
146
147ATF_TC_BODY(vasprintf_print, tc)
148{
149          int i;
150          char *buf;
151
152          buf = NULL;
153          i =  vasp_helper(&buf, "N=%d C=%c S=%s", 7, 'x', "abc");
154          ATF_CHECK_EQ(i, 13);
155          ATF_CHECK(buf != NULL);
156          ATF_CHECK_STREQ(buf, "N=7 C=x S=abc");
157          free(buf);
158}
159
160ATF_TP_ADD_TCS(tp)
161{
162          ATF_TP_ADD_TC(tp, snprintf_print);
163          ATF_TP_ADD_TC(tp, snprintf_print_overflow);
164          ATF_TP_ADD_TC(tp, snprintf_count);
165          ATF_TP_ADD_TC(tp, snprintf_count_overflow);
166          ATF_TP_ADD_TC(tp, vasprintf_print);
167
168          return atf_no_error();
169}
170_EOF
171
172awk '
173/^snprintf\(/ {
174          print prevline
175          out = 1
176}
177{
178          if (out) print
179          else prevline = $0
180}' $1 >>$2
181