chiark - git - mdw - disorder/blob - tests/udplog.c

   1 /*
   2  * This file is part of DisOrder
   3  * Copyright (C) 2007 Richard Kettlewell
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  18  * USA
  19  */
  20
  21 #include <config.h>
  22 #include "types.h"
  23
  24 #include <getopt.h>
  25 #include <sys/types.h>
  26 #include <sys/socket.h>
  27 #include <netinet/in.h>
  28 #include <errno.h>
  29 #include <locale.h>
  30 #include <netdb.h>
  31 #include <unistd.h>
  32 #include <ctype.h>
  33
  34 #include "configuration.h"
  35 #include "syscalls.h"
  36 #include "log.h"
  37 #include "addr.h"
  38 #include "defs.h"
  39 #include "mem.h"
  40
  41 static const struct option options[] = {
  42   { "help", no_argument, 0, 'h' },
  43   { "version", no_argument, 0, 'V' },
  44   { "output", required_argument, 0, 'o' },
  45   { 0, 0, 0, 0 }
  46 };
  47
  48 /* display usage message and terminate */
  49 static void help(void) {
  50   xprintf("Usage:\n"
  51           "  disorder-udplog [OPTIONS] ADDRESS PORT\n"
  52           "Options:\n"
  53           "  --output, -o PATH       Output to PATH (default: stdout)\n"
  54           "  --help, -h              Display usage message\n"
  55           "  --version, -V           Display version number\n"
  56           "\n"
  57           "UDP packet receiver.\n");
  58   xfclose(stdout);
  59   exit(0);
  60 }
  61
  62 /* display version number and terminate */
  63 static void version(void) {
  64   xprintf("%s", disorder_version_string);
  65   xfclose(stdout);
  66   exit(0);
  67 }
  68
  69 int main(int argc, char **argv) {
  70   int n, fd, err, i, j;
  71   struct addrinfo *ai;
  72   struct stringlist a;
  73   char *name, h[4096], s[4096];
  74   uint8_t buffer[4096];
  75   union {
  76     struct sockaddr sa;
  77     struct sockaddr_in sin;
  78     struct sockaddr_in6 sin6;
  79   } sa;
  80   socklen_t len;
  81   fd_set fds;
  82   struct timeval tv;
  83   static const struct addrinfo pref = {
  84     0,                          /* ai_flags */
  85     AF_UNSPEC,                  /* ai_family */
  86     SOCK_DGRAM,                 /* ai_socktype */
  87     IPPROTO_UDP,                /* ai_protocol */
  88     0,
  89     0,
  90     0,
  91     0
  92   };
  93
  94   set_progname(argv);
  95   mem_init();
  96   if(!setlocale(LC_CTYPE, "")) fatal(errno, "error calling setlocale");
  97   while((n = getopt_long(argc, argv, "hVo:", options, 0)) >= 0) {
  98     switch(n) {
  99     case 'h': help();
 100     case 'V': version();
 101     case 'o':
 102       if(!freopen(optarg, "w", stdout))
 103         fatal(errno, "%s", optarg);
 104       break;
 105     default: fatal(0, "invalid option");
 106     }
 107   }
 108   if(optind + 2 != argc)
 109     fatal(0, "missing arguments");
 110   a.n = 2;
 111   a.s = &argv[optind];
 112   if(!(ai = get_address(&a, &pref, &name)))
 113     exit(1);
 114   fd = xsocket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
 115   nonblock(fd);
 116   if(bind(fd, ai->ai_addr, ai->ai_addrlen) < 0)
 117     fatal(errno, "error binding to %s", name);
 118   while(getppid() != 1) {
 119     /* Wait for something to happen.  We don't just block forever in recvfrom()
 120      * as otherwise we'd never die if the parent terminated uncontrolledly. */
 121     FD_ZERO(&fds);
 122     FD_SET(fd, &fds);
 123     tv.tv_sec = 1;
 124     tv.tv_usec = 0;
 125     select(fd + 1, &fds, 0, 0, &tv);
 126     len = sizeof sa;
 127     n = recvfrom(fd, buffer, sizeof buffer, 0, &sa.sa, &len);
 128     if(n < 0) {
 129       if(errno == EINTR || errno == EAGAIN)
 130         continue;
 131       fatal(errno, "%s: recvfrom", name);
 132     }
 133     if((err = getnameinfo(&sa.sa, len, h, sizeof h, s, sizeof s,
 134                           NI_NUMERICHOST|NI_NUMERICSERV|NI_DGRAM)))
 135       fatal(0, "getnameinfo: %s", gai_strerror(err));
 136     xprintf("from host %s service %s: %d bytes\n", h, s, n);
 137     for(i = 0; i < n; i += 16) {
 138       for(j = i; j < n && j < i + 16; ++j)
 139         xprintf(" %02x", buffer[j]);
 140       for(; j < i + 16; ++j)
 141         xprintf("   ");
 142       xprintf("  ");
 143       for(j = i; j < n && j < i + 16; ++j)
 144         xprintf("%c", buffer[j] < 128 && isprint(buffer[j]) ? buffer[j] : '.');
 145       xprintf("\n");
 146       if(fflush(stdout) < 0)
 147         fatal(errno, "stdout");
 148     }
 149   }
 150   return 0;
 151 }
 152
 153 /*
 154 Local Variables:
 155 c-basic-offset:2
 156 comment-column:40
 157 fill-column:79
 158 indent-tabs-mode:nil
 159 End:
 160 */