regress/hfuzzraw.c.m4

   1 m4_dnl hfuzzraw.c.m4
   2 m4_dnl (part of complex test harness, not of the library)
   3 m4_dnl - routines for fuzzing
   4
   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.)
  11 m4_dnl
  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.
  16 m4_dnl
  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.
  21 m4_dnl
  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.
  24
  25 m4_include(hmacros.i4)
  26
  27 #include <assert.h>
  28 #include <string.h>
  29 #include <errno.h>
  30 #include <stdlib.h>
  31
  32 #include <sys/types.h>
  33 #include <sys/socket.h>
  34 #include <netinet/in.h>
  35 #include <arpa/inet.h>
  36 #include <sys/time.h>
  37
  38 #include <unistd.h>
  39 #include <fcntl.h>
  40 #include <limits.h>
  41
  42
  43 #include "harness.h"
  44
  45 static vbuf fdtab;
  46 #define FDF_OPEN     001u
  47 #define FDF_NONBLOCK 002u
  48
  49 static FILE *Tinputfile;
  50 static int stdout_enable;
  51
  52 void Q_vb(void) {
  53   if (!adns__vbuf_append(&vb,"",1)) Tnomem();
  54   if (fprintf(stdout," %s\n",vb.buf) == EOF) Toutputerr();
  55   if (fflush(stdout)) Toutputerr();
  56 }
  57
  58 static void Pformat(const char *what) {
  59   fprintf(stderr,"adns test harness: format error in raw log input file: %s\n",what);
  60   exit(-1);
  61 }
  62
  63 extern void Tshutdown(void) {
  64   int c= fgetc(Tinputfile);
  65   if (c!=EOF) Pformat("unwanted additional syscall reply data");
  66   if (ferror(Tinputfile)) Tfailed("read test log input (at end)");
  67 }
  68
  69 static void Pcheckinput(void) {
  70   if (ferror(Tinputfile)) Tfailed("read test log input file");
  71   if (feof(Tinputfile)) Pformat("eof at syscall reply");
  72 }
  73
  74 void Tensurerecordfile(void) {
  75   static int done;
  76
  77   if (done) return;
  78   done++;
  79
  80   int fd;
  81
  82   fd = Ttestinputfd();
  83   assert(fd >= 0);
  84   Tinputfile= fdopen(fd,"rb");
  85     if (!Tinputfile) Tfailed("fdopen record fd");
  86
  87   while (fdtab.used < 3) {
  88     const char fdfstd = FDF_OPEN;
  89     if (!adns__vbuf_append(&fdtab,&fdfstd,1)) Tnomem();
  90   }
  91
  92   const char *proutstr= getenv("ADNS_TEST_FUZZRAW_STDOUT_ENABLE");
  93   if (proutstr) stdout_enable= atoi(proutstr);
  94 }
  95
  96 static void P_read(void *p, size_t sz) {
  97   ssize_t got = fread(p,1,sz,Tinputfile);
  98   Pcheckinput();
  99   assert(got==sz);
 100 }
 101
 102 #define P_READ(x) (P_read(&(x), sizeof((x))))
 103
 104 static unsigned P_fdf(int fd) {
 105   assert(fd>=0 && fd<fdtab.used);
 106   return fdtab.buf[fd];
 107 }
 108
 109 void T_gettimeofday_hook(void) {
 110   struct timeval delta, sum;
 111   P_READ(delta);
 112   timeradd(&delta, &currenttime, &sum);
 113   currenttime= sum;
 114 }
 115
 116 static void Paddr(struct sockaddr *addr, int *lenr) {
 117   int l, r;
 118   uint16_t port;
 119   char buf[512];
 120   socklen_t sl = *lenr;
 121
 122   P_READ(l);
 123   if (l<0 || l>=sizeof(buf)-1) Pformat("bad addr len");
 124   buf[l]= 0;
 125   P_READ(port);
 126   r= adns_text2addr(buf,port, adns_qf_addrlit_scope_numeric, addr, &sl);
 127   if (r==EINVAL) Pformat("bad addr text");
 128   assert(r==ENOSPC);
 129   *lenr = sl;
 130 }
 131
 132 static int Pbytes(byte *buf, int maxlen) {
 133   int l;
 134   P_READ(l);
 135   if (l<0 || l>maxlen) Pformat("bad byte block len");
 136   P_read(buf, l);
 137   return l;
 138 }
 139
 140 static void Pfdset(fd_set *set, int max) {
 141   uint16_t got;
 142   int fd, ngot=0;
 143
 144   for (fd=0; fd<max; fd++) {
 145     if (!FD_ISSET(fd,set)) continue;
 146     P_fdf(fd);
 147     if (ngot==0) {
 148       P_READ(got);
 149       ngot= 16;
 150     }
 151     if (!(got & 1u)) {
 152       FD_CLR(fd,set);
 153     }
 154     got >>= 1;
 155     ngot--;
 156   }
 157 }
 158
 159 #ifdef HAVE_POLL
 160 static void Ppollfds(struct pollfd *fds, int nfds) {
 161 int fd;
 162   for (fd=0; fd<nfds; fd++) {
 163     if (!fds[fd].events) continue;
 164     P_fdf(fd);
 165     P_READ(fds[fd].revents);
 166   }
 167 }
 168 #endif
 169
 170 static int P_succfail(void) {
 171   int r;
 172   P_READ(r);
 173   if (r<0 && -r<Tnerrnos) {
 174     errno= Terrnos[-r].v;
 175     return -1;
 176   } else if (r>0 && r<=255) {
 177     errno= r;
 178     return -1;
 179   } else if (r) {
 180     Pformat("wrong errno value");
 181   }
 182   return 0;
 183 }
 184
 185 m4_define(`hm_syscall', `
 186  hm_create_proto_h
 187 int H$1(hm_args_massage($3,void)) {
 188  int r;
 189  m4_define(`hm_rv_fd',`')
 190  m4_define(`hm_rv_any',`')
 191  m4_define(`hm_rv_len',`')
 192  m4_define(`hm_rv_must',`')
 193  m4_define(`hm_rv_succfail',`')
 194  m4_define(`hm_rv_fcntl',`')
 195  $2
 196
 197  hm_create_hqcall_vars
 198  $3
 199
 200  hm_create_hqcall_init($1)
 201  $3
 202
 203  Tensurerecordfile();
 204
 205  if (stdout_enable) {
 206    hm_create_hqcall_args
 207    Q$1(hm_args_massage($3));
 208  }
 209
 210  m4_define(`hm_rv_succfail',`
 211   r= P_succfail();
 212   if (r<0) return r;
 213  ')
 214
 215  m4_define(`hm_rv_any',`
 216   hm_rv_succfail
 217   if (!r) {
 218     P_READ(r);
 219     if (r<0) Pformat("negative nonerror syscall return");
 220   }
 221  ')
 222  m4_define(`hm_rv_len',`
 223   hm_rv_any
 224   if (r>($'`1)) Pformat("syscall length return is excessive");
 225  ')
 226  m4_define(`hm_rv_must',`
 227   r= 0;
 228  ')
 229  m4_define(`hm_rv_fcntl',`
 230   unsigned flg = P_fdf(fd);
 231   if (cmd == F_GETFL) {
 232     r= (flg & FDF_NONBLOCK) ? O_NONBLOCK : 0;
 233   } else if (cmd == F_SETFL) {
 234     flg &= ~FDF_NONBLOCK;
 235     if (arg & O_NONBLOCK)
 236       flg |= FDF_NONBLOCK;
 237     fdtab.buf[fd]= flg;
 238     r= 0;
 239   } else {
 240     abort();
 241   }
 242  ')
 243  m4_define(`hm_rv_fd',`
 244   hm_rv_succfail
 245   if (!r) {
 246     int newfd;
 247     P_READ(newfd);
 248     if (newfd<0 || newfd>1000) Pformat("new fd out of range");
 249     adns__vbuf_ensure(&fdtab, newfd+1);
 250     if (fdtab.used <= newfd) {
 251       memset(fdtab.buf+fdtab.used, 0, newfd+1-fdtab.used);
 252       fdtab.used= newfd+1;
 253     }
 254     if (fdtab.buf[newfd]) Pformat("new fd already in use");
 255     fdtab.buf[newfd] |= FDF_OPEN;
 256     r= newfd;
 257  }
 258  ')
 259  $2
 260
 261  hm_create_nothing
 262  m4_define(`hm_arg_fdset_io',`Pfdset($'`1,$'`2);')
 263  m4_define(`hm_arg_pollfds_io',`Ppollfds($'`1,$'`2);')
 264  m4_define(`hm_arg_addr_out',`Paddr($'`1,$'`2);')
 265  $3
 266
 267  hm_create_nothing
 268  m4_define(`hm_arg_bytes_out',`r= Pbytes($'`2,$'`4);')
 269  $3
 270
 271  return r;
 272 }
 273 ')
 274
 275 m4_define(`hm_specsyscall', `')
 276
 277 m4_include(`hsyscalls.i4')
 278
 279 int Hclose(int fd) {
 280   P_fdf(fd);
 281   fdtab.buf[fd]= 0;
 282   return P_succfail();
 283 }