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