1 /*-
2  * Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  */
27 
28 #include <sys/param.h>
29 #include <sys/socket.h>
30 #include <netinet/in.h>
31 #include <err.h>
32 #include <errno.h>
33 #include <netdb.h>
34 #include <signal.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <unistd.h>
39 
40 #include "stress.h"
41 
42 #define NB (400 * 1024 * 1024)
43 
44 static int port;
45 static int bufsize;
46 static int sv[2];
47 
48 static void
49 reader(void) {
50 	int n, *buf;
51 
52 	if ((buf = malloc(bufsize)) == NULL)
53 			err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
54 	while (done_testing == 0) {
55 		if ((n = read(sv[0], buf, bufsize)) < 0)
56 			err(1, "read(), %s:%d", __FILE__, __LINE__);
57 		if (n == 0) break;
58 	}
59 	close(sv[0]);
60 	return;
61 }
62 
63 static void
64 writer(void) {
65 	int i, *buf;
66 
67 	if ((buf = malloc(bufsize)) == NULL)
68 			err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
69 	for (i = 0; i < bufsize / (int)sizeof(int); i++)
70 		buf[i] = i;
71 
72 	for (;;) {
73 		for (i = 0; i < NB; i+= bufsize) {
74 			if (write(sv[1], buf, bufsize) < 0) {
75 				if (errno == EPIPE)
76 					return;
77 				err(1, "write(%d), %s:%d", sv[1],
78 						__FILE__, __LINE__);
79 			}
80 		}
81 	}
82 	return;
83 }
84 
85 int
86 setup(int nb)
87 {
88 	port = 12340 + nb;
89 	bufsize = 2 << random_int(2, 12);
90 	return (0);
91 }
92 
93 void
94 cleanup(void)
95 {
96 }
97 
98 int
99 test(void)
100 {
101 	pid_t pid;
102 
103 	if (socketpair(PF_UNIX, SOCK_STREAM, 0, sv) != 0)
104 		err(1, "socketpair()");
105 	if ((pid = fork()) == 0) {
106 		writer();
107 		_exit(EXIT_SUCCESS);
108 
109 	} else if (pid > 0) {
110 		reader();
111 		kill(pid, SIGINT);
112 	} else
113 		err(1, "fork(), %s:%d",  __FILE__, __LINE__);
114 
115 	return (0);
116 }
117