Add a version of netsend that uses the interval timer rather than

explicit clock reads to set an overall duration to the send, and
blasts rather than trying to clock output.  The goal of netblast,
unlike netsend, is to send as many UDP packets as possible; the
cost is that there's no ability to control the rate, and there's
less accuracy in the timing as the interval timer granularity is
relatively low.

2 files changed, 205 insertions, 0 deletions

diff --git a/tools/tools/netrate/netblast/Makefile b/tools/tools/netrate/netblast/Makefile
new file mode 100644
index 000000000000..910fa6306d18
--- /dev/null
+++ b/tools/tools/netrate/netblast/Makefile
@@ -0,0 +1,8 @@
+#
+# $FreeBSD$
+#
+
+PROG=	netblast
+NOMAN=	yes
+
+.include <bsd.prog.mk>
diff --git a/tools/tools/netrate/netblast/netblast.c b/tools/tools/netrate/netblast/netblast.c
new file mode 100644
index 000000000000..daed7e74e6e6
--- /dev/null
+++ b/tools/tools/netrate/netblast/netblast.c
@@ -0,0 +1,197 @@
+/*-
+ * Copyright (c) 2004 Robert N. M. Watson
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $FreeBSD$
+ */
+
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/time.h>
+
+#include <netinet/in.h>
+
+#include <arpa/inet.h>
+
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+static void
+usage(void)
+{
+
+	fprintf(stderr, "netblast [ip] [port] [payloadsize] [duration]\n");
+	exit(-1);
+}
+
+static int	global_stop_flag;
+
+static void
+signal_handler(int signum)
+{
+
+	global_stop_flag = 1;
+}
+
+/*
+ * Loop that blasts packets: begin by recording time information, resetting
+ * stats.  Set the interval timer for when we want to wake up.  Then go.
+ * SIGALRM will set a flag indicating it's time to stop.  Note that there's
+ * some overhead to the signal and timer setup, so the smaller the duration,
+ * the higher the relative overhead.
+ */
+static int
+blast_loop(int s, long duration, u_char *packet, u_int packet_len)
+{
+	struct timespec starttime, tmptime;
+	struct itimerval it;
+	u_int32_t counter;
+	int send_errors, send_calls;
+
+	if (signal(SIGALRM, signal_handler) == SIG_ERR) {
+		perror("signal");
+		return (-1);
+	}
+
+	if (clock_getres(CLOCK_REALTIME, &tmptime) == -1) {
+		perror("clock_getres");
+		return (-1);
+	}
+
+	if (clock_gettime(CLOCK_REALTIME, &starttime) == -1) {
+		perror("clock_gettime");
+		return (-1);
+	}
+
+	it.it_interval.tv_sec = 0;
+	it.it_interval.tv_usec = 0;
+	it.it_value.tv_sec = duration;
+	it.it_value.tv_usec = 0;
+
+	if (setitimer(ITIMER_REAL, &it, NULL) < 0) {
+		perror("setitimer");
+		return (-1);
+	}
+
+	send_errors = send_calls = 0;
+	counter = 0;
+	while (global_stop_flag == 0) {
+		/*
+		 * We maintain and, if there's room, send a counter.  Note
+		 * that even if the error is purely local, we still increment
+		 * the counter, so missing sequence numbers on the receive
+		 * side should not be assumed to be packets lost in transit.
+		 * For example, if the UDP socket gets back an ICMP from a
+		 * previous send, the error will turn up the current send
+		 * operation, causing the current sequence number also to be
+		 * skipped.
+		 *
+		 * XXXRW: Note alignment assumption.
+		 */
+		if (packet_len >= 4) {
+			*((u_int32_t *)packet) = htonl(counter);
+			counter++;
+		}
+		if (send(s, packet, packet_len, 0) < 0)
+			send_errors++;
+		send_calls++;
+	}
+
+	if (clock_gettime(CLOCK_REALTIME, &tmptime) == -1) {
+		perror("clock_gettime");
+		return (-1);
+	}
+
+	printf("\n");
+	printf("start:             %d.%09lu\n", starttime.tv_sec,
+	    starttime.tv_nsec);
+	printf("finish:            %d.%09lu\n", tmptime.tv_sec,
+	    tmptime.tv_nsec);
+	printf("send calls:        %d\n", send_calls);
+	printf("send errors:       %d\n", send_errors);
+	printf("approx send rate:  %ld\n", (send_calls - send_errors) /
+	    duration);
+	printf("approx error rate: %d\n", (send_errors / send_calls));
+
+	return (0);
+}
+
+int
+main(int argc, char *argv[])
+{
+	long payloadsize, port, duration;
+	struct sockaddr_in sin;
+	char *dummy, *packet;
+	int s;
+
+	if (argc != 5)
+		usage();
+
+	bzero(&sin, sizeof(sin));
+	sin.sin_len = sizeof(sin);
+	sin.sin_family = AF_INET;
+	if (inet_aton(argv[1], &sin.sin_addr) == 0) {
+		perror(argv[1]);
+		return (-1);
+	}
+
+	port = strtoul(argv[2], &dummy, 10);
+	if (port < 1 || port > 65535 || *dummy != '\0')
+		usage();
+	sin.sin_port = htons(port);
+
+	payloadsize = strtoul(argv[3], &dummy, 10);
+	if (payloadsize < 0 || *dummy != '\0')
+		usage();
+	if (payloadsize > 32768) {
+		fprintf(stderr, "payloadsize > 32768\n");
+		return (-1);
+	}
+
+	duration = strtoul(argv[4], &dummy, 10);
+	if (duration < 0 || *dummy != '\0')
+		usage();
+
+	packet = malloc(payloadsize);
+	if (packet == NULL) {
+		perror("malloc");
+		return (-1);
+	}
+	bzero(packet, payloadsize);
+
+	s = socket(PF_INET, SOCK_DGRAM, 0);
+	if (s == -1) {
+		perror("socket");
+		return (-1);
+	}
+
+	if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
+		perror("connect");
+		return (-1);
+	}
+
+	return (blast_loop(s, duration, packet, payloadsize));
+}