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, t, *buf;
51 
52 	t = 0;
53 	if ((buf = malloc(bufsize)) == NULL)
54 			err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
55 	while (done_testing == 0) {
56 		if ((n = read(sv[0], buf, bufsize)) < 0)
57 			err(1, "read(), %s:%d", __FILE__, __LINE__);
58 		t += n;
59 		if (n == 0) break;
60 	}
61 	close(sv[0]);
62 	return;
63 }
64 
65 static void
66 writer(void) {
67 	int i, *buf;
68 
69 	if ((buf = malloc(bufsize)) == NULL)
70 			err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
71 	for (i = 0; i < bufsize / (int)sizeof(int); i++)
72 		buf[i] = i;
73 
74 	for (;;) {
75 		for (i = 0; i < NB; i+= bufsize) {
76 			if (write(sv[1], buf, bufsize) < 0) {
77 				if (errno == EPIPE)
78 					return;
79 				err(1, "write(%d), %s:%d", sv[1],
80 						__FILE__, __LINE__);
81 			}
82 		}
83 	}
84 	return;
85 }
86 
87 int
88 setup(int nb)
89 {
90 	port = 12340 + nb;
91 	bufsize = 2 << random_int(2, 12);
92 	return (0);
93 }
94 
95 void
96 cleanup(void)
97 {
98 }
99 
100 int
101 test(void)
102 {
103 	pid_t pid;
104 
105 	if (socketpair(PF_UNIX, SOCK_STREAM, 0, sv) != 0)
106 		err(1, "socketpair()");
107 	if ((pid = fork()) == 0) {
108 		writer();
109 		_exit(EXIT_SUCCESS);
110 
111 	} else if (pid > 0) {
112 		reader();
113 		kill(pid, SIGINT);
114 	} else
115 		err(1, "fork(), %s:%d",  __FILE__, __LINE__);
116 
117 	return (0);
118 }
119