Commit | Line | Data |
---|---|---|
ac713874 UB |
1 | /* |
2 | * Shared Memory Communications over RDMA (SMC-R) and RoCE | |
3 | * | |
4 | * AF_SMC protocol family socket handler keeping the AF_INET sock address type | |
5 | * applies to SOCK_STREAM sockets only | |
6 | * offers an alternative communication option for TCP-protocol sockets | |
7 | * applicable with RoCE-cards only | |
8 | * | |
a046d57d UB |
9 | * Initial restrictions: |
10 | * - non-blocking connect postponed | |
11 | * - IPv6 support postponed | |
12 | * - support for alternate links postponed | |
13 | * - partial support for non-blocking sockets only | |
14 | * - support for urgent data postponed | |
15 | * | |
ac713874 UB |
16 | * Copyright IBM Corp. 2016 |
17 | * | |
18 | * Author(s): Ursula Braun <ubraun@linux.vnet.ibm.com> | |
19 | * based on prototype from Frank Blaschka | |
20 | */ | |
21 | ||
22 | #define KMSG_COMPONENT "smc" | |
23 | #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt | |
24 | ||
25 | #include <linux/module.h> | |
26 | #include <linux/socket.h> | |
a046d57d UB |
27 | #include <linux/inetdevice.h> |
28 | #include <linux/workqueue.h> | |
5f08318f | 29 | #include <linux/in.h> |
c3edc401 IM |
30 | #include <linux/sched/signal.h> |
31 | ||
ac713874 | 32 | #include <net/sock.h> |
a046d57d | 33 | #include <net/tcp.h> |
f16a7dd5 | 34 | #include <net/smc.h> |
ac713874 UB |
35 | |
36 | #include "smc.h" | |
a046d57d | 37 | #include "smc_clc.h" |
9bf9abea | 38 | #include "smc_llc.h" |
5f08318f | 39 | #include "smc_cdc.h" |
0cfdd8f9 | 40 | #include "smc_core.h" |
a4cf0443 | 41 | #include "smc_ib.h" |
6812baab | 42 | #include "smc_pnet.h" |
e6727f39 | 43 | #include "smc_tx.h" |
952310cc | 44 | #include "smc_rx.h" |
b38d7324 | 45 | #include "smc_close.h" |
ac713874 | 46 | |
0cfdd8f9 UB |
47 | static DEFINE_MUTEX(smc_create_lgr_pending); /* serialize link group |
48 | * creation | |
49 | */ | |
50 | ||
51 | struct smc_lgr_list smc_lgr_list = { /* established link groups */ | |
52 | .lock = __SPIN_LOCK_UNLOCKED(smc_lgr_list.lock), | |
53 | .list = LIST_HEAD_INIT(smc_lgr_list.list), | |
54 | }; | |
55 | ||
a046d57d UB |
56 | static void smc_tcp_listen_work(struct work_struct *); |
57 | ||
ac713874 UB |
58 | static void smc_set_keepalive(struct sock *sk, int val) |
59 | { | |
60 | struct smc_sock *smc = smc_sk(sk); | |
61 | ||
62 | smc->clcsock->sk->sk_prot->keepalive(smc->clcsock->sk, val); | |
63 | } | |
64 | ||
f16a7dd5 UB |
65 | static struct smc_hashinfo smc_v4_hashinfo = { |
66 | .lock = __RW_LOCK_UNLOCKED(smc_v4_hashinfo.lock), | |
67 | }; | |
68 | ||
69 | int smc_hash_sk(struct sock *sk) | |
70 | { | |
71 | struct smc_hashinfo *h = sk->sk_prot->h.smc_hash; | |
72 | struct hlist_head *head; | |
73 | ||
74 | head = &h->ht; | |
75 | ||
76 | write_lock_bh(&h->lock); | |
77 | sk_add_node(sk, head); | |
78 | sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1); | |
79 | write_unlock_bh(&h->lock); | |
80 | ||
81 | return 0; | |
82 | } | |
83 | EXPORT_SYMBOL_GPL(smc_hash_sk); | |
84 | ||
85 | void smc_unhash_sk(struct sock *sk) | |
86 | { | |
87 | struct smc_hashinfo *h = sk->sk_prot->h.smc_hash; | |
88 | ||
89 | write_lock_bh(&h->lock); | |
90 | if (sk_del_node_init(sk)) | |
91 | sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1); | |
92 | write_unlock_bh(&h->lock); | |
93 | } | |
94 | EXPORT_SYMBOL_GPL(smc_unhash_sk); | |
95 | ||
96 | struct proto smc_proto = { | |
ac713874 UB |
97 | .name = "SMC", |
98 | .owner = THIS_MODULE, | |
99 | .keepalive = smc_set_keepalive, | |
f16a7dd5 UB |
100 | .hash = smc_hash_sk, |
101 | .unhash = smc_unhash_sk, | |
ac713874 | 102 | .obj_size = sizeof(struct smc_sock), |
f16a7dd5 | 103 | .h.smc_hash = &smc_v4_hashinfo, |
5f0d5a3a | 104 | .slab_flags = SLAB_TYPESAFE_BY_RCU, |
ac713874 | 105 | }; |
f16a7dd5 | 106 | EXPORT_SYMBOL_GPL(smc_proto); |
ac713874 UB |
107 | |
108 | static int smc_release(struct socket *sock) | |
109 | { | |
110 | struct sock *sk = sock->sk; | |
111 | struct smc_sock *smc; | |
b38d7324 | 112 | int rc = 0; |
ac713874 UB |
113 | |
114 | if (!sk) | |
115 | goto out; | |
116 | ||
117 | smc = smc_sk(sk); | |
b38d7324 UB |
118 | sock_hold(sk); |
119 | if (sk->sk_state == SMC_LISTEN) | |
120 | /* smc_close_non_accepted() is called and acquires | |
121 | * sock lock for child sockets again | |
122 | */ | |
123 | lock_sock_nested(sk, SINGLE_DEPTH_NESTING); | |
124 | else | |
125 | lock_sock(sk); | |
ac713874 | 126 | |
b38d7324 UB |
127 | if (smc->use_fallback) { |
128 | sk->sk_state = SMC_CLOSED; | |
129 | sk->sk_state_change(sk); | |
130 | } else { | |
131 | rc = smc_close_active(smc); | |
132 | sock_set_flag(sk, SOCK_DEAD); | |
133 | sk->sk_shutdown |= SHUTDOWN_MASK; | |
134 | } | |
ac713874 UB |
135 | if (smc->clcsock) { |
136 | sock_release(smc->clcsock); | |
137 | smc->clcsock = NULL; | |
138 | } | |
139 | ||
140 | /* detach socket */ | |
141 | sock_orphan(sk); | |
142 | sock->sk = NULL; | |
b38d7324 UB |
143 | if (smc->use_fallback) { |
144 | schedule_delayed_work(&smc->sock_put_work, TCP_TIMEWAIT_LEN); | |
145 | } else if (sk->sk_state == SMC_CLOSED) { | |
146 | smc_conn_free(&smc->conn); | |
147 | schedule_delayed_work(&smc->sock_put_work, | |
148 | SMC_CLOSE_SOCK_PUT_DELAY); | |
149 | } | |
ac713874 UB |
150 | release_sock(sk); |
151 | ||
152 | sock_put(sk); | |
153 | out: | |
b38d7324 | 154 | return rc; |
ac713874 UB |
155 | } |
156 | ||
157 | static void smc_destruct(struct sock *sk) | |
158 | { | |
159 | if (sk->sk_state != SMC_CLOSED) | |
160 | return; | |
161 | if (!sock_flag(sk, SOCK_DEAD)) | |
162 | return; | |
163 | ||
164 | sk_refcnt_debug_dec(sk); | |
165 | } | |
166 | ||
167 | static struct sock *smc_sock_alloc(struct net *net, struct socket *sock) | |
168 | { | |
169 | struct smc_sock *smc; | |
170 | struct sock *sk; | |
171 | ||
172 | sk = sk_alloc(net, PF_SMC, GFP_KERNEL, &smc_proto, 0); | |
173 | if (!sk) | |
174 | return NULL; | |
175 | ||
176 | sock_init_data(sock, sk); /* sets sk_refcnt to 1 */ | |
177 | sk->sk_state = SMC_INIT; | |
178 | sk->sk_destruct = smc_destruct; | |
179 | sk->sk_protocol = SMCPROTO_SMC; | |
ac713874 | 180 | smc = smc_sk(sk); |
a046d57d UB |
181 | INIT_WORK(&smc->tcp_listen_work, smc_tcp_listen_work); |
182 | INIT_LIST_HEAD(&smc->accept_q); | |
183 | spin_lock_init(&smc->accept_q_lock); | |
b38d7324 | 184 | INIT_DELAYED_WORK(&smc->sock_put_work, smc_close_sock_put_work); |
f16a7dd5 | 185 | sk->sk_prot->hash(sk); |
a046d57d | 186 | sk_refcnt_debug_inc(sk); |
ac713874 UB |
187 | |
188 | return sk; | |
189 | } | |
190 | ||
191 | static int smc_bind(struct socket *sock, struct sockaddr *uaddr, | |
192 | int addr_len) | |
193 | { | |
194 | struct sockaddr_in *addr = (struct sockaddr_in *)uaddr; | |
195 | struct sock *sk = sock->sk; | |
196 | struct smc_sock *smc; | |
197 | int rc; | |
198 | ||
199 | smc = smc_sk(sk); | |
200 | ||
201 | /* replicate tests from inet_bind(), to be safe wrt. future changes */ | |
202 | rc = -EINVAL; | |
203 | if (addr_len < sizeof(struct sockaddr_in)) | |
204 | goto out; | |
205 | ||
206 | rc = -EAFNOSUPPORT; | |
207 | /* accept AF_UNSPEC (mapped to AF_INET) only if s_addr is INADDR_ANY */ | |
208 | if ((addr->sin_family != AF_INET) && | |
209 | ((addr->sin_family != AF_UNSPEC) || | |
210 | (addr->sin_addr.s_addr != htonl(INADDR_ANY)))) | |
211 | goto out; | |
212 | ||
213 | lock_sock(sk); | |
214 | ||
215 | /* Check if socket is already active */ | |
216 | rc = -EINVAL; | |
217 | if (sk->sk_state != SMC_INIT) | |
218 | goto out_rel; | |
219 | ||
220 | smc->clcsock->sk->sk_reuse = sk->sk_reuse; | |
221 | rc = kernel_bind(smc->clcsock, uaddr, addr_len); | |
222 | ||
223 | out_rel: | |
224 | release_sock(sk); | |
225 | out: | |
226 | return rc; | |
227 | } | |
228 | ||
229 | static void smc_copy_sock_settings(struct sock *nsk, struct sock *osk, | |
230 | unsigned long mask) | |
231 | { | |
232 | /* options we don't get control via setsockopt for */ | |
233 | nsk->sk_type = osk->sk_type; | |
234 | nsk->sk_sndbuf = osk->sk_sndbuf; | |
235 | nsk->sk_rcvbuf = osk->sk_rcvbuf; | |
236 | nsk->sk_sndtimeo = osk->sk_sndtimeo; | |
237 | nsk->sk_rcvtimeo = osk->sk_rcvtimeo; | |
238 | nsk->sk_mark = osk->sk_mark; | |
239 | nsk->sk_priority = osk->sk_priority; | |
240 | nsk->sk_rcvlowat = osk->sk_rcvlowat; | |
241 | nsk->sk_bound_dev_if = osk->sk_bound_dev_if; | |
242 | nsk->sk_err = osk->sk_err; | |
243 | ||
244 | nsk->sk_flags &= ~mask; | |
245 | nsk->sk_flags |= osk->sk_flags & mask; | |
246 | } | |
247 | ||
248 | #define SK_FLAGS_SMC_TO_CLC ((1UL << SOCK_URGINLINE) | \ | |
249 | (1UL << SOCK_KEEPOPEN) | \ | |
250 | (1UL << SOCK_LINGER) | \ | |
251 | (1UL << SOCK_BROADCAST) | \ | |
252 | (1UL << SOCK_TIMESTAMP) | \ | |
253 | (1UL << SOCK_DBG) | \ | |
254 | (1UL << SOCK_RCVTSTAMP) | \ | |
255 | (1UL << SOCK_RCVTSTAMPNS) | \ | |
256 | (1UL << SOCK_LOCALROUTE) | \ | |
257 | (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE) | \ | |
258 | (1UL << SOCK_RXQ_OVFL) | \ | |
259 | (1UL << SOCK_WIFI_STATUS) | \ | |
260 | (1UL << SOCK_NOFCS) | \ | |
261 | (1UL << SOCK_FILTER_LOCKED)) | |
262 | /* copy only relevant settings and flags of SOL_SOCKET level from smc to | |
263 | * clc socket (since smc is not called for these options from net/core) | |
264 | */ | |
265 | static void smc_copy_sock_settings_to_clc(struct smc_sock *smc) | |
266 | { | |
267 | smc_copy_sock_settings(smc->clcsock->sk, &smc->sk, SK_FLAGS_SMC_TO_CLC); | |
268 | } | |
269 | ||
270 | #define SK_FLAGS_CLC_TO_SMC ((1UL << SOCK_URGINLINE) | \ | |
271 | (1UL << SOCK_KEEPOPEN) | \ | |
272 | (1UL << SOCK_LINGER) | \ | |
273 | (1UL << SOCK_DBG)) | |
274 | /* copy only settings and flags relevant for smc from clc to smc socket */ | |
275 | static void smc_copy_sock_settings_to_smc(struct smc_sock *smc) | |
276 | { | |
277 | smc_copy_sock_settings(&smc->sk, smc->clcsock->sk, SK_FLAGS_CLC_TO_SMC); | |
278 | } | |
279 | ||
a046d57d UB |
280 | /* determine subnet and mask of internal TCP socket */ |
281 | int smc_netinfo_by_tcpsk(struct socket *clcsock, | |
282 | __be32 *subnet, u8 *prefix_len) | |
283 | { | |
284 | struct dst_entry *dst = sk_dst_get(clcsock->sk); | |
731b0085 | 285 | struct in_device *in_dev; |
a046d57d UB |
286 | struct sockaddr_in addr; |
287 | int rc = -ENOENT; | |
288 | int len; | |
289 | ||
290 | if (!dst) { | |
291 | rc = -ENOTCONN; | |
292 | goto out; | |
293 | } | |
294 | if (!dst->dev) { | |
295 | rc = -ENODEV; | |
296 | goto out_rel; | |
297 | } | |
298 | ||
299 | /* get address to which the internal TCP socket is bound */ | |
300 | kernel_getsockname(clcsock, (struct sockaddr *)&addr, &len); | |
301 | /* analyze IPv4 specific data of net_device belonging to TCP socket */ | |
731b0085 UB |
302 | rcu_read_lock(); |
303 | in_dev = __in_dev_get_rcu(dst->dev); | |
304 | for_ifa(in_dev) { | |
305 | if (!inet_ifa_match(addr.sin_addr.s_addr, ifa)) | |
a046d57d UB |
306 | continue; |
307 | *prefix_len = inet_mask_len(ifa->ifa_mask); | |
308 | *subnet = ifa->ifa_address & ifa->ifa_mask; | |
309 | rc = 0; | |
310 | break; | |
731b0085 UB |
311 | } endfor_ifa(in_dev); |
312 | rcu_read_unlock(); | |
a046d57d UB |
313 | |
314 | out_rel: | |
315 | dst_release(dst); | |
316 | out: | |
317 | return rc; | |
318 | } | |
319 | ||
9bf9abea UB |
320 | static int smc_clnt_conf_first_link(struct smc_sock *smc, union ib_gid *gid) |
321 | { | |
322 | struct smc_link_group *lgr = smc->conn.lgr; | |
323 | struct smc_link *link; | |
324 | int rest; | |
325 | int rc; | |
326 | ||
327 | link = &lgr->lnk[SMC_SINGLE_LINK]; | |
328 | /* receive CONFIRM LINK request from server over RoCE fabric */ | |
329 | rest = wait_for_completion_interruptible_timeout( | |
330 | &link->llc_confirm, | |
331 | SMC_LLC_WAIT_FIRST_TIME); | |
332 | if (rest <= 0) { | |
333 | struct smc_clc_msg_decline dclc; | |
334 | ||
335 | rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc), | |
336 | SMC_CLC_DECLINE); | |
337 | return rc; | |
338 | } | |
339 | ||
340 | rc = smc_ib_modify_qp_rts(link); | |
341 | if (rc) | |
342 | return SMC_CLC_DECL_INTERR; | |
343 | ||
344 | smc_wr_remember_qp_attr(link); | |
652a1e41 UB |
345 | |
346 | rc = smc_wr_reg_send(link, | |
347 | smc->conn.rmb_desc->mr_rx[SMC_SINGLE_LINK]); | |
348 | if (rc) | |
349 | return SMC_CLC_DECL_INTERR; | |
350 | ||
9bf9abea UB |
351 | /* send CONFIRM LINK response over RoCE fabric */ |
352 | rc = smc_llc_send_confirm_link(link, | |
353 | link->smcibdev->mac[link->ibport - 1], | |
354 | gid, SMC_LLC_RESP); | |
355 | if (rc < 0) | |
356 | return SMC_CLC_DECL_TCL; | |
357 | ||
358 | return rc; | |
359 | } | |
360 | ||
0cfdd8f9 UB |
361 | static void smc_conn_save_peer_info(struct smc_sock *smc, |
362 | struct smc_clc_msg_accept_confirm *clc) | |
363 | { | |
364 | smc->conn.peer_conn_idx = clc->conn_idx; | |
5f08318f | 365 | smc->conn.local_tx_ctrl.token = ntohl(clc->rmbe_alert_token); |
cd6851f3 UB |
366 | smc->conn.peer_rmbe_size = smc_uncompress_bufsize(clc->rmbe_size); |
367 | atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size); | |
0cfdd8f9 UB |
368 | } |
369 | ||
370 | static void smc_link_save_peer_info(struct smc_link *link, | |
371 | struct smc_clc_msg_accept_confirm *clc) | |
372 | { | |
373 | link->peer_qpn = ntoh24(clc->qpn); | |
374 | memcpy(link->peer_gid, clc->lcl.gid, SMC_GID_SIZE); | |
375 | memcpy(link->peer_mac, clc->lcl.mac, sizeof(link->peer_mac)); | |
376 | link->peer_psn = ntoh24(clc->psn); | |
377 | link->peer_mtu = clc->qp_mtu; | |
378 | } | |
379 | ||
610db66f UB |
380 | static void smc_lgr_forget(struct smc_link_group *lgr) |
381 | { | |
382 | spin_lock_bh(&smc_lgr_list.lock); | |
383 | /* do not use this link group for new connections */ | |
384 | if (!list_empty(&lgr->list)) | |
385 | list_del_init(&lgr->list); | |
386 | spin_unlock_bh(&smc_lgr_list.lock); | |
387 | } | |
388 | ||
a046d57d UB |
389 | /* setup for RDMA connection of client */ |
390 | static int smc_connect_rdma(struct smc_sock *smc) | |
391 | { | |
0cfdd8f9 | 392 | struct sockaddr_in *inaddr = (struct sockaddr_in *)smc->addr; |
a046d57d | 393 | struct smc_clc_msg_accept_confirm aclc; |
0cfdd8f9 | 394 | int local_contact = SMC_FIRST_CONTACT; |
a046d57d | 395 | struct smc_ib_device *smcibdev; |
0cfdd8f9 UB |
396 | struct smc_link *link; |
397 | u8 srv_first_contact; | |
a046d57d UB |
398 | int reason_code = 0; |
399 | int rc = 0; | |
400 | u8 ibport; | |
401 | ||
c5c1cc9c UB |
402 | if (!tcp_sk(smc->clcsock->sk)->syn_smc) { |
403 | /* peer has not signalled SMC-capability */ | |
404 | smc->use_fallback = true; | |
405 | goto out_connected; | |
406 | } | |
407 | ||
a046d57d UB |
408 | /* IPSec connections opt out of SMC-R optimizations */ |
409 | if (using_ipsec(smc)) { | |
410 | reason_code = SMC_CLC_DECL_IPSEC; | |
411 | goto decline_rdma; | |
412 | } | |
413 | ||
414 | /* PNET table look up: search active ib_device and port | |
415 | * within same PNETID that also contains the ethernet device | |
416 | * used for the internal TCP socket | |
417 | */ | |
418 | smc_pnet_find_roce_resource(smc->clcsock->sk, &smcibdev, &ibport); | |
419 | if (!smcibdev) { | |
420 | reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */ | |
421 | goto decline_rdma; | |
422 | } | |
423 | ||
424 | /* do inband token exchange */ | |
425 | reason_code = smc_clc_send_proposal(smc, smcibdev, ibport); | |
426 | if (reason_code < 0) { | |
427 | rc = reason_code; | |
428 | goto out_err; | |
429 | } | |
430 | if (reason_code > 0) /* configuration error */ | |
431 | goto decline_rdma; | |
432 | /* receive SMC Accept CLC message */ | |
433 | reason_code = smc_clc_wait_msg(smc, &aclc, sizeof(aclc), | |
434 | SMC_CLC_ACCEPT); | |
435 | if (reason_code < 0) { | |
436 | rc = reason_code; | |
437 | goto out_err; | |
438 | } | |
439 | if (reason_code > 0) | |
440 | goto decline_rdma; | |
441 | ||
0cfdd8f9 UB |
442 | srv_first_contact = aclc.hdr.flag; |
443 | mutex_lock(&smc_create_lgr_pending); | |
444 | local_contact = smc_conn_create(smc, inaddr->sin_addr.s_addr, smcibdev, | |
445 | ibport, &aclc.lcl, srv_first_contact); | |
446 | if (local_contact < 0) { | |
447 | rc = local_contact; | |
448 | if (rc == -ENOMEM) | |
449 | reason_code = SMC_CLC_DECL_MEM;/* insufficient memory*/ | |
450 | else if (rc == -ENOLINK) | |
451 | reason_code = SMC_CLC_DECL_SYNCERR; /* synchr. error */ | |
452 | goto decline_rdma_unlock; | |
453 | } | |
454 | link = &smc->conn.lgr->lnk[SMC_SINGLE_LINK]; | |
a046d57d | 455 | |
0cfdd8f9 | 456 | smc_conn_save_peer_info(smc, &aclc); |
cd6851f3 | 457 | |
3e034725 UB |
458 | /* create send buffer and rmb */ |
459 | rc = smc_buf_create(smc); | |
cd6851f3 UB |
460 | if (rc) { |
461 | reason_code = SMC_CLC_DECL_MEM; | |
462 | goto decline_rdma_unlock; | |
463 | } | |
464 | ||
0cfdd8f9 UB |
465 | if (local_contact == SMC_FIRST_CONTACT) |
466 | smc_link_save_peer_info(link, &aclc); | |
bd4ad577 UB |
467 | |
468 | rc = smc_rmb_rtoken_handling(&smc->conn, &aclc); | |
469 | if (rc) { | |
470 | reason_code = SMC_CLC_DECL_INTERR; | |
471 | goto decline_rdma_unlock; | |
472 | } | |
473 | ||
46c28dbd UB |
474 | smc_close_init(smc); |
475 | smc_rx_init(smc); | |
476 | ||
bd4ad577 UB |
477 | if (local_contact == SMC_FIRST_CONTACT) { |
478 | rc = smc_ib_ready_link(link); | |
479 | if (rc) { | |
480 | reason_code = SMC_CLC_DECL_INTERR; | |
481 | goto decline_rdma_unlock; | |
482 | } | |
652a1e41 UB |
483 | } else { |
484 | struct smc_buf_desc *buf_desc = smc->conn.rmb_desc; | |
485 | ||
486 | if (!buf_desc->reused) { | |
487 | /* register memory region for new rmb */ | |
488 | rc = smc_wr_reg_send(link, | |
489 | buf_desc->mr_rx[SMC_SINGLE_LINK]); | |
490 | if (rc) { | |
491 | reason_code = SMC_CLC_DECL_INTERR; | |
492 | goto decline_rdma_unlock; | |
493 | } | |
494 | } | |
bd4ad577 | 495 | } |
10428dd8 | 496 | smc_rmb_sync_sg_for_device(&smc->conn); |
a046d57d UB |
497 | |
498 | rc = smc_clc_send_confirm(smc); | |
499 | if (rc) | |
0cfdd8f9 | 500 | goto out_err_unlock; |
a046d57d | 501 | |
9bf9abea UB |
502 | if (local_contact == SMC_FIRST_CONTACT) { |
503 | /* QP confirmation over RoCE fabric */ | |
504 | reason_code = smc_clnt_conf_first_link( | |
505 | smc, &smcibdev->gid[ibport - 1]); | |
506 | if (reason_code < 0) { | |
507 | rc = reason_code; | |
508 | goto out_err_unlock; | |
509 | } | |
510 | if (reason_code > 0) | |
511 | goto decline_rdma_unlock; | |
512 | } | |
a046d57d | 513 | |
0cfdd8f9 | 514 | mutex_unlock(&smc_create_lgr_pending); |
e6727f39 UB |
515 | smc_tx_init(smc); |
516 | ||
a046d57d UB |
517 | out_connected: |
518 | smc_copy_sock_settings_to_clc(smc); | |
b38d7324 UB |
519 | if (smc->sk.sk_state == SMC_INIT) |
520 | smc->sk.sk_state = SMC_ACTIVE; | |
a046d57d | 521 | |
0cfdd8f9 | 522 | return rc ? rc : local_contact; |
a046d57d | 523 | |
0cfdd8f9 | 524 | decline_rdma_unlock: |
610db66f UB |
525 | if (local_contact == SMC_FIRST_CONTACT) |
526 | smc_lgr_forget(smc->conn.lgr); | |
0cfdd8f9 UB |
527 | mutex_unlock(&smc_create_lgr_pending); |
528 | smc_conn_free(&smc->conn); | |
a046d57d UB |
529 | decline_rdma: |
530 | /* RDMA setup failed, switch back to TCP */ | |
531 | smc->use_fallback = true; | |
532 | if (reason_code && (reason_code != SMC_CLC_DECL_REPLY)) { | |
bfbedfd3 | 533 | rc = smc_clc_send_decline(smc, reason_code); |
0c9f1515 | 534 | if (rc < 0) |
a046d57d UB |
535 | goto out_err; |
536 | } | |
537 | goto out_connected; | |
538 | ||
0cfdd8f9 | 539 | out_err_unlock: |
610db66f UB |
540 | if (local_contact == SMC_FIRST_CONTACT) |
541 | smc_lgr_forget(smc->conn.lgr); | |
0cfdd8f9 UB |
542 | mutex_unlock(&smc_create_lgr_pending); |
543 | smc_conn_free(&smc->conn); | |
a046d57d UB |
544 | out_err: |
545 | return rc; | |
546 | } | |
547 | ||
ac713874 UB |
548 | static int smc_connect(struct socket *sock, struct sockaddr *addr, |
549 | int alen, int flags) | |
550 | { | |
551 | struct sock *sk = sock->sk; | |
552 | struct smc_sock *smc; | |
553 | int rc = -EINVAL; | |
554 | ||
555 | smc = smc_sk(sk); | |
556 | ||
557 | /* separate smc parameter checking to be safe */ | |
558 | if (alen < sizeof(addr->sa_family)) | |
559 | goto out_err; | |
560 | if (addr->sa_family != AF_INET) | |
561 | goto out_err; | |
a046d57d | 562 | smc->addr = addr; /* needed for nonblocking connect */ |
ac713874 UB |
563 | |
564 | lock_sock(sk); | |
565 | switch (sk->sk_state) { | |
566 | default: | |
567 | goto out; | |
568 | case SMC_ACTIVE: | |
569 | rc = -EISCONN; | |
570 | goto out; | |
571 | case SMC_INIT: | |
572 | rc = 0; | |
573 | break; | |
574 | } | |
575 | ||
576 | smc_copy_sock_settings_to_clc(smc); | |
c5c1cc9c | 577 | tcp_sk(smc->clcsock->sk)->syn_smc = 1; |
ac713874 UB |
578 | rc = kernel_connect(smc->clcsock, addr, alen, flags); |
579 | if (rc) | |
580 | goto out; | |
581 | ||
a046d57d UB |
582 | /* setup RDMA connection */ |
583 | rc = smc_connect_rdma(smc); | |
584 | if (rc < 0) | |
585 | goto out; | |
586 | else | |
587 | rc = 0; /* success cases including fallback */ | |
ac713874 UB |
588 | |
589 | out: | |
590 | release_sock(sk); | |
591 | out_err: | |
592 | return rc; | |
593 | } | |
594 | ||
595 | static int smc_clcsock_accept(struct smc_sock *lsmc, struct smc_sock **new_smc) | |
596 | { | |
3163c507 UB |
597 | struct socket *new_clcsock = NULL; |
598 | struct sock *lsk = &lsmc->sk; | |
ac713874 UB |
599 | struct sock *new_sk; |
600 | int rc; | |
601 | ||
3163c507 UB |
602 | release_sock(lsk); |
603 | new_sk = smc_sock_alloc(sock_net(lsk), NULL); | |
ac713874 UB |
604 | if (!new_sk) { |
605 | rc = -ENOMEM; | |
3163c507 | 606 | lsk->sk_err = ENOMEM; |
ac713874 | 607 | *new_smc = NULL; |
3163c507 | 608 | lock_sock(lsk); |
ac713874 UB |
609 | goto out; |
610 | } | |
611 | *new_smc = smc_sk(new_sk); | |
612 | ||
613 | rc = kernel_accept(lsmc->clcsock, &new_clcsock, 0); | |
3163c507 | 614 | lock_sock(lsk); |
35a6b178 | 615 | if (rc < 0) |
3163c507 | 616 | lsk->sk_err = -rc; |
35a6b178 | 617 | if (rc < 0 || lsk->sk_state == SMC_CLOSED) { |
a046d57d UB |
618 | if (new_clcsock) |
619 | sock_release(new_clcsock); | |
620 | new_sk->sk_state = SMC_CLOSED; | |
621 | sock_set_flag(new_sk, SOCK_DEAD); | |
3163c507 | 622 | new_sk->sk_prot->unhash(new_sk); |
ac713874 UB |
623 | sock_put(new_sk); |
624 | *new_smc = NULL; | |
625 | goto out; | |
626 | } | |
627 | ||
628 | (*new_smc)->clcsock = new_clcsock; | |
629 | out: | |
630 | return rc; | |
631 | } | |
632 | ||
a046d57d UB |
633 | /* add a just created sock to the accept queue of the listen sock as |
634 | * candidate for a following socket accept call from user space | |
635 | */ | |
636 | static void smc_accept_enqueue(struct sock *parent, struct sock *sk) | |
637 | { | |
638 | struct smc_sock *par = smc_sk(parent); | |
639 | ||
640 | sock_hold(sk); | |
641 | spin_lock(&par->accept_q_lock); | |
642 | list_add_tail(&smc_sk(sk)->accept_q, &par->accept_q); | |
643 | spin_unlock(&par->accept_q_lock); | |
644 | sk_acceptq_added(parent); | |
645 | } | |
646 | ||
647 | /* remove a socket from the accept queue of its parental listening socket */ | |
648 | static void smc_accept_unlink(struct sock *sk) | |
649 | { | |
650 | struct smc_sock *par = smc_sk(sk)->listen_smc; | |
651 | ||
652 | spin_lock(&par->accept_q_lock); | |
653 | list_del_init(&smc_sk(sk)->accept_q); | |
654 | spin_unlock(&par->accept_q_lock); | |
655 | sk_acceptq_removed(&smc_sk(sk)->listen_smc->sk); | |
656 | sock_put(sk); | |
657 | } | |
658 | ||
659 | /* remove a sock from the accept queue to bind it to a new socket created | |
660 | * for a socket accept call from user space | |
661 | */ | |
b38d7324 UB |
662 | struct sock *smc_accept_dequeue(struct sock *parent, |
663 | struct socket *new_sock) | |
a046d57d UB |
664 | { |
665 | struct smc_sock *isk, *n; | |
666 | struct sock *new_sk; | |
667 | ||
668 | list_for_each_entry_safe(isk, n, &smc_sk(parent)->accept_q, accept_q) { | |
669 | new_sk = (struct sock *)isk; | |
670 | ||
671 | smc_accept_unlink(new_sk); | |
672 | if (new_sk->sk_state == SMC_CLOSED) { | |
288c8390 UB |
673 | new_sk->sk_prot->unhash(new_sk); |
674 | sock_put(new_sk); | |
a046d57d UB |
675 | continue; |
676 | } | |
677 | if (new_sock) | |
678 | sock_graft(new_sk, new_sock); | |
679 | return new_sk; | |
680 | } | |
681 | return NULL; | |
682 | } | |
683 | ||
684 | /* clean up for a created but never accepted sock */ | |
b38d7324 | 685 | void smc_close_non_accepted(struct sock *sk) |
a046d57d UB |
686 | { |
687 | struct smc_sock *smc = smc_sk(sk); | |
688 | ||
689 | sock_hold(sk); | |
b38d7324 UB |
690 | lock_sock(sk); |
691 | if (!sk->sk_lingertime) | |
692 | /* wait for peer closing */ | |
693 | sk->sk_lingertime = SMC_MAX_STREAM_WAIT_TIMEOUT; | |
288c8390 UB |
694 | if (smc->use_fallback) { |
695 | sk->sk_state = SMC_CLOSED; | |
696 | } else { | |
b38d7324 | 697 | smc_close_active(smc); |
288c8390 UB |
698 | sock_set_flag(sk, SOCK_DEAD); |
699 | sk->sk_shutdown |= SHUTDOWN_MASK; | |
700 | } | |
a046d57d UB |
701 | if (smc->clcsock) { |
702 | struct socket *tcp; | |
703 | ||
704 | tcp = smc->clcsock; | |
705 | smc->clcsock = NULL; | |
706 | sock_release(tcp); | |
707 | } | |
b38d7324 UB |
708 | if (smc->use_fallback) { |
709 | schedule_delayed_work(&smc->sock_put_work, TCP_TIMEWAIT_LEN); | |
288c8390 | 710 | } else if (sk->sk_state == SMC_CLOSED) { |
b38d7324 UB |
711 | smc_conn_free(&smc->conn); |
712 | schedule_delayed_work(&smc->sock_put_work, | |
713 | SMC_CLOSE_SOCK_PUT_DELAY); | |
714 | } | |
715 | release_sock(sk); | |
a046d57d UB |
716 | sock_put(sk); |
717 | } | |
718 | ||
9bf9abea UB |
719 | static int smc_serv_conf_first_link(struct smc_sock *smc) |
720 | { | |
721 | struct smc_link_group *lgr = smc->conn.lgr; | |
722 | struct smc_link *link; | |
723 | int rest; | |
724 | int rc; | |
725 | ||
726 | link = &lgr->lnk[SMC_SINGLE_LINK]; | |
652a1e41 UB |
727 | |
728 | rc = smc_wr_reg_send(link, | |
729 | smc->conn.rmb_desc->mr_rx[SMC_SINGLE_LINK]); | |
730 | if (rc) | |
731 | return SMC_CLC_DECL_INTERR; | |
732 | ||
9bf9abea UB |
733 | /* send CONFIRM LINK request to client over the RoCE fabric */ |
734 | rc = smc_llc_send_confirm_link(link, | |
735 | link->smcibdev->mac[link->ibport - 1], | |
736 | &link->smcibdev->gid[link->ibport - 1], | |
737 | SMC_LLC_REQ); | |
738 | if (rc < 0) | |
739 | return SMC_CLC_DECL_TCL; | |
740 | ||
741 | /* receive CONFIRM LINK response from client over the RoCE fabric */ | |
742 | rest = wait_for_completion_interruptible_timeout( | |
743 | &link->llc_confirm_resp, | |
744 | SMC_LLC_WAIT_FIRST_TIME); | |
745 | if (rest <= 0) { | |
746 | struct smc_clc_msg_decline dclc; | |
747 | ||
748 | rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc), | |
749 | SMC_CLC_DECLINE); | |
750 | } | |
751 | ||
752 | return rc; | |
753 | } | |
754 | ||
a046d57d UB |
755 | /* setup for RDMA connection of server */ |
756 | static void smc_listen_work(struct work_struct *work) | |
757 | { | |
758 | struct smc_sock *new_smc = container_of(work, struct smc_sock, | |
759 | smc_listen_work); | |
e7b7a64a | 760 | struct smc_clc_msg_proposal_prefix *pclc_prfx; |
a046d57d UB |
761 | struct socket *newclcsock = new_smc->clcsock; |
762 | struct smc_sock *lsmc = new_smc->listen_smc; | |
763 | struct smc_clc_msg_accept_confirm cclc; | |
0cfdd8f9 | 764 | int local_contact = SMC_REUSE_CONTACT; |
a046d57d | 765 | struct sock *newsmcsk = &new_smc->sk; |
e7b7a64a | 766 | struct smc_clc_msg_proposal *pclc; |
a046d57d UB |
767 | struct smc_ib_device *smcibdev; |
768 | struct sockaddr_in peeraddr; | |
e7b7a64a | 769 | u8 buf[SMC_CLC_MAX_LEN]; |
0cfdd8f9 | 770 | struct smc_link *link; |
a046d57d UB |
771 | int reason_code = 0; |
772 | int rc = 0, len; | |
773 | __be32 subnet; | |
774 | u8 prefix_len; | |
775 | u8 ibport; | |
776 | ||
c5c1cc9c UB |
777 | /* check if peer is smc capable */ |
778 | if (!tcp_sk(newclcsock->sk)->syn_smc) { | |
779 | new_smc->use_fallback = true; | |
780 | goto out_connected; | |
781 | } | |
782 | ||
a046d57d UB |
783 | /* do inband token exchange - |
784 | *wait for and receive SMC Proposal CLC message | |
785 | */ | |
e7b7a64a | 786 | reason_code = smc_clc_wait_msg(new_smc, &buf, sizeof(buf), |
a046d57d UB |
787 | SMC_CLC_PROPOSAL); |
788 | if (reason_code < 0) | |
789 | goto out_err; | |
790 | if (reason_code > 0) | |
791 | goto decline_rdma; | |
792 | ||
793 | /* IPSec connections opt out of SMC-R optimizations */ | |
794 | if (using_ipsec(new_smc)) { | |
795 | reason_code = SMC_CLC_DECL_IPSEC; | |
796 | goto decline_rdma; | |
797 | } | |
798 | ||
799 | /* PNET table look up: search active ib_device and port | |
800 | * within same PNETID that also contains the ethernet device | |
801 | * used for the internal TCP socket | |
802 | */ | |
803 | smc_pnet_find_roce_resource(newclcsock->sk, &smcibdev, &ibport); | |
804 | if (!smcibdev) { | |
805 | reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */ | |
806 | goto decline_rdma; | |
807 | } | |
808 | ||
809 | /* determine subnet and mask from internal TCP socket */ | |
810 | rc = smc_netinfo_by_tcpsk(newclcsock, &subnet, &prefix_len); | |
811 | if (rc) { | |
812 | reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */ | |
813 | goto decline_rdma; | |
814 | } | |
e7b7a64a UB |
815 | |
816 | pclc = (struct smc_clc_msg_proposal *)&buf; | |
817 | pclc_prfx = smc_clc_proposal_get_prefix(pclc); | |
818 | if (pclc_prfx->outgoing_subnet != subnet || | |
819 | pclc_prfx->prefix_len != prefix_len) { | |
a046d57d UB |
820 | reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */ |
821 | goto decline_rdma; | |
822 | } | |
823 | ||
824 | /* get address of the peer connected to the internal TCP socket */ | |
825 | kernel_getpeername(newclcsock, (struct sockaddr *)&peeraddr, &len); | |
826 | ||
0cfdd8f9 UB |
827 | /* allocate connection / link group */ |
828 | mutex_lock(&smc_create_lgr_pending); | |
829 | local_contact = smc_conn_create(new_smc, peeraddr.sin_addr.s_addr, | |
e7b7a64a | 830 | smcibdev, ibport, &pclc->lcl, 0); |
0cfdd8f9 UB |
831 | if (local_contact < 0) { |
832 | rc = local_contact; | |
833 | if (rc == -ENOMEM) | |
834 | reason_code = SMC_CLC_DECL_MEM;/* insufficient memory*/ | |
145686ba | 835 | goto decline_rdma_unlock; |
0cfdd8f9 UB |
836 | } |
837 | link = &new_smc->conn.lgr->lnk[SMC_SINGLE_LINK]; | |
a046d57d | 838 | |
3e034725 UB |
839 | /* create send buffer and rmb */ |
840 | rc = smc_buf_create(new_smc); | |
cd6851f3 UB |
841 | if (rc) { |
842 | reason_code = SMC_CLC_DECL_MEM; | |
145686ba | 843 | goto decline_rdma_unlock; |
cd6851f3 | 844 | } |
a046d57d | 845 | |
46c28dbd UB |
846 | smc_close_init(new_smc); |
847 | smc_rx_init(new_smc); | |
848 | ||
652a1e41 UB |
849 | if (local_contact != SMC_FIRST_CONTACT) { |
850 | struct smc_buf_desc *buf_desc = new_smc->conn.rmb_desc; | |
851 | ||
852 | if (!buf_desc->reused) { | |
853 | /* register memory region for new rmb */ | |
854 | rc = smc_wr_reg_send(link, | |
855 | buf_desc->mr_rx[SMC_SINGLE_LINK]); | |
856 | if (rc) { | |
857 | reason_code = SMC_CLC_DECL_INTERR; | |
145686ba | 858 | goto decline_rdma_unlock; |
652a1e41 UB |
859 | } |
860 | } | |
861 | } | |
10428dd8 | 862 | smc_rmb_sync_sg_for_device(&new_smc->conn); |
652a1e41 | 863 | |
0cfdd8f9 | 864 | rc = smc_clc_send_accept(new_smc, local_contact); |
a046d57d | 865 | if (rc) |
145686ba | 866 | goto out_err_unlock; |
a046d57d UB |
867 | |
868 | /* receive SMC Confirm CLC message */ | |
869 | reason_code = smc_clc_wait_msg(new_smc, &cclc, sizeof(cclc), | |
870 | SMC_CLC_CONFIRM); | |
871 | if (reason_code < 0) | |
145686ba | 872 | goto out_err_unlock; |
a046d57d | 873 | if (reason_code > 0) |
145686ba | 874 | goto decline_rdma_unlock; |
0cfdd8f9 UB |
875 | smc_conn_save_peer_info(new_smc, &cclc); |
876 | if (local_contact == SMC_FIRST_CONTACT) | |
877 | smc_link_save_peer_info(link, &cclc); | |
a046d57d | 878 | |
bd4ad577 UB |
879 | rc = smc_rmb_rtoken_handling(&new_smc->conn, &cclc); |
880 | if (rc) { | |
881 | reason_code = SMC_CLC_DECL_INTERR; | |
145686ba | 882 | goto decline_rdma_unlock; |
bd4ad577 UB |
883 | } |
884 | ||
bd4ad577 UB |
885 | if (local_contact == SMC_FIRST_CONTACT) { |
886 | rc = smc_ib_ready_link(link); | |
887 | if (rc) { | |
888 | reason_code = SMC_CLC_DECL_INTERR; | |
145686ba | 889 | goto decline_rdma_unlock; |
bd4ad577 | 890 | } |
9bf9abea UB |
891 | /* QP confirmation over RoCE fabric */ |
892 | reason_code = smc_serv_conf_first_link(new_smc); | |
0c9f1515 | 893 | if (reason_code < 0) |
9bf9abea | 894 | /* peer is not aware of a problem */ |
145686ba | 895 | goto out_err_unlock; |
9bf9abea | 896 | if (reason_code > 0) |
145686ba | 897 | goto decline_rdma_unlock; |
bd4ad577 | 898 | } |
a046d57d | 899 | |
e6727f39 | 900 | smc_tx_init(new_smc); |
145686ba | 901 | mutex_unlock(&smc_create_lgr_pending); |
e6727f39 | 902 | |
a046d57d UB |
903 | out_connected: |
904 | sk_refcnt_debug_inc(newsmcsk); | |
b38d7324 UB |
905 | if (newsmcsk->sk_state == SMC_INIT) |
906 | newsmcsk->sk_state = SMC_ACTIVE; | |
a046d57d | 907 | enqueue: |
b38d7324 | 908 | lock_sock_nested(&lsmc->sk, SINGLE_DEPTH_NESTING); |
a046d57d UB |
909 | if (lsmc->sk.sk_state == SMC_LISTEN) { |
910 | smc_accept_enqueue(&lsmc->sk, newsmcsk); | |
911 | } else { /* no longer listening */ | |
912 | smc_close_non_accepted(newsmcsk); | |
913 | } | |
914 | release_sock(&lsmc->sk); | |
915 | ||
916 | /* Wake up accept */ | |
917 | lsmc->sk.sk_data_ready(&lsmc->sk); | |
918 | sock_put(&lsmc->sk); /* sock_hold in smc_tcp_listen_work */ | |
919 | return; | |
920 | ||
145686ba | 921 | decline_rdma_unlock: |
610db66f UB |
922 | if (local_contact == SMC_FIRST_CONTACT) |
923 | smc_lgr_forget(new_smc->conn.lgr); | |
145686ba | 924 | mutex_unlock(&smc_create_lgr_pending); |
a046d57d UB |
925 | decline_rdma: |
926 | /* RDMA setup failed, switch back to TCP */ | |
0cfdd8f9 | 927 | smc_conn_free(&new_smc->conn); |
a046d57d UB |
928 | new_smc->use_fallback = true; |
929 | if (reason_code && (reason_code != SMC_CLC_DECL_REPLY)) { | |
0c9f1515 | 930 | if (smc_clc_send_decline(new_smc, reason_code) < 0) |
a046d57d UB |
931 | goto out_err; |
932 | } | |
933 | goto out_connected; | |
934 | ||
145686ba | 935 | out_err_unlock: |
610db66f UB |
936 | if (local_contact == SMC_FIRST_CONTACT) |
937 | smc_lgr_forget(new_smc->conn.lgr); | |
145686ba | 938 | mutex_unlock(&smc_create_lgr_pending); |
a046d57d UB |
939 | out_err: |
940 | newsmcsk->sk_state = SMC_CLOSED; | |
b38d7324 | 941 | smc_conn_free(&new_smc->conn); |
a046d57d UB |
942 | goto enqueue; /* queue new sock with sk_err set */ |
943 | } | |
944 | ||
945 | static void smc_tcp_listen_work(struct work_struct *work) | |
946 | { | |
947 | struct smc_sock *lsmc = container_of(work, struct smc_sock, | |
948 | tcp_listen_work); | |
3163c507 | 949 | struct sock *lsk = &lsmc->sk; |
a046d57d UB |
950 | struct smc_sock *new_smc; |
951 | int rc = 0; | |
952 | ||
3163c507 UB |
953 | lock_sock(lsk); |
954 | while (lsk->sk_state == SMC_LISTEN) { | |
a046d57d UB |
955 | rc = smc_clcsock_accept(lsmc, &new_smc); |
956 | if (rc) | |
957 | goto out; | |
958 | if (!new_smc) | |
959 | continue; | |
960 | ||
961 | new_smc->listen_smc = lsmc; | |
962 | new_smc->use_fallback = false; /* assume rdma capability first*/ | |
3163c507 | 963 | sock_hold(lsk); /* sock_put in smc_listen_work */ |
a046d57d UB |
964 | INIT_WORK(&new_smc->smc_listen_work, smc_listen_work); |
965 | smc_copy_sock_settings_to_smc(new_smc); | |
966 | schedule_work(&new_smc->smc_listen_work); | |
967 | } | |
968 | ||
969 | out: | |
3163c507 UB |
970 | release_sock(lsk); |
971 | lsk->sk_data_ready(lsk); /* no more listening, wake accept */ | |
a046d57d UB |
972 | } |
973 | ||
ac713874 UB |
974 | static int smc_listen(struct socket *sock, int backlog) |
975 | { | |
976 | struct sock *sk = sock->sk; | |
977 | struct smc_sock *smc; | |
978 | int rc; | |
979 | ||
980 | smc = smc_sk(sk); | |
981 | lock_sock(sk); | |
982 | ||
983 | rc = -EINVAL; | |
984 | if ((sk->sk_state != SMC_INIT) && (sk->sk_state != SMC_LISTEN)) | |
985 | goto out; | |
986 | ||
987 | rc = 0; | |
988 | if (sk->sk_state == SMC_LISTEN) { | |
989 | sk->sk_max_ack_backlog = backlog; | |
990 | goto out; | |
991 | } | |
992 | /* some socket options are handled in core, so we could not apply | |
993 | * them to the clc socket -- copy smc socket options to clc socket | |
994 | */ | |
995 | smc_copy_sock_settings_to_clc(smc); | |
c5c1cc9c | 996 | tcp_sk(smc->clcsock->sk)->syn_smc = 1; |
ac713874 UB |
997 | |
998 | rc = kernel_listen(smc->clcsock, backlog); | |
999 | if (rc) | |
1000 | goto out; | |
1001 | sk->sk_max_ack_backlog = backlog; | |
1002 | sk->sk_ack_backlog = 0; | |
1003 | sk->sk_state = SMC_LISTEN; | |
a046d57d UB |
1004 | INIT_WORK(&smc->tcp_listen_work, smc_tcp_listen_work); |
1005 | schedule_work(&smc->tcp_listen_work); | |
ac713874 UB |
1006 | |
1007 | out: | |
1008 | release_sock(sk); | |
1009 | return rc; | |
1010 | } | |
1011 | ||
1012 | static int smc_accept(struct socket *sock, struct socket *new_sock, | |
cdfbabfb | 1013 | int flags, bool kern) |
ac713874 | 1014 | { |
a046d57d UB |
1015 | struct sock *sk = sock->sk, *nsk; |
1016 | DECLARE_WAITQUEUE(wait, current); | |
ac713874 | 1017 | struct smc_sock *lsmc; |
a046d57d UB |
1018 | long timeo; |
1019 | int rc = 0; | |
ac713874 UB |
1020 | |
1021 | lsmc = smc_sk(sk); | |
1022 | lock_sock(sk); | |
1023 | ||
1024 | if (lsmc->sk.sk_state != SMC_LISTEN) { | |
1025 | rc = -EINVAL; | |
1026 | goto out; | |
1027 | } | |
1028 | ||
a046d57d UB |
1029 | /* Wait for an incoming connection */ |
1030 | timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK); | |
1031 | add_wait_queue_exclusive(sk_sleep(sk), &wait); | |
1032 | while (!(nsk = smc_accept_dequeue(sk, new_sock))) { | |
1033 | set_current_state(TASK_INTERRUPTIBLE); | |
1034 | if (!timeo) { | |
1035 | rc = -EAGAIN; | |
1036 | break; | |
1037 | } | |
1038 | release_sock(sk); | |
1039 | timeo = schedule_timeout(timeo); | |
1040 | /* wakeup by sk_data_ready in smc_listen_work() */ | |
1041 | sched_annotate_sleep(); | |
1042 | lock_sock(sk); | |
1043 | if (signal_pending(current)) { | |
1044 | rc = sock_intr_errno(timeo); | |
1045 | break; | |
1046 | } | |
1047 | } | |
1048 | set_current_state(TASK_RUNNING); | |
1049 | remove_wait_queue(sk_sleep(sk), &wait); | |
ac713874 | 1050 | |
a046d57d UB |
1051 | if (!rc) |
1052 | rc = sock_error(nsk); | |
ac713874 UB |
1053 | |
1054 | out: | |
1055 | release_sock(sk); | |
1056 | return rc; | |
1057 | } | |
1058 | ||
1059 | static int smc_getname(struct socket *sock, struct sockaddr *addr, | |
1060 | int *len, int peer) | |
1061 | { | |
1062 | struct smc_sock *smc; | |
1063 | ||
b38d7324 UB |
1064 | if (peer && (sock->sk->sk_state != SMC_ACTIVE) && |
1065 | (sock->sk->sk_state != SMC_APPCLOSEWAIT1)) | |
ac713874 UB |
1066 | return -ENOTCONN; |
1067 | ||
1068 | smc = smc_sk(sock->sk); | |
1069 | ||
1070 | return smc->clcsock->ops->getname(smc->clcsock, addr, len, peer); | |
1071 | } | |
1072 | ||
1073 | static int smc_sendmsg(struct socket *sock, struct msghdr *msg, size_t len) | |
1074 | { | |
1075 | struct sock *sk = sock->sk; | |
1076 | struct smc_sock *smc; | |
1077 | int rc = -EPIPE; | |
1078 | ||
1079 | smc = smc_sk(sk); | |
1080 | lock_sock(sk); | |
b38d7324 UB |
1081 | if ((sk->sk_state != SMC_ACTIVE) && |
1082 | (sk->sk_state != SMC_APPCLOSEWAIT1) && | |
1083 | (sk->sk_state != SMC_INIT)) | |
ac713874 UB |
1084 | goto out; |
1085 | if (smc->use_fallback) | |
1086 | rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len); | |
1087 | else | |
e6727f39 | 1088 | rc = smc_tx_sendmsg(smc, msg, len); |
ac713874 UB |
1089 | out: |
1090 | release_sock(sk); | |
1091 | return rc; | |
1092 | } | |
1093 | ||
1094 | static int smc_recvmsg(struct socket *sock, struct msghdr *msg, size_t len, | |
1095 | int flags) | |
1096 | { | |
1097 | struct sock *sk = sock->sk; | |
1098 | struct smc_sock *smc; | |
1099 | int rc = -ENOTCONN; | |
1100 | ||
1101 | smc = smc_sk(sk); | |
1102 | lock_sock(sk); | |
b38d7324 UB |
1103 | if ((sk->sk_state == SMC_INIT) || |
1104 | (sk->sk_state == SMC_LISTEN) || | |
1105 | (sk->sk_state == SMC_CLOSED)) | |
ac713874 UB |
1106 | goto out; |
1107 | ||
b38d7324 UB |
1108 | if (sk->sk_state == SMC_PEERFINCLOSEWAIT) { |
1109 | rc = 0; | |
1110 | goto out; | |
1111 | } | |
1112 | ||
ac713874 UB |
1113 | if (smc->use_fallback) |
1114 | rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags); | |
1115 | else | |
952310cc | 1116 | rc = smc_rx_recvmsg(smc, msg, len, flags); |
b38d7324 | 1117 | |
ac713874 UB |
1118 | out: |
1119 | release_sock(sk); | |
1120 | return rc; | |
1121 | } | |
1122 | ||
a046d57d UB |
1123 | static unsigned int smc_accept_poll(struct sock *parent) |
1124 | { | |
8dce2786 UB |
1125 | struct smc_sock *isk = smc_sk(parent); |
1126 | int mask = 0; | |
a046d57d | 1127 | |
8dce2786 UB |
1128 | spin_lock(&isk->accept_q_lock); |
1129 | if (!list_empty(&isk->accept_q)) | |
1130 | mask = POLLIN | POLLRDNORM; | |
1131 | spin_unlock(&isk->accept_q_lock); | |
a046d57d | 1132 | |
8dce2786 | 1133 | return mask; |
a046d57d UB |
1134 | } |
1135 | ||
ac713874 UB |
1136 | static unsigned int smc_poll(struct file *file, struct socket *sock, |
1137 | poll_table *wait) | |
1138 | { | |
1139 | struct sock *sk = sock->sk; | |
1140 | unsigned int mask = 0; | |
1141 | struct smc_sock *smc; | |
a046d57d | 1142 | int rc; |
ac713874 | 1143 | |
8dce2786 UB |
1144 | if (!sk) |
1145 | return POLLNVAL; | |
1146 | ||
ac713874 | 1147 | smc = smc_sk(sock->sk); |
8dce2786 UB |
1148 | sock_hold(sk); |
1149 | lock_sock(sk); | |
a046d57d UB |
1150 | if ((sk->sk_state == SMC_INIT) || smc->use_fallback) { |
1151 | /* delegate to CLC child sock */ | |
8dce2786 | 1152 | release_sock(sk); |
ac713874 UB |
1153 | mask = smc->clcsock->ops->poll(file, smc->clcsock, wait); |
1154 | /* if non-blocking connect finished ... */ | |
1155 | lock_sock(sk); | |
1156 | if ((sk->sk_state == SMC_INIT) && (mask & POLLOUT)) { | |
a046d57d UB |
1157 | sk->sk_err = smc->clcsock->sk->sk_err; |
1158 | if (sk->sk_err) { | |
1159 | mask |= POLLERR; | |
1160 | } else { | |
1161 | rc = smc_connect_rdma(smc); | |
1162 | if (rc < 0) | |
1163 | mask |= POLLERR; | |
8dce2786 UB |
1164 | /* success cases including fallback */ |
1165 | mask |= POLLOUT | POLLWRNORM; | |
a046d57d | 1166 | } |
ac713874 | 1167 | } |
ac713874 | 1168 | } else { |
8dce2786 UB |
1169 | if (sk->sk_state != SMC_CLOSED) { |
1170 | release_sock(sk); | |
1171 | sock_poll_wait(file, sk_sleep(sk), wait); | |
1172 | lock_sock(sk); | |
1173 | } | |
a046d57d UB |
1174 | if (sk->sk_err) |
1175 | mask |= POLLERR; | |
b38d7324 UB |
1176 | if ((sk->sk_shutdown == SHUTDOWN_MASK) || |
1177 | (sk->sk_state == SMC_CLOSED)) | |
1178 | mask |= POLLHUP; | |
8dce2786 UB |
1179 | if (sk->sk_state == SMC_LISTEN) { |
1180 | /* woken up by sk_data_ready in smc_listen_work() */ | |
1181 | mask = smc_accept_poll(sk); | |
1182 | } else { | |
1183 | if (atomic_read(&smc->conn.sndbuf_space) || | |
1184 | sk->sk_shutdown & SEND_SHUTDOWN) { | |
1185 | mask |= POLLOUT | POLLWRNORM; | |
1186 | } else { | |
1187 | sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk); | |
1188 | set_bit(SOCK_NOSPACE, &sk->sk_socket->flags); | |
1189 | } | |
1190 | if (atomic_read(&smc->conn.bytes_to_rcv)) | |
1191 | mask |= POLLIN | POLLRDNORM; | |
1192 | if (sk->sk_shutdown & RCV_SHUTDOWN) | |
1193 | mask |= POLLIN | POLLRDNORM | POLLRDHUP; | |
1194 | if (sk->sk_state == SMC_APPCLOSEWAIT1) | |
1195 | mask |= POLLIN; | |
1196 | } | |
b38d7324 | 1197 | |
ac713874 | 1198 | } |
8dce2786 UB |
1199 | release_sock(sk); |
1200 | sock_put(sk); | |
ac713874 UB |
1201 | |
1202 | return mask; | |
1203 | } | |
1204 | ||
1205 | static int smc_shutdown(struct socket *sock, int how) | |
1206 | { | |
1207 | struct sock *sk = sock->sk; | |
1208 | struct smc_sock *smc; | |
1209 | int rc = -EINVAL; | |
b38d7324 | 1210 | int rc1 = 0; |
ac713874 UB |
1211 | |
1212 | smc = smc_sk(sk); | |
1213 | ||
1214 | if ((how < SHUT_RD) || (how > SHUT_RDWR)) | |
b38d7324 | 1215 | return rc; |
ac713874 UB |
1216 | |
1217 | lock_sock(sk); | |
1218 | ||
1219 | rc = -ENOTCONN; | |
b38d7324 UB |
1220 | if ((sk->sk_state != SMC_LISTEN) && |
1221 | (sk->sk_state != SMC_ACTIVE) && | |
1222 | (sk->sk_state != SMC_PEERCLOSEWAIT1) && | |
1223 | (sk->sk_state != SMC_PEERCLOSEWAIT2) && | |
1224 | (sk->sk_state != SMC_APPCLOSEWAIT1) && | |
1225 | (sk->sk_state != SMC_APPCLOSEWAIT2) && | |
1226 | (sk->sk_state != SMC_APPFINCLOSEWAIT)) | |
ac713874 UB |
1227 | goto out; |
1228 | if (smc->use_fallback) { | |
1229 | rc = kernel_sock_shutdown(smc->clcsock, how); | |
1230 | sk->sk_shutdown = smc->clcsock->sk->sk_shutdown; | |
1231 | if (sk->sk_shutdown == SHUTDOWN_MASK) | |
1232 | sk->sk_state = SMC_CLOSED; | |
b38d7324 | 1233 | goto out; |
ac713874 | 1234 | } |
b38d7324 UB |
1235 | switch (how) { |
1236 | case SHUT_RDWR: /* shutdown in both directions */ | |
1237 | rc = smc_close_active(smc); | |
1238 | break; | |
1239 | case SHUT_WR: | |
1240 | rc = smc_close_shutdown_write(smc); | |
1241 | break; | |
1242 | case SHUT_RD: | |
1243 | if (sk->sk_state == SMC_LISTEN) | |
1244 | rc = smc_close_active(smc); | |
1245 | else | |
1246 | rc = 0; | |
1247 | /* nothing more to do because peer is not involved */ | |
1248 | break; | |
1249 | } | |
1250 | rc1 = kernel_sock_shutdown(smc->clcsock, how); | |
1251 | /* map sock_shutdown_cmd constants to sk_shutdown value range */ | |
1252 | sk->sk_shutdown |= how + 1; | |
ac713874 UB |
1253 | |
1254 | out: | |
1255 | release_sock(sk); | |
b38d7324 | 1256 | return rc ? rc : rc1; |
ac713874 UB |
1257 | } |
1258 | ||
1259 | static int smc_setsockopt(struct socket *sock, int level, int optname, | |
1260 | char __user *optval, unsigned int optlen) | |
1261 | { | |
1262 | struct sock *sk = sock->sk; | |
1263 | struct smc_sock *smc; | |
1264 | ||
1265 | smc = smc_sk(sk); | |
1266 | ||
1267 | /* generic setsockopts reaching us here always apply to the | |
1268 | * CLC socket | |
1269 | */ | |
1270 | return smc->clcsock->ops->setsockopt(smc->clcsock, level, optname, | |
1271 | optval, optlen); | |
1272 | } | |
1273 | ||
1274 | static int smc_getsockopt(struct socket *sock, int level, int optname, | |
1275 | char __user *optval, int __user *optlen) | |
1276 | { | |
1277 | struct smc_sock *smc; | |
1278 | ||
1279 | smc = smc_sk(sock->sk); | |
1280 | /* socket options apply to the CLC socket */ | |
1281 | return smc->clcsock->ops->getsockopt(smc->clcsock, level, optname, | |
1282 | optval, optlen); | |
1283 | } | |
1284 | ||
1285 | static int smc_ioctl(struct socket *sock, unsigned int cmd, | |
1286 | unsigned long arg) | |
1287 | { | |
1288 | struct smc_sock *smc; | |
1289 | ||
1290 | smc = smc_sk(sock->sk); | |
1291 | if (smc->use_fallback) | |
1292 | return smc->clcsock->ops->ioctl(smc->clcsock, cmd, arg); | |
1293 | else | |
1294 | return sock_no_ioctl(sock, cmd, arg); | |
1295 | } | |
1296 | ||
1297 | static ssize_t smc_sendpage(struct socket *sock, struct page *page, | |
1298 | int offset, size_t size, int flags) | |
1299 | { | |
1300 | struct sock *sk = sock->sk; | |
1301 | struct smc_sock *smc; | |
1302 | int rc = -EPIPE; | |
1303 | ||
1304 | smc = smc_sk(sk); | |
1305 | lock_sock(sk); | |
1306 | if (sk->sk_state != SMC_ACTIVE) | |
1307 | goto out; | |
1308 | if (smc->use_fallback) | |
1309 | rc = kernel_sendpage(smc->clcsock, page, offset, | |
1310 | size, flags); | |
1311 | else | |
1312 | rc = sock_no_sendpage(sock, page, offset, size, flags); | |
1313 | ||
1314 | out: | |
1315 | release_sock(sk); | |
1316 | return rc; | |
1317 | } | |
1318 | ||
1319 | static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos, | |
1320 | struct pipe_inode_info *pipe, size_t len, | |
1321 | unsigned int flags) | |
1322 | { | |
1323 | struct sock *sk = sock->sk; | |
1324 | struct smc_sock *smc; | |
1325 | int rc = -ENOTCONN; | |
1326 | ||
1327 | smc = smc_sk(sk); | |
1328 | lock_sock(sk); | |
1329 | if ((sk->sk_state != SMC_ACTIVE) && (sk->sk_state != SMC_CLOSED)) | |
1330 | goto out; | |
1331 | if (smc->use_fallback) { | |
1332 | rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos, | |
1333 | pipe, len, flags); | |
1334 | } else { | |
1335 | rc = -EOPNOTSUPP; | |
1336 | } | |
1337 | out: | |
1338 | release_sock(sk); | |
1339 | return rc; | |
1340 | } | |
1341 | ||
1342 | /* must look like tcp */ | |
1343 | static const struct proto_ops smc_sock_ops = { | |
1344 | .family = PF_SMC, | |
1345 | .owner = THIS_MODULE, | |
1346 | .release = smc_release, | |
1347 | .bind = smc_bind, | |
1348 | .connect = smc_connect, | |
1349 | .socketpair = sock_no_socketpair, | |
1350 | .accept = smc_accept, | |
1351 | .getname = smc_getname, | |
1352 | .poll = smc_poll, | |
1353 | .ioctl = smc_ioctl, | |
1354 | .listen = smc_listen, | |
1355 | .shutdown = smc_shutdown, | |
1356 | .setsockopt = smc_setsockopt, | |
1357 | .getsockopt = smc_getsockopt, | |
1358 | .sendmsg = smc_sendmsg, | |
1359 | .recvmsg = smc_recvmsg, | |
1360 | .mmap = sock_no_mmap, | |
1361 | .sendpage = smc_sendpage, | |
1362 | .splice_read = smc_splice_read, | |
1363 | }; | |
1364 | ||
1365 | static int smc_create(struct net *net, struct socket *sock, int protocol, | |
1366 | int kern) | |
1367 | { | |
1368 | struct smc_sock *smc; | |
1369 | struct sock *sk; | |
1370 | int rc; | |
1371 | ||
1372 | rc = -ESOCKTNOSUPPORT; | |
1373 | if (sock->type != SOCK_STREAM) | |
1374 | goto out; | |
1375 | ||
1376 | rc = -EPROTONOSUPPORT; | |
1377 | if ((protocol != IPPROTO_IP) && (protocol != IPPROTO_TCP)) | |
1378 | goto out; | |
1379 | ||
1380 | rc = -ENOBUFS; | |
1381 | sock->ops = &smc_sock_ops; | |
1382 | sk = smc_sock_alloc(net, sock); | |
1383 | if (!sk) | |
1384 | goto out; | |
1385 | ||
1386 | /* create internal TCP socket for CLC handshake and fallback */ | |
1387 | smc = smc_sk(sk); | |
a046d57d | 1388 | smc->use_fallback = false; /* assume rdma capability first */ |
ac713874 UB |
1389 | rc = sock_create_kern(net, PF_INET, SOCK_STREAM, |
1390 | IPPROTO_TCP, &smc->clcsock); | |
1391 | if (rc) | |
1392 | sk_common_release(sk); | |
cd6851f3 UB |
1393 | smc->sk.sk_sndbuf = max(smc->clcsock->sk->sk_sndbuf, SMC_BUF_MIN_SIZE); |
1394 | smc->sk.sk_rcvbuf = max(smc->clcsock->sk->sk_rcvbuf, SMC_BUF_MIN_SIZE); | |
ac713874 UB |
1395 | |
1396 | out: | |
1397 | return rc; | |
1398 | } | |
1399 | ||
1400 | static const struct net_proto_family smc_sock_family_ops = { | |
1401 | .family = PF_SMC, | |
1402 | .owner = THIS_MODULE, | |
1403 | .create = smc_create, | |
1404 | }; | |
1405 | ||
1406 | static int __init smc_init(void) | |
1407 | { | |
1408 | int rc; | |
1409 | ||
6812baab TR |
1410 | rc = smc_pnet_init(); |
1411 | if (rc) | |
1412 | return rc; | |
1413 | ||
9bf9abea UB |
1414 | rc = smc_llc_init(); |
1415 | if (rc) { | |
1416 | pr_err("%s: smc_llc_init fails with %d\n", __func__, rc); | |
1417 | goto out_pnet; | |
1418 | } | |
1419 | ||
5f08318f UB |
1420 | rc = smc_cdc_init(); |
1421 | if (rc) { | |
1422 | pr_err("%s: smc_cdc_init fails with %d\n", __func__, rc); | |
1423 | goto out_pnet; | |
1424 | } | |
1425 | ||
ac713874 UB |
1426 | rc = proto_register(&smc_proto, 1); |
1427 | if (rc) { | |
1428 | pr_err("%s: proto_register fails with %d\n", __func__, rc); | |
6812baab | 1429 | goto out_pnet; |
ac713874 UB |
1430 | } |
1431 | ||
1432 | rc = sock_register(&smc_sock_family_ops); | |
1433 | if (rc) { | |
1434 | pr_err("%s: sock_register fails with %d\n", __func__, rc); | |
1435 | goto out_proto; | |
1436 | } | |
f16a7dd5 | 1437 | INIT_HLIST_HEAD(&smc_v4_hashinfo.ht); |
ac713874 | 1438 | |
a4cf0443 UB |
1439 | rc = smc_ib_register_client(); |
1440 | if (rc) { | |
1441 | pr_err("%s: ib_register fails with %d\n", __func__, rc); | |
1442 | goto out_sock; | |
1443 | } | |
1444 | ||
c5c1cc9c | 1445 | static_branch_enable(&tcp_have_smc); |
ac713874 UB |
1446 | return 0; |
1447 | ||
a4cf0443 UB |
1448 | out_sock: |
1449 | sock_unregister(PF_SMC); | |
ac713874 UB |
1450 | out_proto: |
1451 | proto_unregister(&smc_proto); | |
6812baab TR |
1452 | out_pnet: |
1453 | smc_pnet_exit(); | |
ac713874 UB |
1454 | return rc; |
1455 | } | |
1456 | ||
1457 | static void __exit smc_exit(void) | |
1458 | { | |
0cfdd8f9 UB |
1459 | struct smc_link_group *lgr, *lg; |
1460 | LIST_HEAD(lgr_freeing_list); | |
1461 | ||
1462 | spin_lock_bh(&smc_lgr_list.lock); | |
1463 | if (!list_empty(&smc_lgr_list.list)) | |
1464 | list_splice_init(&smc_lgr_list.list, &lgr_freeing_list); | |
1465 | spin_unlock_bh(&smc_lgr_list.lock); | |
1466 | list_for_each_entry_safe(lgr, lg, &lgr_freeing_list, list) { | |
1467 | list_del_init(&lgr->list); | |
1468 | smc_lgr_free(lgr); /* free link group */ | |
1469 | } | |
c5c1cc9c | 1470 | static_branch_disable(&tcp_have_smc); |
a4cf0443 | 1471 | smc_ib_unregister_client(); |
ac713874 UB |
1472 | sock_unregister(PF_SMC); |
1473 | proto_unregister(&smc_proto); | |
6812baab | 1474 | smc_pnet_exit(); |
ac713874 UB |
1475 | } |
1476 | ||
1477 | module_init(smc_init); | |
1478 | module_exit(smc_exit); | |
1479 | ||
1480 | MODULE_AUTHOR("Ursula Braun <ubraun@linux.vnet.ibm.com>"); | |
1481 | MODULE_DESCRIPTION("smc socket address family"); | |
1482 | MODULE_LICENSE("GPL"); | |
1483 | MODULE_ALIAS_NETPROTO(PF_SMC); |