gecloud.ch/security-scripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added dnsperf

Browse source
This commit is contained in:
Peter Baumann 2013-06-04 15:22:42 +02:00
parent 4b465e4c1a
commit 651bb1703b
37 changed files with 13825 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

185
dns/dnsperf-src-2.0.0.0-1/net.c Normal file
View file

@ -0,0 +1,185 @@
/*
* Copyright (C) 2000, 2001 Nominum, Inc.
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM
* DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
* INTERNET SOFTWARE CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
* FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
* NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
* WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
/*
* Copyright (C) 2004 - 2012 Nominum, Inc.
*
* Permission to use, copy, modify, and distribute this software and its
* documentation for any purpose with or without fee is hereby granted,
* provided that the above copyright notice and this permission notice
* appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <isc/result.h>
#include <isc/sockaddr.h>
#include <bind9/getaddresses.h>
#include "log.h"
#include "net.h"
#include "opt.h"
int
perf_net_parsefamily(const char *family)
{
if (family == NULL || strcmp(family, "any") == 0)
return AF_UNSPEC;
else if (strcmp(family, "inet") == 0)
return AF_INET;
#ifdef AF_INET6
else if (strcmp(family, "inet6") == 0)
return AF_INET6;
#endif
else {
fprintf(stderr, "invalid family %s\n", family);
perf_opt_usage();
exit(1);
}
}
void
perf_net_parseserver(int family, const char *name, unsigned int port,
isc_sockaddr_t *addr)
{
isc_sockaddr_t addrs[8];
int count, i;
isc_result_t result;
if (port == 0) {
fprintf(stderr, "server port cannot be 0\n");
perf_opt_usage();
exit(1);
}
count = 0;
result = bind9_getaddresses(name, port, addrs, 8, &count);
if (result == ISC_R_SUCCESS) {
for (i = 0; i < count; i++) {
if (isc_sockaddr_pf(&addrs[i]) == family ||
family == AF_UNSPEC)
{
*addr = addrs[i];
return;
}
}
}
fprintf(stderr, "invalid server address %s\n", name);
perf_opt_usage();
exit(1);
}
void
perf_net_parselocal(int family, const char *name, unsigned int port,
isc_sockaddr_t *addr)
{
struct in_addr in4a;
struct in6_addr in6a;
if (name == NULL) {
isc_sockaddr_anyofpf(addr, family);
isc_sockaddr_setport(addr, port);
} else if (inet_pton(AF_INET, name, &in4a) == 1) {
isc_sockaddr_fromin(addr, &in4a, port);
} else if (inet_pton(AF_INET6, name, &in6a) == 1) {
isc_sockaddr_fromin6(addr, &in6a, port);
} else {
fprintf(stderr, "invalid local address %s\n", name);
perf_opt_usage();
exit(1);
}
}
int
perf_net_opensocket(const isc_sockaddr_t *server, const isc_sockaddr_t *local,
unsigned int offset, int bufsize)
{
int family;
int sock;
isc_sockaddr_t tmp;
int port;
int ret;
int flags;
family = isc_sockaddr_pf(server);
if (isc_sockaddr_pf(local) != family)
perf_log_fatal("server and local addresses have "
"different families");
sock = socket(family, SOCK_DGRAM, 0);
if (sock == -1)
perf_log_fatal("Error: socket: %s\n", strerror(errno));
#if defined(AF_INET6) && defined(IPV6_V6ONLY)
if (family == AF_INET6) {
int on = 1;
if (setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY,
&on, sizeof(on)) == -1) {
perf_log_warning("setsockopt(IPV6_V6ONLY) failed");
}
}
#endif
tmp = *local;
port = isc_sockaddr_getport(&tmp);
if (port != 0)
port += offset;
if (port >= 0xFFFF)
perf_log_fatal("port %d out of range", port);
if (bind(sock, &tmp.type.sa, tmp.length) == -1)
perf_log_fatal("bind: %s\n", strerror(errno));
if (bufsize > 0) {
bufsize *= 1024;
ret = setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
&bufsize, sizeof(bufsize));
if (ret < 0)
perf_log_warning("setsockbuf(SO_RCVBUF) failed");
ret = setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
&bufsize, sizeof(bufsize));
if (ret < 0)
perf_log_warning("setsockbuf(SO_SNDBUF) failed");
}
flags = fcntl(sock, F_GETFL, 0);
if (flags < 0)
perf_log_fatal("fcntl(F_GETFL)");
ret = fcntl(sock, F_SETFL, flags | O_NONBLOCK);
if (ret < 0)
perf_log_fatal("fcntl(F_SETFL)");
return sock;
}