xref: /openbmc/linux/net/mptcp/subflow.c (revision aa5b395b)
1 // SPDX-License-Identifier: GPL-2.0
2 /* Multipath TCP
3  *
4  * Copyright (c) 2017 - 2019, Intel Corporation.
5  */
6 
7 #define pr_fmt(fmt) "MPTCP: " fmt
8 
9 #include <linux/kernel.h>
10 #include <linux/module.h>
11 #include <linux/netdevice.h>
12 #include <net/sock.h>
13 #include <net/inet_common.h>
14 #include <net/inet_hashtables.h>
15 #include <net/protocol.h>
16 #include <net/tcp.h>
17 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
18 #include <net/ip6_route.h>
19 #endif
20 #include <net/mptcp.h>
21 #include "protocol.h"
22 
23 static int subflow_rebuild_header(struct sock *sk)
24 {
25 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
26 	int err = 0;
27 
28 	if (subflow->request_mptcp && !subflow->token) {
29 		pr_debug("subflow=%p", sk);
30 		err = mptcp_token_new_connect(sk);
31 	}
32 
33 	if (err)
34 		return err;
35 
36 	return subflow->icsk_af_ops->rebuild_header(sk);
37 }
38 
39 static void subflow_req_destructor(struct request_sock *req)
40 {
41 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
42 
43 	pr_debug("subflow_req=%p", subflow_req);
44 
45 	if (subflow_req->mp_capable)
46 		mptcp_token_destroy_request(subflow_req->token);
47 	tcp_request_sock_ops.destructor(req);
48 }
49 
50 static void subflow_init_req(struct request_sock *req,
51 			     const struct sock *sk_listener,
52 			     struct sk_buff *skb)
53 {
54 	struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk_listener);
55 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
56 	struct tcp_options_received rx_opt;
57 
58 	pr_debug("subflow_req=%p, listener=%p", subflow_req, listener);
59 
60 	memset(&rx_opt.mptcp, 0, sizeof(rx_opt.mptcp));
61 	mptcp_get_options(skb, &rx_opt);
62 
63 	subflow_req->mp_capable = 0;
64 	subflow_req->remote_key_valid = 0;
65 
66 #ifdef CONFIG_TCP_MD5SIG
67 	/* no MPTCP if MD5SIG is enabled on this socket or we may run out of
68 	 * TCP option space.
69 	 */
70 	if (rcu_access_pointer(tcp_sk(sk_listener)->md5sig_info))
71 		return;
72 #endif
73 
74 	if (rx_opt.mptcp.mp_capable && listener->request_mptcp) {
75 		int err;
76 
77 		err = mptcp_token_new_request(req);
78 		if (err == 0)
79 			subflow_req->mp_capable = 1;
80 
81 		subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq;
82 	}
83 }
84 
85 static void subflow_v4_init_req(struct request_sock *req,
86 				const struct sock *sk_listener,
87 				struct sk_buff *skb)
88 {
89 	tcp_rsk(req)->is_mptcp = 1;
90 
91 	tcp_request_sock_ipv4_ops.init_req(req, sk_listener, skb);
92 
93 	subflow_init_req(req, sk_listener, skb);
94 }
95 
96 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
97 static void subflow_v6_init_req(struct request_sock *req,
98 				const struct sock *sk_listener,
99 				struct sk_buff *skb)
100 {
101 	tcp_rsk(req)->is_mptcp = 1;
102 
103 	tcp_request_sock_ipv6_ops.init_req(req, sk_listener, skb);
104 
105 	subflow_init_req(req, sk_listener, skb);
106 }
107 #endif
108 
109 static void subflow_finish_connect(struct sock *sk, const struct sk_buff *skb)
110 {
111 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
112 
113 	subflow->icsk_af_ops->sk_rx_dst_set(sk, skb);
114 
115 	if (subflow->conn && !subflow->conn_finished) {
116 		pr_debug("subflow=%p, remote_key=%llu", mptcp_subflow_ctx(sk),
117 			 subflow->remote_key);
118 		mptcp_finish_connect(sk);
119 		subflow->conn_finished = 1;
120 
121 		if (skb) {
122 			pr_debug("synack seq=%u", TCP_SKB_CB(skb)->seq);
123 			subflow->ssn_offset = TCP_SKB_CB(skb)->seq;
124 		}
125 	}
126 }
127 
128 static struct request_sock_ops subflow_request_sock_ops;
129 static struct tcp_request_sock_ops subflow_request_sock_ipv4_ops;
130 
131 static int subflow_v4_conn_request(struct sock *sk, struct sk_buff *skb)
132 {
133 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
134 
135 	pr_debug("subflow=%p", subflow);
136 
137 	/* Never answer to SYNs sent to broadcast or multicast */
138 	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
139 		goto drop;
140 
141 	return tcp_conn_request(&subflow_request_sock_ops,
142 				&subflow_request_sock_ipv4_ops,
143 				sk, skb);
144 drop:
145 	tcp_listendrop(sk);
146 	return 0;
147 }
148 
149 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
150 static struct tcp_request_sock_ops subflow_request_sock_ipv6_ops;
151 static struct inet_connection_sock_af_ops subflow_v6_specific;
152 static struct inet_connection_sock_af_ops subflow_v6m_specific;
153 
154 static int subflow_v6_conn_request(struct sock *sk, struct sk_buff *skb)
155 {
156 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
157 
158 	pr_debug("subflow=%p", subflow);
159 
160 	if (skb->protocol == htons(ETH_P_IP))
161 		return subflow_v4_conn_request(sk, skb);
162 
163 	if (!ipv6_unicast_destination(skb))
164 		goto drop;
165 
166 	return tcp_conn_request(&subflow_request_sock_ops,
167 				&subflow_request_sock_ipv6_ops, sk, skb);
168 
169 drop:
170 	tcp_listendrop(sk);
171 	return 0; /* don't send reset */
172 }
173 #endif
174 
175 static struct sock *subflow_syn_recv_sock(const struct sock *sk,
176 					  struct sk_buff *skb,
177 					  struct request_sock *req,
178 					  struct dst_entry *dst,
179 					  struct request_sock *req_unhash,
180 					  bool *own_req)
181 {
182 	struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk);
183 	struct mptcp_subflow_request_sock *subflow_req;
184 	struct tcp_options_received opt_rx;
185 	struct sock *child;
186 
187 	pr_debug("listener=%p, req=%p, conn=%p", listener, req, listener->conn);
188 
189 	/* if the sk is MP_CAPABLE, we try to fetch the client key */
190 	subflow_req = mptcp_subflow_rsk(req);
191 	if (subflow_req->mp_capable) {
192 		if (TCP_SKB_CB(skb)->seq != subflow_req->ssn_offset + 1) {
193 			/* here we can receive and accept an in-window,
194 			 * out-of-order pkt, which will not carry the MP_CAPABLE
195 			 * opt even on mptcp enabled paths
196 			 */
197 			goto create_child;
198 		}
199 
200 		opt_rx.mptcp.mp_capable = 0;
201 		mptcp_get_options(skb, &opt_rx);
202 		if (opt_rx.mptcp.mp_capable) {
203 			subflow_req->remote_key = opt_rx.mptcp.sndr_key;
204 			subflow_req->remote_key_valid = 1;
205 		} else {
206 			subflow_req->mp_capable = 0;
207 		}
208 	}
209 
210 create_child:
211 	child = listener->icsk_af_ops->syn_recv_sock(sk, skb, req, dst,
212 						     req_unhash, own_req);
213 
214 	if (child && *own_req) {
215 		struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(child);
216 
217 		/* we have null ctx on TCP fallback, not fatal on MPC
218 		 * handshake
219 		 */
220 		if (!ctx)
221 			return child;
222 
223 		if (ctx->mp_capable) {
224 			if (mptcp_token_new_accept(ctx->token))
225 				goto close_child;
226 		}
227 	}
228 
229 	return child;
230 
231 close_child:
232 	pr_debug("closing child socket");
233 	tcp_send_active_reset(child, GFP_ATOMIC);
234 	inet_csk_prepare_forced_close(child);
235 	tcp_done(child);
236 	return NULL;
237 }
238 
239 static struct inet_connection_sock_af_ops subflow_specific;
240 
241 enum mapping_status {
242 	MAPPING_OK,
243 	MAPPING_INVALID,
244 	MAPPING_EMPTY,
245 	MAPPING_DATA_FIN
246 };
247 
248 static u64 expand_seq(u64 old_seq, u16 old_data_len, u64 seq)
249 {
250 	if ((u32)seq == (u32)old_seq)
251 		return old_seq;
252 
253 	/* Assume map covers data not mapped yet. */
254 	return seq | ((old_seq + old_data_len + 1) & GENMASK_ULL(63, 32));
255 }
256 
257 static void warn_bad_map(struct mptcp_subflow_context *subflow, u32 ssn)
258 {
259 	WARN_ONCE(1, "Bad mapping: ssn=%d map_seq=%d map_data_len=%d",
260 		  ssn, subflow->map_subflow_seq, subflow->map_data_len);
261 }
262 
263 static bool skb_is_fully_mapped(struct sock *ssk, struct sk_buff *skb)
264 {
265 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
266 	unsigned int skb_consumed;
267 
268 	skb_consumed = tcp_sk(ssk)->copied_seq - TCP_SKB_CB(skb)->seq;
269 	if (WARN_ON_ONCE(skb_consumed >= skb->len))
270 		return true;
271 
272 	return skb->len - skb_consumed <= subflow->map_data_len -
273 					  mptcp_subflow_get_map_offset(subflow);
274 }
275 
276 static bool validate_mapping(struct sock *ssk, struct sk_buff *skb)
277 {
278 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
279 	u32 ssn = tcp_sk(ssk)->copied_seq - subflow->ssn_offset;
280 
281 	if (unlikely(before(ssn, subflow->map_subflow_seq))) {
282 		/* Mapping covers data later in the subflow stream,
283 		 * currently unsupported.
284 		 */
285 		warn_bad_map(subflow, ssn);
286 		return false;
287 	}
288 	if (unlikely(!before(ssn, subflow->map_subflow_seq +
289 				  subflow->map_data_len))) {
290 		/* Mapping does covers past subflow data, invalid */
291 		warn_bad_map(subflow, ssn + skb->len);
292 		return false;
293 	}
294 	return true;
295 }
296 
297 static enum mapping_status get_mapping_status(struct sock *ssk)
298 {
299 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
300 	struct mptcp_ext *mpext;
301 	struct sk_buff *skb;
302 	u16 data_len;
303 	u64 map_seq;
304 
305 	skb = skb_peek(&ssk->sk_receive_queue);
306 	if (!skb)
307 		return MAPPING_EMPTY;
308 
309 	mpext = mptcp_get_ext(skb);
310 	if (!mpext || !mpext->use_map) {
311 		if (!subflow->map_valid && !skb->len) {
312 			/* the TCP stack deliver 0 len FIN pkt to the receive
313 			 * queue, that is the only 0len pkts ever expected here,
314 			 * and we can admit no mapping only for 0 len pkts
315 			 */
316 			if (!(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN))
317 				WARN_ONCE(1, "0len seq %d:%d flags %x",
318 					  TCP_SKB_CB(skb)->seq,
319 					  TCP_SKB_CB(skb)->end_seq,
320 					  TCP_SKB_CB(skb)->tcp_flags);
321 			sk_eat_skb(ssk, skb);
322 			return MAPPING_EMPTY;
323 		}
324 
325 		if (!subflow->map_valid)
326 			return MAPPING_INVALID;
327 
328 		goto validate_seq;
329 	}
330 
331 	pr_debug("seq=%llu is64=%d ssn=%u data_len=%u data_fin=%d",
332 		 mpext->data_seq, mpext->dsn64, mpext->subflow_seq,
333 		 mpext->data_len, mpext->data_fin);
334 
335 	data_len = mpext->data_len;
336 	if (data_len == 0) {
337 		pr_err("Infinite mapping not handled");
338 		return MAPPING_INVALID;
339 	}
340 
341 	if (mpext->data_fin == 1) {
342 		if (data_len == 1) {
343 			pr_debug("DATA_FIN with no payload");
344 			if (subflow->map_valid) {
345 				/* A DATA_FIN might arrive in a DSS
346 				 * option before the previous mapping
347 				 * has been fully consumed. Continue
348 				 * handling the existing mapping.
349 				 */
350 				skb_ext_del(skb, SKB_EXT_MPTCP);
351 				return MAPPING_OK;
352 			} else {
353 				return MAPPING_DATA_FIN;
354 			}
355 		}
356 
357 		/* Adjust for DATA_FIN using 1 byte of sequence space */
358 		data_len--;
359 	}
360 
361 	if (!mpext->dsn64) {
362 		map_seq = expand_seq(subflow->map_seq, subflow->map_data_len,
363 				     mpext->data_seq);
364 		pr_debug("expanded seq=%llu", subflow->map_seq);
365 	} else {
366 		map_seq = mpext->data_seq;
367 	}
368 
369 	if (subflow->map_valid) {
370 		/* Allow replacing only with an identical map */
371 		if (subflow->map_seq == map_seq &&
372 		    subflow->map_subflow_seq == mpext->subflow_seq &&
373 		    subflow->map_data_len == data_len) {
374 			skb_ext_del(skb, SKB_EXT_MPTCP);
375 			return MAPPING_OK;
376 		}
377 
378 		/* If this skb data are fully covered by the current mapping,
379 		 * the new map would need caching, which is not supported
380 		 */
381 		if (skb_is_fully_mapped(ssk, skb))
382 			return MAPPING_INVALID;
383 
384 		/* will validate the next map after consuming the current one */
385 		return MAPPING_OK;
386 	}
387 
388 	subflow->map_seq = map_seq;
389 	subflow->map_subflow_seq = mpext->subflow_seq;
390 	subflow->map_data_len = data_len;
391 	subflow->map_valid = 1;
392 	subflow->mpc_map = mpext->mpc_map;
393 	pr_debug("new map seq=%llu subflow_seq=%u data_len=%u",
394 		 subflow->map_seq, subflow->map_subflow_seq,
395 		 subflow->map_data_len);
396 
397 validate_seq:
398 	/* we revalidate valid mapping on new skb, because we must ensure
399 	 * the current skb is completely covered by the available mapping
400 	 */
401 	if (!validate_mapping(ssk, skb))
402 		return MAPPING_INVALID;
403 
404 	skb_ext_del(skb, SKB_EXT_MPTCP);
405 	return MAPPING_OK;
406 }
407 
408 static bool subflow_check_data_avail(struct sock *ssk)
409 {
410 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
411 	enum mapping_status status;
412 	struct mptcp_sock *msk;
413 	struct sk_buff *skb;
414 
415 	pr_debug("msk=%p ssk=%p data_avail=%d skb=%p", subflow->conn, ssk,
416 		 subflow->data_avail, skb_peek(&ssk->sk_receive_queue));
417 	if (subflow->data_avail)
418 		return true;
419 
420 	if (!subflow->conn)
421 		return false;
422 
423 	msk = mptcp_sk(subflow->conn);
424 	for (;;) {
425 		u32 map_remaining;
426 		size_t delta;
427 		u64 ack_seq;
428 		u64 old_ack;
429 
430 		status = get_mapping_status(ssk);
431 		pr_debug("msk=%p ssk=%p status=%d", msk, ssk, status);
432 		if (status == MAPPING_INVALID) {
433 			ssk->sk_err = EBADMSG;
434 			goto fatal;
435 		}
436 
437 		if (status != MAPPING_OK)
438 			return false;
439 
440 		skb = skb_peek(&ssk->sk_receive_queue);
441 		if (WARN_ON_ONCE(!skb))
442 			return false;
443 
444 		/* if msk lacks the remote key, this subflow must provide an
445 		 * MP_CAPABLE-based mapping
446 		 */
447 		if (unlikely(!READ_ONCE(msk->can_ack))) {
448 			if (!subflow->mpc_map) {
449 				ssk->sk_err = EBADMSG;
450 				goto fatal;
451 			}
452 			WRITE_ONCE(msk->remote_key, subflow->remote_key);
453 			WRITE_ONCE(msk->ack_seq, subflow->map_seq);
454 			WRITE_ONCE(msk->can_ack, true);
455 		}
456 
457 		old_ack = READ_ONCE(msk->ack_seq);
458 		ack_seq = mptcp_subflow_get_mapped_dsn(subflow);
459 		pr_debug("msk ack_seq=%llx subflow ack_seq=%llx", old_ack,
460 			 ack_seq);
461 		if (ack_seq == old_ack)
462 			break;
463 
464 		/* only accept in-sequence mapping. Old values are spurious
465 		 * retransmission; we can hit "future" values on active backup
466 		 * subflow switch, we relay on retransmissions to get
467 		 * in-sequence data.
468 		 * Cuncurrent subflows support will require subflow data
469 		 * reordering
470 		 */
471 		map_remaining = subflow->map_data_len -
472 				mptcp_subflow_get_map_offset(subflow);
473 		if (before64(ack_seq, old_ack))
474 			delta = min_t(size_t, old_ack - ack_seq, map_remaining);
475 		else
476 			delta = min_t(size_t, ack_seq - old_ack, map_remaining);
477 
478 		/* discard mapped data */
479 		pr_debug("discarding %zu bytes, current map len=%d", delta,
480 			 map_remaining);
481 		if (delta) {
482 			struct mptcp_read_arg arg = {
483 				.msg = NULL,
484 			};
485 			read_descriptor_t desc = {
486 				.count = delta,
487 				.arg.data = &arg,
488 			};
489 			int ret;
490 
491 			ret = tcp_read_sock(ssk, &desc, mptcp_read_actor);
492 			if (ret < 0) {
493 				ssk->sk_err = -ret;
494 				goto fatal;
495 			}
496 			if (ret < delta)
497 				return false;
498 			if (delta == map_remaining)
499 				subflow->map_valid = 0;
500 		}
501 	}
502 	return true;
503 
504 fatal:
505 	/* fatal protocol error, close the socket */
506 	/* This barrier is coupled with smp_rmb() in tcp_poll() */
507 	smp_wmb();
508 	ssk->sk_error_report(ssk);
509 	tcp_set_state(ssk, TCP_CLOSE);
510 	tcp_send_active_reset(ssk, GFP_ATOMIC);
511 	return false;
512 }
513 
514 bool mptcp_subflow_data_available(struct sock *sk)
515 {
516 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
517 	struct sk_buff *skb;
518 
519 	/* check if current mapping is still valid */
520 	if (subflow->map_valid &&
521 	    mptcp_subflow_get_map_offset(subflow) >= subflow->map_data_len) {
522 		subflow->map_valid = 0;
523 		subflow->data_avail = 0;
524 
525 		pr_debug("Done with mapping: seq=%u data_len=%u",
526 			 subflow->map_subflow_seq,
527 			 subflow->map_data_len);
528 	}
529 
530 	if (!subflow_check_data_avail(sk)) {
531 		subflow->data_avail = 0;
532 		return false;
533 	}
534 
535 	skb = skb_peek(&sk->sk_receive_queue);
536 	subflow->data_avail = skb &&
537 		       before(tcp_sk(sk)->copied_seq, TCP_SKB_CB(skb)->end_seq);
538 	return subflow->data_avail;
539 }
540 
541 static void subflow_data_ready(struct sock *sk)
542 {
543 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
544 	struct sock *parent = subflow->conn;
545 
546 	if (!parent || !subflow->mp_capable) {
547 		subflow->tcp_data_ready(sk);
548 
549 		if (parent)
550 			parent->sk_data_ready(parent);
551 		return;
552 	}
553 
554 	if (mptcp_subflow_data_available(sk)) {
555 		set_bit(MPTCP_DATA_READY, &mptcp_sk(parent)->flags);
556 
557 		parent->sk_data_ready(parent);
558 	}
559 }
560 
561 static void subflow_write_space(struct sock *sk)
562 {
563 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
564 	struct sock *parent = subflow->conn;
565 
566 	sk_stream_write_space(sk);
567 	if (parent && sk_stream_is_writeable(sk)) {
568 		set_bit(MPTCP_SEND_SPACE, &mptcp_sk(parent)->flags);
569 		smp_mb__after_atomic();
570 		/* set SEND_SPACE before sk_stream_write_space clears NOSPACE */
571 		sk_stream_write_space(parent);
572 	}
573 }
574 
575 static struct inet_connection_sock_af_ops *
576 subflow_default_af_ops(struct sock *sk)
577 {
578 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
579 	if (sk->sk_family == AF_INET6)
580 		return &subflow_v6_specific;
581 #endif
582 	return &subflow_specific;
583 }
584 
585 void mptcp_handle_ipv6_mapped(struct sock *sk, bool mapped)
586 {
587 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
588 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
589 	struct inet_connection_sock *icsk = inet_csk(sk);
590 	struct inet_connection_sock_af_ops *target;
591 
592 	target = mapped ? &subflow_v6m_specific : subflow_default_af_ops(sk);
593 
594 	pr_debug("subflow=%p family=%d ops=%p target=%p mapped=%d",
595 		 subflow, sk->sk_family, icsk->icsk_af_ops, target, mapped);
596 
597 	if (likely(icsk->icsk_af_ops == target))
598 		return;
599 
600 	subflow->icsk_af_ops = icsk->icsk_af_ops;
601 	icsk->icsk_af_ops = target;
602 #endif
603 }
604 
605 int mptcp_subflow_create_socket(struct sock *sk, struct socket **new_sock)
606 {
607 	struct mptcp_subflow_context *subflow;
608 	struct net *net = sock_net(sk);
609 	struct socket *sf;
610 	int err;
611 
612 	err = sock_create_kern(net, sk->sk_family, SOCK_STREAM, IPPROTO_TCP,
613 			       &sf);
614 	if (err)
615 		return err;
616 
617 	lock_sock(sf->sk);
618 
619 	/* kernel sockets do not by default acquire net ref, but TCP timer
620 	 * needs it.
621 	 */
622 	sf->sk->sk_net_refcnt = 1;
623 	get_net(net);
624 	this_cpu_add(*net->core.sock_inuse, 1);
625 	err = tcp_set_ulp(sf->sk, "mptcp");
626 	release_sock(sf->sk);
627 
628 	if (err)
629 		return err;
630 
631 	subflow = mptcp_subflow_ctx(sf->sk);
632 	pr_debug("subflow=%p", subflow);
633 
634 	*new_sock = sf;
635 	sock_hold(sk);
636 	subflow->conn = sk;
637 
638 	return 0;
639 }
640 
641 static struct mptcp_subflow_context *subflow_create_ctx(struct sock *sk,
642 							gfp_t priority)
643 {
644 	struct inet_connection_sock *icsk = inet_csk(sk);
645 	struct mptcp_subflow_context *ctx;
646 
647 	ctx = kzalloc(sizeof(*ctx), priority);
648 	if (!ctx)
649 		return NULL;
650 
651 	rcu_assign_pointer(icsk->icsk_ulp_data, ctx);
652 	INIT_LIST_HEAD(&ctx->node);
653 
654 	pr_debug("subflow=%p", ctx);
655 
656 	ctx->tcp_sock = sk;
657 
658 	return ctx;
659 }
660 
661 static void __subflow_state_change(struct sock *sk)
662 {
663 	struct socket_wq *wq;
664 
665 	rcu_read_lock();
666 	wq = rcu_dereference(sk->sk_wq);
667 	if (skwq_has_sleeper(wq))
668 		wake_up_interruptible_all(&wq->wait);
669 	rcu_read_unlock();
670 }
671 
672 static bool subflow_is_done(const struct sock *sk)
673 {
674 	return sk->sk_shutdown & RCV_SHUTDOWN || sk->sk_state == TCP_CLOSE;
675 }
676 
677 static void subflow_state_change(struct sock *sk)
678 {
679 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
680 	struct sock *parent = READ_ONCE(subflow->conn);
681 
682 	__subflow_state_change(sk);
683 
684 	/* as recvmsg() does not acquire the subflow socket for ssk selection
685 	 * a fin packet carrying a DSS can be unnoticed if we don't trigger
686 	 * the data available machinery here.
687 	 */
688 	if (parent && subflow->mp_capable && mptcp_subflow_data_available(sk)) {
689 		set_bit(MPTCP_DATA_READY, &mptcp_sk(parent)->flags);
690 
691 		parent->sk_data_ready(parent);
692 	}
693 
694 	if (parent && !(parent->sk_shutdown & RCV_SHUTDOWN) &&
695 	    !subflow->rx_eof && subflow_is_done(sk)) {
696 		subflow->rx_eof = 1;
697 		parent->sk_shutdown |= RCV_SHUTDOWN;
698 		__subflow_state_change(parent);
699 	}
700 }
701 
702 static int subflow_ulp_init(struct sock *sk)
703 {
704 	struct inet_connection_sock *icsk = inet_csk(sk);
705 	struct mptcp_subflow_context *ctx;
706 	struct tcp_sock *tp = tcp_sk(sk);
707 	int err = 0;
708 
709 	/* disallow attaching ULP to a socket unless it has been
710 	 * created with sock_create_kern()
711 	 */
712 	if (!sk->sk_kern_sock) {
713 		err = -EOPNOTSUPP;
714 		goto out;
715 	}
716 
717 	ctx = subflow_create_ctx(sk, GFP_KERNEL);
718 	if (!ctx) {
719 		err = -ENOMEM;
720 		goto out;
721 	}
722 
723 	pr_debug("subflow=%p, family=%d", ctx, sk->sk_family);
724 
725 	tp->is_mptcp = 1;
726 	ctx->icsk_af_ops = icsk->icsk_af_ops;
727 	icsk->icsk_af_ops = subflow_default_af_ops(sk);
728 	ctx->tcp_data_ready = sk->sk_data_ready;
729 	ctx->tcp_state_change = sk->sk_state_change;
730 	ctx->tcp_write_space = sk->sk_write_space;
731 	sk->sk_data_ready = subflow_data_ready;
732 	sk->sk_write_space = subflow_write_space;
733 	sk->sk_state_change = subflow_state_change;
734 out:
735 	return err;
736 }
737 
738 static void subflow_ulp_release(struct sock *sk)
739 {
740 	struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(sk);
741 
742 	if (!ctx)
743 		return;
744 
745 	if (ctx->conn)
746 		sock_put(ctx->conn);
747 
748 	kfree_rcu(ctx, rcu);
749 }
750 
751 static void subflow_ulp_fallback(struct sock *sk,
752 				 struct mptcp_subflow_context *old_ctx)
753 {
754 	struct inet_connection_sock *icsk = inet_csk(sk);
755 
756 	mptcp_subflow_tcp_fallback(sk, old_ctx);
757 	icsk->icsk_ulp_ops = NULL;
758 	rcu_assign_pointer(icsk->icsk_ulp_data, NULL);
759 	tcp_sk(sk)->is_mptcp = 0;
760 }
761 
762 static void subflow_ulp_clone(const struct request_sock *req,
763 			      struct sock *newsk,
764 			      const gfp_t priority)
765 {
766 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
767 	struct mptcp_subflow_context *old_ctx = mptcp_subflow_ctx(newsk);
768 	struct mptcp_subflow_context *new_ctx;
769 
770 	if (!subflow_req->mp_capable) {
771 		subflow_ulp_fallback(newsk, old_ctx);
772 		return;
773 	}
774 
775 	new_ctx = subflow_create_ctx(newsk, priority);
776 	if (!new_ctx) {
777 		subflow_ulp_fallback(newsk, old_ctx);
778 		return;
779 	}
780 
781 	/* see comments in subflow_syn_recv_sock(), MPTCP connection is fully
782 	 * established only after we receive the remote key
783 	 */
784 	new_ctx->conn_finished = 1;
785 	new_ctx->icsk_af_ops = old_ctx->icsk_af_ops;
786 	new_ctx->tcp_data_ready = old_ctx->tcp_data_ready;
787 	new_ctx->tcp_state_change = old_ctx->tcp_state_change;
788 	new_ctx->tcp_write_space = old_ctx->tcp_write_space;
789 	new_ctx->mp_capable = 1;
790 	new_ctx->fourth_ack = subflow_req->remote_key_valid;
791 	new_ctx->can_ack = subflow_req->remote_key_valid;
792 	new_ctx->remote_key = subflow_req->remote_key;
793 	new_ctx->local_key = subflow_req->local_key;
794 	new_ctx->token = subflow_req->token;
795 	new_ctx->ssn_offset = subflow_req->ssn_offset;
796 	new_ctx->idsn = subflow_req->idsn;
797 }
798 
799 static struct tcp_ulp_ops subflow_ulp_ops __read_mostly = {
800 	.name		= "mptcp",
801 	.owner		= THIS_MODULE,
802 	.init		= subflow_ulp_init,
803 	.release	= subflow_ulp_release,
804 	.clone		= subflow_ulp_clone,
805 };
806 
807 static int subflow_ops_init(struct request_sock_ops *subflow_ops)
808 {
809 	subflow_ops->obj_size = sizeof(struct mptcp_subflow_request_sock);
810 	subflow_ops->slab_name = "request_sock_subflow";
811 
812 	subflow_ops->slab = kmem_cache_create(subflow_ops->slab_name,
813 					      subflow_ops->obj_size, 0,
814 					      SLAB_ACCOUNT |
815 					      SLAB_TYPESAFE_BY_RCU,
816 					      NULL);
817 	if (!subflow_ops->slab)
818 		return -ENOMEM;
819 
820 	subflow_ops->destructor = subflow_req_destructor;
821 
822 	return 0;
823 }
824 
825 void mptcp_subflow_init(void)
826 {
827 	subflow_request_sock_ops = tcp_request_sock_ops;
828 	if (subflow_ops_init(&subflow_request_sock_ops) != 0)
829 		panic("MPTCP: failed to init subflow request sock ops\n");
830 
831 	subflow_request_sock_ipv4_ops = tcp_request_sock_ipv4_ops;
832 	subflow_request_sock_ipv4_ops.init_req = subflow_v4_init_req;
833 
834 	subflow_specific = ipv4_specific;
835 	subflow_specific.conn_request = subflow_v4_conn_request;
836 	subflow_specific.syn_recv_sock = subflow_syn_recv_sock;
837 	subflow_specific.sk_rx_dst_set = subflow_finish_connect;
838 	subflow_specific.rebuild_header = subflow_rebuild_header;
839 
840 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
841 	subflow_request_sock_ipv6_ops = tcp_request_sock_ipv6_ops;
842 	subflow_request_sock_ipv6_ops.init_req = subflow_v6_init_req;
843 
844 	subflow_v6_specific = ipv6_specific;
845 	subflow_v6_specific.conn_request = subflow_v6_conn_request;
846 	subflow_v6_specific.syn_recv_sock = subflow_syn_recv_sock;
847 	subflow_v6_specific.sk_rx_dst_set = subflow_finish_connect;
848 	subflow_v6_specific.rebuild_header = subflow_rebuild_header;
849 
850 	subflow_v6m_specific = subflow_v6_specific;
851 	subflow_v6m_specific.queue_xmit = ipv4_specific.queue_xmit;
852 	subflow_v6m_specific.send_check = ipv4_specific.send_check;
853 	subflow_v6m_specific.net_header_len = ipv4_specific.net_header_len;
854 	subflow_v6m_specific.mtu_reduced = ipv4_specific.mtu_reduced;
855 	subflow_v6m_specific.net_frag_header_len = 0;
856 #endif
857 
858 	if (tcp_register_ulp(&subflow_ulp_ops) != 0)
859 		panic("MPTCP: failed to register subflows to ULP\n");
860 }
861