2 m4_dnl (part of complex test harness, not of the library)
3 m4_dnl - routines for fuzzing
5 m4_dnl This file is part of adns, which is
6 m4_dnl Copyright (C) 1997-2000,2003,2006,2014-2016 Ian Jackson
7 m4_dnl Copyright (C) 2014 Mark Wooding
8 m4_dnl Copyright (C) 1999-2000,2003,2006 Tony Finch
9 m4_dnl Copyright (C) 1991 Massachusetts Institute of Technology
10 m4_dnl (See the file INSTALL for full details.)
12 m4_dnl This program is free software; you can redistribute it and/or modify
13 m4_dnl it under the terms of the GNU General Public License as published by
14 m4_dnl the Free Software Foundation; either version 3, or (at your option)
15 m4_dnl any later version.
17 m4_dnl This program is distributed in the hope that it will be useful,
18 m4_dnl but WITHOUT ANY WARRANTY; without even the implied warranty of
19 m4_dnl MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 m4_dnl GNU General Public License for more details.
22 m4_dnl You should have received a copy of the GNU General Public License
23 m4_dnl along with this program; if not, write to the Free Software Foundation.
25 m4_include(hmacros.i4)
32 #include <sys/types.h>
33 #include <sys/socket.h>
34 #include <netinet/in.h>
35 #include <arpa/inet.h>
47 #define FDF_NONBLOCK 002u
49 static FILE *Tinputfile, *traceout;
50 static int traceprint;
52 static void Tflushtrace( void) {
53 if (fflush(traceout)) Toutputerr();
56 void Tensuresetup(void) {
66 Tinputfile= fdopen(fd,"rb");
67 if (!Tinputfile) Tfailed("fdopen record fd");
69 while (fdtab.used < 3) {
70 const char fdfstd = FDF_OPEN;
71 if (!adns__vbuf_append(&fdtab,&fdfstd,1)) Tnomem();
74 const char *traceprintstr= getenv("ADNS_TEST_FUZZRAW_TRACEPRINT");
76 traceprint= atoi(traceprintstr);
78 if (tracefd<0) Tfailed("dup for tracefd");
79 traceout= fdopen(tracefd,"w");
80 if (!traceout) Tfailed("fdopen for traceout");
85 if (!traceprint) return; /* hcommon.c.m4 can call Q_vb directly */
86 if (!adns__vbuf_append(&vb,"",1)) Tnomem();
87 if (fprintf(traceout," %s\n",vb.buf) == EOF) Toutputerr();
91 static void Pformat(const char *what) {
92 fprintf(stderr,"adns test harness: format error in raw log input file: %s\n",what);
96 extern void Tshutdown(void) {
97 if (!Tinputfile) return;
98 int c= fgetc(Tinputfile);
99 if (c!=EOF) Pformat("unwanted additional syscall reply data");
100 if (ferror(Tinputfile)) Tfailed("read test log input (at end)");
103 static void Pcheckinput(void) {
104 if (ferror(Tinputfile)) Tfailed("read test log input file");
105 if (feof(Tinputfile)) Pformat("eof at syscall reply");
108 static void P_read_dump(const unsigned char *p0, size_t count, ssize_t d) {
109 fputs(" | ",traceout);
111 fprintf(traceout,"%02x", *p0);
117 static void P_read(void *p, size_t sz, const char *what) {
118 long pos = ftell(Tinputfile);
119 ssize_t got = fread(p,1,sz,Tinputfile);
122 if (traceprint>1 && sz) {
123 fprintf(traceout,"%8lx %8s:",pos,what);
124 P_read_dump(p, sz, +1);
126 P_read_dump((const unsigned char *)p+sz-1, sz, -1);
128 fputs(" |\n",traceout);
133 #define P_READ(x) (P_read(&(x), sizeof((x)), #x))
135 static unsigned P_fdf(int fd) {
136 assert(fd>=0 && fd<fdtab.used);
137 return fdtab.buf[fd];
140 void T_gettimeofday_hook(void) {
141 struct timeval delta, sum;
143 timeradd(&delta, ¤ttime, &sum);
144 sum.tv_usec %= 1000000;
148 static void Paddr(struct sockaddr *addr, int *lenr) {
152 socklen_t sl = *lenr;
155 if (al<0 || al>=sizeof(buf)-1) Pformat("bad addr len");
156 P_read(buf,al,"addrtext");
159 r= adns_text2addr(buf,port, adns_qf_addrlit_scope_numeric, addr, &sl);
160 if (r==EINVAL) Pformat("bad addr text");
161 assert(r==0 || r==ENOSPC);
165 static int Pbytes(byte *buf, int maxlen) {
168 if (bl<0 || bl>maxlen) Pformat("bad byte block len");
169 P_read(buf, bl, "bytes");
173 static void Pfdset(fd_set *set, int max, int *r_io) {
180 for (fd=max-1; fd>=0; fd--) {
185 _Bool y = fdmap & 1u;
189 if (!FD_ISSET(fd,set)) continue;
202 static void Psync(const char *exp, char *got, size_t sz, const char *what) {
203 P_read(got,sz,"syscall");
204 if (memcmp(exp,got,sz)) Pformat(what);
207 m4_define(`syscall_sync',`
209 hm_fr_syscall_ident($'`1)
210 static char sync_got[sizeof(sync_expect)];
211 Psync(sync_expect, sync_got, sizeof(sync_got), "sync lost: program did $1");
216 static void Ppollfds(struct pollfd *fds, int nfds, int *r_io) {
218 for (fd=0; fd<nfds; fd++) {
219 if (!fds[fd].events) continue;
221 P_READ(fds[fd].revents);
228 static int P_succfail(void) {
235 Pformat("wrong errno value");
240 m4_define(`hm_syscall', `
242 int H$1(hm_args_massage($3,void)) {
247 hm_create_hqcall_vars
250 hm_create_hqcall_init($1)
256 hm_create_hqcall_args
257 Q$1(hm_args_massage($3));
262 m4_define(`hm_rv_succfail',`
267 m4_define(`hm_rv_any',`
271 if (r<0) Pformat("negative nonerror syscall return");
274 m4_define(`hm_rv_len',`
277 m4_define(`hm_rv_must',`
280 m4_define(`hm_rv_select',`hm_rv_succfail')
281 m4_define(`hm_rv_poll',`hm_rv_succfail')
282 m4_define(`hm_rv_fcntl',`
283 unsigned flg = P_fdf(fd);
284 if (cmd == F_GETFL) {
285 r= (flg & FDF_NONBLOCK) ? O_NONBLOCK : 0;
286 } else if (cmd == F_SETFL) {
287 flg &= ~FDF_NONBLOCK;
288 if (arg & O_NONBLOCK)
296 m4_define(`hm_rv_fd',`
301 if (newfd<0 || newfd>1000) Pformat("new fd out of range");
302 adns__vbuf_ensure(&fdtab, newfd+1);
303 if (fdtab.used <= newfd) {
304 memset(fdtab.buf+fdtab.used, 0, newfd+1-fdtab.used);
307 if (fdtab.buf[newfd]) Pformat("new fd already in use");
308 fdtab.buf[newfd] |= FDF_OPEN;
312 m4_define(`hm_rv_wlen',`
314 if (r>$'`1) Pformat("write return value too large");
319 m4_define(`hm_arg_fdset_io',`Pfdset($'`1,$'`2,&r);')
320 m4_define(`hm_arg_pollfds_io',`Ppollfds($'`1,$'`2,&r);')
321 m4_define(`hm_arg_addr_out',`Paddr($'`1,$'`2);')
325 m4_define(`hm_arg_bytes_out',`r= Pbytes($'`2,$'`4);')
329 m4_define(`hm_rv_selectpoll',`
330 if (($'`1) && !r) Pformat("select/poll returning 0 but infinite timeout");
332 m4_define(`hm_rv_select',`hm_rv_selectpoll(!to)')
333 m4_define(`hm_rv_poll',`hm_rv_selectpoll(timeout<0)')
340 m4_define(`hm_specsyscall', `')
342 m4_include(`hsyscalls.i4')