1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Check that writev on a socket that has been shutdown(2) fails
4 *
5 */
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <stdint.h>
9 #include <string.h>
10 #include <assert.h>
11
12 #include <errno.h>
13 #include <fcntl.h>
14 #include <unistd.h>
15 #include <sys/socket.h>
16 #include <sys/un.h>
17 #include <netinet/tcp.h>
18 #include <netinet/in.h>
19 #include <arpa/inet.h>
20
21 #include "liburing.h"
22
sig_pipe(int sig)23 static void sig_pipe(int sig)
24 {
25 }
26
main(int argc,char * argv[])27 int main(int argc, char *argv[])
28 {
29 int p_fd[2], ret;
30 int32_t recv_s0;
31 int32_t val = 1;
32 struct sockaddr_in addr;
33
34 if (argc > 1)
35 return 0;
36
37 srand(getpid());
38
39 recv_s0 = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
40
41 ret = setsockopt(recv_s0, SOL_SOCKET, SO_REUSEPORT, &val, sizeof(val));
42 assert(ret != -1);
43 ret = setsockopt(recv_s0, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
44 assert(ret != -1);
45
46 addr.sin_family = AF_INET;
47 addr.sin_port = htons((rand() % 61440) + 4096);
48 addr.sin_addr.s_addr = inet_addr("127.0.0.1");
49
50 ret = bind(recv_s0, (struct sockaddr*)&addr, sizeof(addr));
51 assert(ret != -1);
52 ret = listen(recv_s0, 128);
53 assert(ret != -1);
54
55 p_fd[1] = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, IPPROTO_TCP);
56
57 val = 1;
58 ret = setsockopt(p_fd[1], IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val));
59 assert(ret != -1);
60
61 int32_t flags = fcntl(p_fd[1], F_GETFL, 0);
62 assert(flags != -1);
63
64 flags |= O_NONBLOCK;
65 ret = fcntl(p_fd[1], F_SETFL, flags);
66 assert(ret != -1);
67
68 ret = connect(p_fd[1], (struct sockaddr*)&addr, sizeof(addr));
69 assert(ret == -1);
70
71 flags = fcntl(p_fd[1], F_GETFL, 0);
72 assert(flags != -1);
73
74 flags &= ~O_NONBLOCK;
75 ret = fcntl(p_fd[1], F_SETFL, flags);
76 assert(ret != -1);
77
78 p_fd[0] = accept(recv_s0, NULL, NULL);
79 assert(p_fd[0] != -1);
80
81 signal(SIGPIPE, sig_pipe);
82
83 while (1) {
84 int32_t code;
85 socklen_t code_len = sizeof(code);
86
87 ret = getsockopt(p_fd[1], SOL_SOCKET, SO_ERROR, &code, &code_len);
88 assert(ret != -1);
89
90 if (!code)
91 break;
92 }
93
94 struct io_uring m_io_uring;
95
96 ret = io_uring_queue_init(32, &m_io_uring, 0);
97 assert(ret >= 0);
98
99 {
100 struct io_uring_cqe *cqe;
101 struct io_uring_sqe *sqe;
102 int res;
103
104 sqe = io_uring_get_sqe(&m_io_uring);
105 io_uring_prep_shutdown(sqe, p_fd[1], SHUT_WR);
106 sqe->user_data = 1;
107
108 res = io_uring_submit_and_wait(&m_io_uring, 1);
109 assert(res != -1);
110
111 res = io_uring_wait_cqe(&m_io_uring, &cqe);
112 if (res < 0) {
113 fprintf(stderr, "wait: %s\n", strerror(-ret));
114 goto err;
115 }
116
117 if (cqe->res) {
118 if (cqe->res == -EINVAL) {
119 fprintf(stdout, "Shutdown not supported, skipping\n");
120 goto done;
121 }
122 fprintf(stderr, "writev: %d\n", cqe->res);
123 goto err;
124 }
125
126 io_uring_cqe_seen(&m_io_uring, cqe);
127 }
128
129 {
130 struct io_uring_cqe *cqe;
131 struct io_uring_sqe *sqe;
132 struct iovec iov[1];
133 char send_buff[128];
134 int res;
135
136 iov[0].iov_base = send_buff;
137 iov[0].iov_len = sizeof(send_buff);
138
139 sqe = io_uring_get_sqe(&m_io_uring);
140 assert(sqe != NULL);
141
142 io_uring_prep_writev(sqe, p_fd[1], iov, 1, 0);
143 res = io_uring_submit_and_wait(&m_io_uring, 1);
144 assert(res != -1);
145
146 res = io_uring_wait_cqe(&m_io_uring, &cqe);
147 if (res < 0) {
148 fprintf(stderr, "wait: %s\n", strerror(-ret));
149 goto err;
150 }
151
152 if (cqe->res != -EPIPE) {
153 fprintf(stderr, "writev: %d\n", cqe->res);
154 goto err;
155 }
156 io_uring_cqe_seen(&m_io_uring, cqe);
157 }
158
159 done:
160 io_uring_queue_exit(&m_io_uring);
161 return 0;
162 err:
163 io_uring_queue_exit(&m_io_uring);
164 return 1;
165 }
166