1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* SCTP kernel Implementation
3 * Copyright (c) 2003 Hewlett-Packard Development Company, L.P
4 * (C) Copyright IBM Corp. 2004
5 * Copyright (c) 2019 Martin Doucha <[email protected]>
6 *
7 * When init timeout is set to zero, a connect () crashed the system. This case
8 * tests the fix for the same.
9 */
10
11 #include <stdio.h>
12 #include <unistd.h>
13 #include <fcntl.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <sys/types.h>
17 #include <sys/socket.h>
18 #include <netinet/in.h> /* for sockaddr_in */
19 #include <arpa/inet.h>
20 #include <netinet/sctp.h>
21 #include "tst_test.h"
22 #include "tst_net.h"
23
24 #ifdef PROT_SOCK
25 #define SCTP_TESTPORT_1 PROT_SOCK
26 #else
27 #define SCTP_TESTPORT_1 1024
28 #endif
29
30 #define SCTP_IP_LOOPBACK htonl(0x7f000001)
31
32 static const struct test_case {
33 __u16 streams;
34 int accept_err;
35 } testcase_list[] = {
36 {10, 0},
37 {65535, ENOMEM}
38 };
39
test_sctp(unsigned int n)40 static void test_sctp(unsigned int n)
41 {
42 int sk1, sk2, sk3, msglen;
43 socklen_t len;
44 struct sockaddr_in lstn_addr, acpt_addr;
45 char *buffer_rcv;
46 struct sctp_initmsg sinmsg;
47 const char *message = "Hello World!\n";
48 const struct test_case *tc = testcase_list + n;
49
50 tst_res(TINFO, "Running test with %u streams", tc->streams);
51
52 sk1 = SAFE_SOCKET(PF_INET, SOCK_STREAM, IPPROTO_SCTP);
53 sk3 = SAFE_SOCKET(PF_INET, SOCK_STREAM, IPPROTO_SCTP);
54
55 lstn_addr.sin_family = AF_INET;
56 lstn_addr.sin_addr.s_addr = SCTP_IP_LOOPBACK;
57 lstn_addr.sin_port = htons(SCTP_TESTPORT_1);
58
59 SAFE_BIND(sk3, (struct sockaddr *) &lstn_addr, sizeof(lstn_addr));
60
61 len = sizeof(struct sctp_initmsg);
62 sinmsg.sinit_num_ostreams = tc->streams;
63 sinmsg.sinit_max_instreams = 10;
64 sinmsg.sinit_max_attempts = 1;
65 sinmsg.sinit_max_init_timeo = 0;
66 SAFE_SETSOCKOPT(sk1, SOL_SCTP, SCTP_INITMSG, &sinmsg, len);
67 sinmsg.sinit_num_ostreams = 10;
68 sinmsg.sinit_max_instreams = tc->streams;
69 SAFE_SETSOCKOPT(sk3, SOL_SCTP, SCTP_INITMSG, &sinmsg, len);
70
71 SAFE_LISTEN(sk3, 1);
72
73 len = sizeof(struct sockaddr_in);
74 TEST(connect(sk1, (struct sockaddr *) &lstn_addr, len));
75
76 if (TST_RET == -1 && tc->accept_err && TST_ERR == tc->accept_err) {
77 tst_res(TPASS, "connect() failed in an acceptable way");
78 SAFE_CLOSE(sk1);
79 SAFE_CLOSE(sk3);
80 return;
81 } else if (TST_RET < 0) {
82 tst_brk(TBROK | TTERRNO, "connect() failed");
83 }
84
85 sk2 = SAFE_ACCEPT(sk3, (struct sockaddr *) &acpt_addr, &len);
86
87 msglen = strlen(message) + 1;
88 TEST(sctp_sendmsg(sk1, message, msglen, (struct sockaddr *)&lstn_addr,
89 len, 0, 0, tc->streams - 1, 0, 0));
90
91 if (TST_RET != msglen) {
92 tst_brk(TBROK | TTERRNO, "sctp_sendmsg() failed");
93 }
94
95 buffer_rcv = malloc(msglen);
96 TEST(recv(sk2, buffer_rcv, msglen, MSG_NOSIGNAL));
97
98 if (TST_RET != msglen || strncmp(buffer_rcv, message, msglen)) {
99 tst_res(TFAIL | TTERRNO, "recv() failed");
100 } else {
101 tst_res(TPASS, "connect() with init timeout set to 0");
102 }
103
104 free(buffer_rcv);
105 SAFE_CLOSE(sk1);
106 SAFE_CLOSE(sk2);
107 SAFE_CLOSE(sk3);
108 }
109
110 static struct tst_test test = {
111 .test = test_sctp,
112 .tcnt = ARRAY_SIZE(testcase_list),
113 .needs_drivers = (const char *[]) {
114 "sctp",
115 NULL
116 },
117 };
118