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 <netdb.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <unistd.h>
35 #include <signal.h>
36 #include <errno.h>
37 #include <string.h>
38 #include <err.h>
39 
40 #include "stress.h"
41 
42 #define NB (400 * 1024 * 1024)
43 
44 int port;
45 int bufsize;
46 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, r;
68 
69 	if (r < 0)
70 		err(1, "connect(), %s:%d", __FILE__, __LINE__);
71 
72 	if ((buf = malloc(bufsize)) == NULL)
73 			err(1, "malloc(%d), %s:%d", bufsize, __FILE__, __LINE__);
74 	for (i = 0; i < bufsize / sizeof(int); i++)
75 		buf[i] = i;
76 
77 	for (;;) {
78 		for (i = 0; i < NB; i+= bufsize) {
79 			if (write(sv[1], buf, bufsize) < 0) {
80 				if (errno == EPIPE)
81 					return;
82 				err(1, "write(%d), %s:%d", sv[1],
83 						__FILE__, __LINE__);
84 			}
85 		}
86 	}
87 	return;
88 }
89 
90 int
91 setup(int nb)
92 {
93 	port = 12340 + nb;
94 	bufsize = 2 << random_int(2, 12);
95 	return (0);
96 }
97 
98 void
99 cleanup(void)
100 {
101 }
102 
103 int
104 test(void)
105 {
106 	pid_t pid;
107 
108 	if (socketpair(PF_UNIX, SOCK_STREAM, 0, sv) != 0)
109 		err(1, "socketpair()");
110 	if ((pid = fork()) == 0) {
111 		writer();
112 		exit(EXIT_SUCCESS);
113 
114 	} else if (pid > 0) {
115 		reader();
116 		kill(pid, SIGINT);
117 	} else
118 		err(1, "fork(), %s:%d",  __FILE__, __LINE__);
119 
120 	return (0);
121 }
122