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 "common.h"
  22
  23 #include <getopt.h>
  24 #include <sys/types.h>
  25 #include <sys/socket.h>
  26 #include <netinet/in.h>
  27 #include <errno.h>
  28 #include <locale.h>
  29 #include <netdb.h>
  30 #include <unistd.h>
  31 #include <ctype.h>
  32
  33 #include "configuration.h"
  34 #include "syscalls.h"
  35 #include "log.h"
  36 #include "addr.h"
  37 #include "defs.h"
  38 #include "mem.h"
  39
  40 static const struct option options[] = {
  41   { "help", no_argument, 0, 'h' },
  42   { "version", no_argument, 0, 'V' },
  43   { "output", required_argument, 0, 'o' },
  44   { 0, 0, 0, 0 }
  45 };
  46
  47 /* display usage message and terminate */
  48 static void help(void) {
  49   xprintf("Usage:\n"
  50           "  disorder-udplog [OPTIONS] ADDRESS PORT\n"
  51           "Options:\n"
  52           "  --output, -o PATH       Output to PATH (default: stdout)\n"
  53           "  --help, -h              Display usage message\n"
  54           "  --version, -V           Display version number\n"
  55           "\n"
  56           "UDP packet receiver.\n");
  57   xfclose(stdout);
  58   exit(0);
  59 }
  60
  61 /* display version number and terminate */
  62 static void version(void) {
  63   xprintf("%s", disorder_version_string);
  64   xfclose(stdout);
  65   exit(0);
  66 }
  67
  68 int main(int argc, char **argv) {
  69   int n, fd, err, i, j;
  70   struct addrinfo *ai;
  71   struct stringlist a;
  72   char *name, h[4096], s[4096];
  73   uint8_t buffer[4096];
  74   union {
  75     struct sockaddr sa;
  76     struct sockaddr_in sin;
  77     struct sockaddr_in6 sin6;
  78   } sa;
  79   socklen_t len;
  80   fd_set fds;
  81   struct timeval tv;
  82   static const struct addrinfo pref = {
  83     .ai_flags = 0,
  84     .ai_family = AF_UNSPEC,
  85     .ai_socktype = SOCK_DGRAM,
  86     .ai_protocol = IPPROTO_UDP,
  87   };
  88
  89   set_progname(argv);
  90   mem_init();
  91   if(!setlocale(LC_CTYPE, "")) fatal(errno, "error calling setlocale");
  92   while((n = getopt_long(argc, argv, "hVo:", options, 0)) >= 0) {
  93     switch(n) {
  94     case 'h': help();
  95     case 'V': version();
  96     case 'o':
  97       if(!freopen(optarg, "w", stdout))
  98         fatal(errno, "%s", optarg);
  99       break;
 100     default: fatal(0, "invalid option");
 101     }
 102   }
 103   if(optind + 2 != argc)
 104     fatal(0, "missing arguments");
 105   a.n = 2;
 106   a.s = &argv[optind];
 107   if(!(ai = get_address(&a, &pref, &name)))
 108     exit(1);
 109   fd = xsocket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
 110   nonblock(fd);
 111   if(bind(fd, ai->ai_addr, ai->ai_addrlen) < 0)
 112     fatal(errno, "error binding to %s", name);
 113   while(getppid() != 1) {
 114     /* Wait for something to happen.  We don't just block forever in recvfrom()
 115      * as otherwise we'd never die if the parent terminated uncontrolledly. */
 116     FD_ZERO(&fds);
 117     FD_SET(fd, &fds);
 118     tv.tv_sec = 1;
 119     tv.tv_usec = 0;
 120     select(fd + 1, &fds, 0, 0, &tv);
 121     len = sizeof sa;
 122     n = recvfrom(fd, buffer, sizeof buffer, 0, &sa.sa, &len);
 123     if(n < 0) {
 124       if(errno == EINTR || errno == EAGAIN)
 125         continue;
 126       fatal(errno, "%s: recvfrom", name);
 127     }
 128     if((err = getnameinfo(&sa.sa, len, h, sizeof h, s, sizeof s,
 129                           NI_NUMERICHOST|NI_NUMERICSERV|NI_DGRAM)))
 130       fatal(0, "getnameinfo: %s", gai_strerror(err));
 131     xprintf("from host %s service %s: %d bytes\n", h, s, n);
 132     for(i = 0; i < n; i += 16) {
 133       for(j = i; j < n && j < i + 16; ++j)
 134         xprintf(" %02x", buffer[j]);
 135       for(; j < i + 16; ++j)
 136         xprintf("   ");
 137       xprintf("  ");
 138       for(j = i; j < n && j < i + 16; ++j)
 139         xprintf("%c", buffer[j] < 128 && isprint(buffer[j]) ? buffer[j] : '.');
 140       xprintf("\n");
 141       if(fflush(stdout) < 0)
 142         fatal(errno, "stdout");
 143     }
 144   }
 145   return 0;
 146 }
 147
 148 /*
 149 Local Variables:
 150 c-basic-offset:2
 151 comment-column:40
 152 fill-column:79
 153 indent-tabs-mode:nil
 154 End:
 155 */