1*86d7f5d3SJohn Marino /*-
2*86d7f5d3SJohn Marino * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
3*86d7f5d3SJohn Marino * All rights reserved.
4*86d7f5d3SJohn Marino *
5*86d7f5d3SJohn Marino * Redistribution and use in source and binary forms, with or without
6*86d7f5d3SJohn Marino * modification, are permitted provided that the following conditions
7*86d7f5d3SJohn Marino * are met:
8*86d7f5d3SJohn Marino * 1. Redistributions of source code must retain the above copyright
9*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer.
10*86d7f5d3SJohn Marino * 2. Redistributions in binary form must reproduce the above copyright
11*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer in the
12*86d7f5d3SJohn Marino * documentation and/or other materials provided with the distribution.
13*86d7f5d3SJohn Marino *
14*86d7f5d3SJohn Marino * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*86d7f5d3SJohn Marino * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*86d7f5d3SJohn Marino * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*86d7f5d3SJohn Marino * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*86d7f5d3SJohn Marino * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*86d7f5d3SJohn Marino * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*86d7f5d3SJohn Marino * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*86d7f5d3SJohn Marino * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*86d7f5d3SJohn Marino * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*86d7f5d3SJohn Marino * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*86d7f5d3SJohn Marino * SUCH DAMAGE.
25*86d7f5d3SJohn Marino *
26*86d7f5d3SJohn Marino */
27*86d7f5d3SJohn Marino
28*86d7f5d3SJohn Marino #include <sys/types.h>
29*86d7f5d3SJohn Marino #include <sys/socket.h>
30*86d7f5d3SJohn Marino #include <netinet/in.h>
31*86d7f5d3SJohn Marino #include <arpa/inet.h>
32*86d7f5d3SJohn Marino /*#include <sys/param.h>*/
33*86d7f5d3SJohn Marino #include <netdb.h>
34*86d7f5d3SJohn Marino #include <stdio.h>
35*86d7f5d3SJohn Marino #include <stdlib.h>
36*86d7f5d3SJohn Marino #include <unistd.h>
37*86d7f5d3SJohn Marino #include <errno.h>
38*86d7f5d3SJohn Marino #include <string.h>
39*86d7f5d3SJohn Marino #include <err.h>
40*86d7f5d3SJohn Marino
41*86d7f5d3SJohn Marino #include "stress.h"
42*86d7f5d3SJohn Marino
43*86d7f5d3SJohn Marino static int bufsize;
44*86d7f5d3SJohn Marino
45*86d7f5d3SJohn Marino int
setup(int nb)46*86d7f5d3SJohn Marino setup(int nb)
47*86d7f5d3SJohn Marino {
48*86d7f5d3SJohn Marino bufsize = 2 << random_int(1, 12);
49*86d7f5d3SJohn Marino return (0);
50*86d7f5d3SJohn Marino }
51*86d7f5d3SJohn Marino
52*86d7f5d3SJohn Marino void
cleanup(void)53*86d7f5d3SJohn Marino cleanup(void)
54*86d7f5d3SJohn Marino {
55*86d7f5d3SJohn Marino }
56*86d7f5d3SJohn Marino
57*86d7f5d3SJohn Marino int
test(void)58*86d7f5d3SJohn Marino test(void)
59*86d7f5d3SJohn Marino {
60*86d7f5d3SJohn Marino struct sockaddr_in sock_in;
61*86d7f5d3SJohn Marino struct hostent *host;
62*86d7f5d3SJohn Marino char *hostname;
63*86d7f5d3SJohn Marino int f, i, n;
64*86d7f5d3SJohn Marino int *buf;
65*86d7f5d3SJohn Marino
66*86d7f5d3SJohn Marino bzero((char *)&sock_in, sizeof(sock_in));
67*86d7f5d3SJohn Marino sock_in.sin_family = AF_INET;
68*86d7f5d3SJohn Marino f = socket(AF_INET, SOCK_DGRAM, 0);
69*86d7f5d3SJohn Marino if (f < 0)
70*86d7f5d3SJohn Marino err(1, "socket");
71*86d7f5d3SJohn Marino if (bind(f, (struct sockaddr *)&sock_in, sizeof(sock_in)) < 0) {
72*86d7f5d3SJohn Marino warn("bind");
73*86d7f5d3SJohn Marino return (1);
74*86d7f5d3SJohn Marino }
75*86d7f5d3SJohn Marino if (getenv("BLASTHOST") == NULL)
76*86d7f5d3SJohn Marino hostname = "localhost";
77*86d7f5d3SJohn Marino else
78*86d7f5d3SJohn Marino hostname = getenv("BLASTHOST");
79*86d7f5d3SJohn Marino host = gethostbyname(hostname);
80*86d7f5d3SJohn Marino if (host) {
81*86d7f5d3SJohn Marino sock_in.sin_family = host->h_addrtype;
82*86d7f5d3SJohn Marino bcopy(host->h_addr, &sock_in.sin_addr, host->h_length);
83*86d7f5d3SJohn Marino } else {
84*86d7f5d3SJohn Marino sock_in.sin_family = AF_INET;
85*86d7f5d3SJohn Marino sock_in.sin_addr.s_addr = inet_addr(hostname);
86*86d7f5d3SJohn Marino if (sock_in.sin_addr.s_addr == -1) {
87*86d7f5d3SJohn Marino err(1, "host: %s", hostname);
88*86d7f5d3SJohn Marino }
89*86d7f5d3SJohn Marino }
90*86d7f5d3SJohn Marino sock_in.sin_port = htons(9);
91*86d7f5d3SJohn Marino
92*86d7f5d3SJohn Marino if (connect(f, (struct sockaddr *)&sock_in, sizeof(sock_in)) < 0)
93*86d7f5d3SJohn Marino err(1, "connect");
94*86d7f5d3SJohn Marino
95*86d7f5d3SJohn Marino if ((buf = calloc(1, bufsize)) == NULL)
96*86d7f5d3SJohn Marino err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
97*86d7f5d3SJohn Marino
98*86d7f5d3SJohn Marino if (op->verbose > 1)
99*86d7f5d3SJohn Marino printf("udp %s:9 with %d bytes\n", hostname, bufsize);
100*86d7f5d3SJohn Marino for (i = 0; i < 128 && done_testing == 0; i++) {
101*86d7f5d3SJohn Marino n = write(f, buf, bufsize);
102*86d7f5d3SJohn Marino if (n == -1 && errno == ENOBUFS)
103*86d7f5d3SJohn Marino continue;
104*86d7f5d3SJohn Marino if (n == -1 && errno == ECONNREFUSED)
105*86d7f5d3SJohn Marino break;
106*86d7f5d3SJohn Marino if (n == -1)
107*86d7f5d3SJohn Marino err(1, "write(%d) #%d", bufsize, i);
108*86d7f5d3SJohn Marino if (n == 0) break;
109*86d7f5d3SJohn Marino }
110*86d7f5d3SJohn Marino free(buf);
111*86d7f5d3SJohn Marino close(f);
112*86d7f5d3SJohn Marino return (0);
113*86d7f5d3SJohn Marino }
114