1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
#!/bin/sh
# Based on https://gist.github.com/zonque/7d03568eab14a2bb57cb by
# Daniel Mack github@zonque.org
# "panic: general protection fault" seen:
# https://people.freebsd.org/~pho/stress/log/sctp.txt
# Fixed by r350626
kldstat -v | grep -q sctp || kldload sctp.ko
cat > /tmp/sctp.c <<EOF
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <libgen.h>
#include <netinet/in.h>
#include <netinet/sctp.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
static int my_port_num;
static void
die(const char *s)
{
perror(s);
exit(1);
}
static void
server(void)
{
struct sctp_sndrcvinfo sndrcvinfo;
struct sockaddr_in servaddr = {
.sin_family = AF_INET,
.sin_addr.s_addr = htonl(INADDR_ANY),
.sin_port = htons(my_port_num),
};
struct sctp_initmsg initmsg = {
.sinit_num_ostreams = 5,
.sinit_max_instreams = 5,
.sinit_max_attempts = 4,
};
int listen_fd, conn_fd, flags, ret, in;
listen_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_SCTP);
if (listen_fd < 0)
die("socket");
ret = bind(listen_fd, (struct sockaddr *) &servaddr, sizeof(servaddr));
if (ret < 0)
die("bind");
ret = setsockopt(listen_fd, IPPROTO_SCTP, SCTP_INITMSG, &initmsg,
sizeof(initmsg));
if (ret < 0)
die("setsockopt");
ret = listen(listen_fd, initmsg.sinit_max_instreams);
if (ret < 0)
die("listen");
for (;;) {
char buffer[1024];
printf("Waiting for connection\n");
fflush(stdout);
conn_fd = accept(listen_fd, (struct sockaddr *) NULL, NULL);
if(conn_fd < 0)
die("accept()");
printf("New client connected\n");
fflush(stdout);
/* Note that flags is uninitialized here */
in = sctp_recvmsg(conn_fd, buffer, sizeof(buffer), NULL, 0,
&sndrcvinfo, &flags);
if (in > 0) {
printf("Received data: %s\n", buffer);
fflush(stdout);
}
close(conn_fd);
}
}
static void
client(void)
{
struct sockaddr_in servaddr = {
.sin_family = AF_INET,
.sin_port = htons(my_port_num),
.sin_addr.s_addr = inet_addr("127.0.0.1"),
};
int conn_fd, ret;
const char *msg = "Hello, Server!";
conn_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_SCTP);
if (conn_fd < 0)
die("socket()");
ret = connect(conn_fd, (struct sockaddr *) &servaddr, sizeof(servaddr));
if (ret < 0)
die("connect()");
ret = sctp_sendmsg(conn_fd, (void *) msg, strlen(msg) + 1, NULL, 0, 0, 0, 0, 0, 0 );
if (ret < 0)
die("sctp_sendmsg");
close(conn_fd);
}
int
main(int argc __unused, char *argv[])
{
my_port_num = atoi(argv[1]);
if (strstr(basename(argv[0]), "server"))
server();
else
client();
return (0);
}
EOF
cc -o /tmp/server -Wall -Wextra -O2 /tmp/sctp.c || exit
ln -sf /tmp/server /tmp/client
parallel=100
for i in `jot $parallel 62324`; do
/tmp/server $i > /dev/null &
done
(cd ../testcases/swap; ./swap -t 1m -i 20 -l 100) &
sleep 2
start=`date +%s`
while [ $((`date +%s` - start)) -lt 60 ]; do
pids=
for i in `jot 50`; do
for j in `jot $parallel 62324`; do
/tmp/client $j &
pids="$pids $!"
done
done
for i in $pids; do
wait $i
done
done
pkill server
wait
while pkill swap; do :; done
wait
rm -f /tmp/sctp.c /tmp/server /tmp/client
exit 0
|
