xref: /openbmc/linux/net/mptcp/subflow.c (revision 43eca11b)
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 <crypto/algapi.h>
13 #include <crypto/sha2.h>
14 #include <net/sock.h>
15 #include <net/inet_common.h>
16 #include <net/inet_hashtables.h>
17 #include <net/protocol.h>
18 #include <net/tcp.h>
19 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
20 #include <net/ip6_route.h>
21 #include <net/transp_v6.h>
22 #endif
23 #include <net/mptcp.h>
24 #include <uapi/linux/mptcp.h>
25 #include "protocol.h"
26 #include "mib.h"
27 
28 #include <trace/events/mptcp.h>
29 #include <trace/events/sock.h>
30 
31 static void mptcp_subflow_ops_undo_override(struct sock *ssk);
32 
SUBFLOW_REQ_INC_STATS(struct request_sock * req,enum linux_mptcp_mib_field field)33 static void SUBFLOW_REQ_INC_STATS(struct request_sock *req,
34 				  enum linux_mptcp_mib_field field)
35 {
36 	MPTCP_INC_STATS(sock_net(req_to_sk(req)), field);
37 }
38 
subflow_req_destructor(struct request_sock * req)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->msk)
46 		sock_put((struct sock *)subflow_req->msk);
47 
48 	mptcp_token_destroy_request(req);
49 }
50 
subflow_generate_hmac(u64 key1,u64 key2,u32 nonce1,u32 nonce2,void * hmac)51 static void subflow_generate_hmac(u64 key1, u64 key2, u32 nonce1, u32 nonce2,
52 				  void *hmac)
53 {
54 	u8 msg[8];
55 
56 	put_unaligned_be32(nonce1, &msg[0]);
57 	put_unaligned_be32(nonce2, &msg[4]);
58 
59 	mptcp_crypto_hmac_sha(key1, key2, msg, 8, hmac);
60 }
61 
mptcp_can_accept_new_subflow(const struct mptcp_sock * msk)62 static bool mptcp_can_accept_new_subflow(const struct mptcp_sock *msk)
63 {
64 	return mptcp_is_fully_established((void *)msk) &&
65 		((mptcp_pm_is_userspace(msk) &&
66 		  mptcp_userspace_pm_active(msk)) ||
67 		 READ_ONCE(msk->pm.accept_subflow));
68 }
69 
70 /* validate received token and create truncated hmac and nonce for SYN-ACK */
subflow_req_create_thmac(struct mptcp_subflow_request_sock * subflow_req)71 static void subflow_req_create_thmac(struct mptcp_subflow_request_sock *subflow_req)
72 {
73 	struct mptcp_sock *msk = subflow_req->msk;
74 	u8 hmac[SHA256_DIGEST_SIZE];
75 
76 	get_random_bytes(&subflow_req->local_nonce, sizeof(u32));
77 
78 	subflow_generate_hmac(msk->local_key, msk->remote_key,
79 			      subflow_req->local_nonce,
80 			      subflow_req->remote_nonce, hmac);
81 
82 	subflow_req->thmac = get_unaligned_be64(hmac);
83 }
84 
subflow_token_join_request(struct request_sock * req)85 static struct mptcp_sock *subflow_token_join_request(struct request_sock *req)
86 {
87 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
88 	struct mptcp_sock *msk;
89 	int local_id;
90 
91 	msk = mptcp_token_get_sock(sock_net(req_to_sk(req)), subflow_req->token);
92 	if (!msk) {
93 		SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINNOTOKEN);
94 		return NULL;
95 	}
96 
97 	local_id = mptcp_pm_get_local_id(msk, (struct sock_common *)req);
98 	if (local_id < 0) {
99 		sock_put((struct sock *)msk);
100 		return NULL;
101 	}
102 	subflow_req->local_id = local_id;
103 
104 	return msk;
105 }
106 
subflow_init_req(struct request_sock * req,const struct sock * sk_listener)107 static void subflow_init_req(struct request_sock *req, const struct sock *sk_listener)
108 {
109 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
110 
111 	subflow_req->mp_capable = 0;
112 	subflow_req->mp_join = 0;
113 	subflow_req->csum_reqd = mptcp_is_checksum_enabled(sock_net(sk_listener));
114 	subflow_req->allow_join_id0 = mptcp_allow_join_id0(sock_net(sk_listener));
115 	subflow_req->msk = NULL;
116 	mptcp_token_init_request(req);
117 }
118 
subflow_use_different_sport(struct mptcp_sock * msk,const struct sock * sk)119 static bool subflow_use_different_sport(struct mptcp_sock *msk, const struct sock *sk)
120 {
121 	return inet_sk(sk)->inet_sport != inet_sk((struct sock *)msk)->inet_sport;
122 }
123 
subflow_add_reset_reason(struct sk_buff * skb,u8 reason)124 static void subflow_add_reset_reason(struct sk_buff *skb, u8 reason)
125 {
126 	struct mptcp_ext *mpext = skb_ext_add(skb, SKB_EXT_MPTCP);
127 
128 	if (mpext) {
129 		memset(mpext, 0, sizeof(*mpext));
130 		mpext->reset_reason = reason;
131 	}
132 }
133 
134 /* Init mptcp request socket.
135  *
136  * Returns an error code if a JOIN has failed and a TCP reset
137  * should be sent.
138  */
subflow_check_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)139 static int subflow_check_req(struct request_sock *req,
140 			     const struct sock *sk_listener,
141 			     struct sk_buff *skb)
142 {
143 	struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk_listener);
144 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
145 	struct mptcp_options_received mp_opt;
146 	bool opt_mp_capable, opt_mp_join;
147 
148 	pr_debug("subflow_req=%p, listener=%p", subflow_req, listener);
149 
150 #ifdef CONFIG_TCP_MD5SIG
151 	/* no MPTCP if MD5SIG is enabled on this socket or we may run out of
152 	 * TCP option space.
153 	 */
154 	if (rcu_access_pointer(tcp_sk(sk_listener)->md5sig_info))
155 		return -EINVAL;
156 #endif
157 
158 	mptcp_get_options(skb, &mp_opt);
159 
160 	opt_mp_capable = !!(mp_opt.suboptions & OPTION_MPTCP_MPC_SYN);
161 	opt_mp_join = !!(mp_opt.suboptions & OPTION_MPTCP_MPJ_SYN);
162 	if (opt_mp_capable) {
163 		SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_MPCAPABLEPASSIVE);
164 
165 		if (opt_mp_join)
166 			return 0;
167 	} else if (opt_mp_join) {
168 		SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINSYNRX);
169 	}
170 
171 	if (opt_mp_capable && listener->request_mptcp) {
172 		int err, retries = MPTCP_TOKEN_MAX_RETRIES;
173 
174 		subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq;
175 again:
176 		do {
177 			get_random_bytes(&subflow_req->local_key, sizeof(subflow_req->local_key));
178 		} while (subflow_req->local_key == 0);
179 
180 		if (unlikely(req->syncookie)) {
181 			mptcp_crypto_key_sha(subflow_req->local_key,
182 					     &subflow_req->token,
183 					     &subflow_req->idsn);
184 			if (mptcp_token_exists(subflow_req->token)) {
185 				if (retries-- > 0)
186 					goto again;
187 				SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_TOKENFALLBACKINIT);
188 			} else {
189 				subflow_req->mp_capable = 1;
190 			}
191 			return 0;
192 		}
193 
194 		err = mptcp_token_new_request(req);
195 		if (err == 0)
196 			subflow_req->mp_capable = 1;
197 		else if (retries-- > 0)
198 			goto again;
199 		else
200 			SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_TOKENFALLBACKINIT);
201 
202 	} else if (opt_mp_join && listener->request_mptcp) {
203 		subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq;
204 		subflow_req->mp_join = 1;
205 		subflow_req->backup = mp_opt.backup;
206 		subflow_req->remote_id = mp_opt.join_id;
207 		subflow_req->token = mp_opt.token;
208 		subflow_req->remote_nonce = mp_opt.nonce;
209 		subflow_req->msk = subflow_token_join_request(req);
210 
211 		/* Can't fall back to TCP in this case. */
212 		if (!subflow_req->msk) {
213 			subflow_add_reset_reason(skb, MPTCP_RST_EMPTCP);
214 			return -EPERM;
215 		}
216 
217 		if (subflow_use_different_sport(subflow_req->msk, sk_listener)) {
218 			pr_debug("syn inet_sport=%d %d",
219 				 ntohs(inet_sk(sk_listener)->inet_sport),
220 				 ntohs(inet_sk((struct sock *)subflow_req->msk)->inet_sport));
221 			if (!mptcp_pm_sport_in_anno_list(subflow_req->msk, sk_listener)) {
222 				SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_MISMATCHPORTSYNRX);
223 				return -EPERM;
224 			}
225 			SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINPORTSYNRX);
226 		}
227 
228 		subflow_req_create_thmac(subflow_req);
229 
230 		if (unlikely(req->syncookie)) {
231 			if (mptcp_can_accept_new_subflow(subflow_req->msk))
232 				subflow_init_req_cookie_join_save(subflow_req, skb);
233 			else
234 				return -EPERM;
235 		}
236 
237 		pr_debug("token=%u, remote_nonce=%u msk=%p", subflow_req->token,
238 			 subflow_req->remote_nonce, subflow_req->msk);
239 	}
240 
241 	return 0;
242 }
243 
mptcp_subflow_init_cookie_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)244 int mptcp_subflow_init_cookie_req(struct request_sock *req,
245 				  const struct sock *sk_listener,
246 				  struct sk_buff *skb)
247 {
248 	struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk_listener);
249 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
250 	struct mptcp_options_received mp_opt;
251 	bool opt_mp_capable, opt_mp_join;
252 	int err;
253 
254 	subflow_init_req(req, sk_listener);
255 	mptcp_get_options(skb, &mp_opt);
256 
257 	opt_mp_capable = !!(mp_opt.suboptions & OPTION_MPTCP_MPC_ACK);
258 	opt_mp_join = !!(mp_opt.suboptions & OPTION_MPTCP_MPJ_ACK);
259 	if (opt_mp_capable && opt_mp_join)
260 		return -EINVAL;
261 
262 	if (opt_mp_capable && listener->request_mptcp) {
263 		if (mp_opt.sndr_key == 0)
264 			return -EINVAL;
265 
266 		subflow_req->local_key = mp_opt.rcvr_key;
267 		err = mptcp_token_new_request(req);
268 		if (err)
269 			return err;
270 
271 		subflow_req->mp_capable = 1;
272 		subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq - 1;
273 	} else if (opt_mp_join && listener->request_mptcp) {
274 		if (!mptcp_token_join_cookie_init_state(subflow_req, skb))
275 			return -EINVAL;
276 
277 		subflow_req->mp_join = 1;
278 		subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq - 1;
279 	}
280 
281 	return 0;
282 }
283 EXPORT_SYMBOL_GPL(mptcp_subflow_init_cookie_req);
284 
subflow_v4_route_req(const struct sock * sk,struct sk_buff * skb,struct flowi * fl,struct request_sock * req)285 static struct dst_entry *subflow_v4_route_req(const struct sock *sk,
286 					      struct sk_buff *skb,
287 					      struct flowi *fl,
288 					      struct request_sock *req)
289 {
290 	struct dst_entry *dst;
291 	int err;
292 
293 	tcp_rsk(req)->is_mptcp = 1;
294 	subflow_init_req(req, sk);
295 
296 	dst = tcp_request_sock_ipv4_ops.route_req(sk, skb, fl, req);
297 	if (!dst)
298 		return NULL;
299 
300 	err = subflow_check_req(req, sk, skb);
301 	if (err == 0)
302 		return dst;
303 
304 	dst_release(dst);
305 	if (!req->syncookie)
306 		tcp_request_sock_ops.send_reset(sk, skb);
307 	return NULL;
308 }
309 
subflow_prep_synack(const struct sock * sk,struct request_sock * req,struct tcp_fastopen_cookie * foc,enum tcp_synack_type synack_type)310 static void subflow_prep_synack(const struct sock *sk, struct request_sock *req,
311 				struct tcp_fastopen_cookie *foc,
312 				enum tcp_synack_type synack_type)
313 {
314 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
315 	struct inet_request_sock *ireq = inet_rsk(req);
316 
317 	/* clear tstamp_ok, as needed depending on cookie */
318 	if (foc && foc->len > -1)
319 		ireq->tstamp_ok = 0;
320 
321 	if (synack_type == TCP_SYNACK_FASTOPEN)
322 		mptcp_fastopen_subflow_synack_set_params(subflow, req);
323 }
324 
subflow_v4_send_synack(const struct sock * sk,struct dst_entry * dst,struct flowi * fl,struct request_sock * req,struct tcp_fastopen_cookie * foc,enum tcp_synack_type synack_type,struct sk_buff * syn_skb)325 static int subflow_v4_send_synack(const struct sock *sk, struct dst_entry *dst,
326 				  struct flowi *fl,
327 				  struct request_sock *req,
328 				  struct tcp_fastopen_cookie *foc,
329 				  enum tcp_synack_type synack_type,
330 				  struct sk_buff *syn_skb)
331 {
332 	subflow_prep_synack(sk, req, foc, synack_type);
333 
334 	return tcp_request_sock_ipv4_ops.send_synack(sk, dst, fl, req, foc,
335 						     synack_type, syn_skb);
336 }
337 
338 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
subflow_v6_send_synack(const struct sock * sk,struct dst_entry * dst,struct flowi * fl,struct request_sock * req,struct tcp_fastopen_cookie * foc,enum tcp_synack_type synack_type,struct sk_buff * syn_skb)339 static int subflow_v6_send_synack(const struct sock *sk, struct dst_entry *dst,
340 				  struct flowi *fl,
341 				  struct request_sock *req,
342 				  struct tcp_fastopen_cookie *foc,
343 				  enum tcp_synack_type synack_type,
344 				  struct sk_buff *syn_skb)
345 {
346 	subflow_prep_synack(sk, req, foc, synack_type);
347 
348 	return tcp_request_sock_ipv6_ops.send_synack(sk, dst, fl, req, foc,
349 						     synack_type, syn_skb);
350 }
351 
subflow_v6_route_req(const struct sock * sk,struct sk_buff * skb,struct flowi * fl,struct request_sock * req)352 static struct dst_entry *subflow_v6_route_req(const struct sock *sk,
353 					      struct sk_buff *skb,
354 					      struct flowi *fl,
355 					      struct request_sock *req)
356 {
357 	struct dst_entry *dst;
358 	int err;
359 
360 	tcp_rsk(req)->is_mptcp = 1;
361 	subflow_init_req(req, sk);
362 
363 	dst = tcp_request_sock_ipv6_ops.route_req(sk, skb, fl, req);
364 	if (!dst)
365 		return NULL;
366 
367 	err = subflow_check_req(req, sk, skb);
368 	if (err == 0)
369 		return dst;
370 
371 	dst_release(dst);
372 	if (!req->syncookie)
373 		tcp6_request_sock_ops.send_reset(sk, skb);
374 	return NULL;
375 }
376 #endif
377 
378 /* validate received truncated hmac and create hmac for third ACK */
subflow_thmac_valid(struct mptcp_subflow_context * subflow)379 static bool subflow_thmac_valid(struct mptcp_subflow_context *subflow)
380 {
381 	u8 hmac[SHA256_DIGEST_SIZE];
382 	u64 thmac;
383 
384 	subflow_generate_hmac(subflow->remote_key, subflow->local_key,
385 			      subflow->remote_nonce, subflow->local_nonce,
386 			      hmac);
387 
388 	thmac = get_unaligned_be64(hmac);
389 	pr_debug("subflow=%p, token=%u, thmac=%llu, subflow->thmac=%llu\n",
390 		 subflow, subflow->token, thmac, subflow->thmac);
391 
392 	return thmac == subflow->thmac;
393 }
394 
mptcp_subflow_reset(struct sock * ssk)395 void mptcp_subflow_reset(struct sock *ssk)
396 {
397 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
398 	struct sock *sk = subflow->conn;
399 
400 	/* mptcp_mp_fail_no_response() can reach here on an already closed
401 	 * socket
402 	 */
403 	if (ssk->sk_state == TCP_CLOSE)
404 		return;
405 
406 	/* must hold: tcp_done() could drop last reference on parent */
407 	sock_hold(sk);
408 
409 	tcp_send_active_reset(ssk, GFP_ATOMIC);
410 	tcp_done(ssk);
411 	if (!test_and_set_bit(MPTCP_WORK_CLOSE_SUBFLOW, &mptcp_sk(sk)->flags))
412 		mptcp_schedule_work(sk);
413 
414 	sock_put(sk);
415 }
416 
subflow_use_different_dport(struct mptcp_sock * msk,const struct sock * sk)417 static bool subflow_use_different_dport(struct mptcp_sock *msk, const struct sock *sk)
418 {
419 	return inet_sk(sk)->inet_dport != inet_sk((struct sock *)msk)->inet_dport;
420 }
421 
__mptcp_sync_state(struct sock * sk,int state)422 void __mptcp_sync_state(struct sock *sk, int state)
423 {
424 	struct mptcp_subflow_context *subflow;
425 	struct mptcp_sock *msk = mptcp_sk(sk);
426 	struct sock *ssk = msk->first;
427 
428 	subflow = mptcp_subflow_ctx(ssk);
429 	__mptcp_propagate_sndbuf(sk, ssk);
430 	if (!msk->rcvspace_init)
431 		mptcp_rcv_space_init(msk, ssk);
432 
433 	if (sk->sk_state == TCP_SYN_SENT) {
434 		/* subflow->idsn is always available is TCP_SYN_SENT state,
435 		 * even for the FASTOPEN scenarios
436 		 */
437 		WRITE_ONCE(msk->write_seq, subflow->idsn + 1);
438 		WRITE_ONCE(msk->snd_nxt, msk->write_seq);
439 		mptcp_set_state(sk, state);
440 		sk->sk_state_change(sk);
441 	}
442 }
443 
subflow_set_remote_key(struct mptcp_sock * msk,struct mptcp_subflow_context * subflow,const struct mptcp_options_received * mp_opt)444 static void subflow_set_remote_key(struct mptcp_sock *msk,
445 				   struct mptcp_subflow_context *subflow,
446 				   const struct mptcp_options_received *mp_opt)
447 {
448 	/* active MPC subflow will reach here multiple times:
449 	 * at subflow_finish_connect() time and at 4th ack time
450 	 */
451 	if (subflow->remote_key_valid)
452 		return;
453 
454 	subflow->remote_key_valid = 1;
455 	subflow->remote_key = mp_opt->sndr_key;
456 	mptcp_crypto_key_sha(subflow->remote_key, NULL, &subflow->iasn);
457 	subflow->iasn++;
458 
459 	WRITE_ONCE(msk->remote_key, subflow->remote_key);
460 	WRITE_ONCE(msk->ack_seq, subflow->iasn);
461 	WRITE_ONCE(msk->can_ack, true);
462 	atomic64_set(&msk->rcv_wnd_sent, subflow->iasn);
463 }
464 
mptcp_propagate_state(struct sock * sk,struct sock * ssk,struct mptcp_subflow_context * subflow,const struct mptcp_options_received * mp_opt)465 static void mptcp_propagate_state(struct sock *sk, struct sock *ssk,
466 				  struct mptcp_subflow_context *subflow,
467 				  const struct mptcp_options_received *mp_opt)
468 {
469 	struct mptcp_sock *msk = mptcp_sk(sk);
470 
471 	mptcp_data_lock(sk);
472 	if (mp_opt) {
473 		/* Options are available only in the non fallback cases
474 		 * avoid updating rx path fields otherwise
475 		 */
476 		WRITE_ONCE(msk->snd_una, subflow->idsn + 1);
477 		WRITE_ONCE(msk->wnd_end, subflow->idsn + 1 + tcp_sk(ssk)->snd_wnd);
478 		subflow_set_remote_key(msk, subflow, mp_opt);
479 	}
480 
481 	if (!sock_owned_by_user(sk)) {
482 		__mptcp_sync_state(sk, ssk->sk_state);
483 	} else {
484 		msk->pending_state = ssk->sk_state;
485 		__set_bit(MPTCP_SYNC_STATE, &msk->cb_flags);
486 	}
487 	mptcp_data_unlock(sk);
488 }
489 
subflow_finish_connect(struct sock * sk,const struct sk_buff * skb)490 static void subflow_finish_connect(struct sock *sk, const struct sk_buff *skb)
491 {
492 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
493 	struct mptcp_options_received mp_opt;
494 	struct sock *parent = subflow->conn;
495 	struct mptcp_sock *msk;
496 
497 	subflow->icsk_af_ops->sk_rx_dst_set(sk, skb);
498 
499 	/* be sure no special action on any packet other than syn-ack */
500 	if (subflow->conn_finished)
501 		return;
502 
503 	msk = mptcp_sk(parent);
504 	subflow->rel_write_seq = 1;
505 	subflow->conn_finished = 1;
506 	subflow->ssn_offset = TCP_SKB_CB(skb)->seq;
507 	pr_debug("subflow=%p synack seq=%x", subflow, subflow->ssn_offset);
508 
509 	mptcp_get_options(skb, &mp_opt);
510 	if (subflow->request_mptcp) {
511 		if (!(mp_opt.suboptions & OPTION_MPTCP_MPC_SYNACK)) {
512 			MPTCP_INC_STATS(sock_net(sk),
513 					MPTCP_MIB_MPCAPABLEACTIVEFALLBACK);
514 			mptcp_do_fallback(sk);
515 			pr_fallback(msk);
516 			goto fallback;
517 		}
518 
519 		if (mp_opt.suboptions & OPTION_MPTCP_CSUMREQD)
520 			WRITE_ONCE(msk->csum_enabled, true);
521 		if (mp_opt.deny_join_id0)
522 			WRITE_ONCE(msk->pm.remote_deny_join_id0, true);
523 		subflow->mp_capable = 1;
524 		MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_MPCAPABLEACTIVEACK);
525 		mptcp_finish_connect(sk);
526 		mptcp_propagate_state(parent, sk, subflow, &mp_opt);
527 	} else if (subflow->request_join) {
528 		u8 hmac[SHA256_DIGEST_SIZE];
529 
530 		if (!(mp_opt.suboptions & OPTION_MPTCP_MPJ_SYNACK)) {
531 			subflow->reset_reason = MPTCP_RST_EMPTCP;
532 			goto do_reset;
533 		}
534 
535 		subflow->backup = mp_opt.backup;
536 		subflow->thmac = mp_opt.thmac;
537 		subflow->remote_nonce = mp_opt.nonce;
538 		WRITE_ONCE(subflow->remote_id, mp_opt.join_id);
539 		pr_debug("subflow=%p, thmac=%llu, remote_nonce=%u backup=%d",
540 			 subflow, subflow->thmac, subflow->remote_nonce,
541 			 subflow->backup);
542 
543 		if (!subflow_thmac_valid(subflow)) {
544 			MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_JOINACKMAC);
545 			subflow->reset_reason = MPTCP_RST_EMPTCP;
546 			goto do_reset;
547 		}
548 
549 		if (!mptcp_finish_join(sk))
550 			goto do_reset;
551 
552 		subflow_generate_hmac(subflow->local_key, subflow->remote_key,
553 				      subflow->local_nonce,
554 				      subflow->remote_nonce,
555 				      hmac);
556 		memcpy(subflow->hmac, hmac, MPTCPOPT_HMAC_LEN);
557 
558 		subflow->mp_join = 1;
559 		MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_JOINSYNACKRX);
560 
561 		if (subflow_use_different_dport(msk, sk)) {
562 			pr_debug("synack inet_dport=%d %d",
563 				 ntohs(inet_sk(sk)->inet_dport),
564 				 ntohs(inet_sk(parent)->inet_dport));
565 			MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_JOINPORTSYNACKRX);
566 		}
567 	} else if (mptcp_check_fallback(sk)) {
568 fallback:
569 		mptcp_propagate_state(parent, sk, subflow, NULL);
570 	}
571 	return;
572 
573 do_reset:
574 	subflow->reset_transient = 0;
575 	mptcp_subflow_reset(sk);
576 }
577 
subflow_set_local_id(struct mptcp_subflow_context * subflow,int local_id)578 static void subflow_set_local_id(struct mptcp_subflow_context *subflow, int local_id)
579 {
580 	WARN_ON_ONCE(local_id < 0 || local_id > 255);
581 	WRITE_ONCE(subflow->local_id, local_id);
582 }
583 
subflow_chk_local_id(struct sock * sk)584 static int subflow_chk_local_id(struct sock *sk)
585 {
586 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
587 	struct mptcp_sock *msk = mptcp_sk(subflow->conn);
588 	int err;
589 
590 	if (likely(subflow->local_id >= 0))
591 		return 0;
592 
593 	err = mptcp_pm_get_local_id(msk, (struct sock_common *)sk);
594 	if (err < 0)
595 		return err;
596 
597 	subflow_set_local_id(subflow, err);
598 	return 0;
599 }
600 
subflow_rebuild_header(struct sock * sk)601 static int subflow_rebuild_header(struct sock *sk)
602 {
603 	int err = subflow_chk_local_id(sk);
604 
605 	if (unlikely(err < 0))
606 		return err;
607 
608 	return inet_sk_rebuild_header(sk);
609 }
610 
611 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
subflow_v6_rebuild_header(struct sock * sk)612 static int subflow_v6_rebuild_header(struct sock *sk)
613 {
614 	int err = subflow_chk_local_id(sk);
615 
616 	if (unlikely(err < 0))
617 		return err;
618 
619 	return inet6_sk_rebuild_header(sk);
620 }
621 #endif
622 
623 static struct request_sock_ops mptcp_subflow_v4_request_sock_ops __ro_after_init;
624 static struct tcp_request_sock_ops subflow_request_sock_ipv4_ops __ro_after_init;
625 
subflow_v4_conn_request(struct sock * sk,struct sk_buff * skb)626 static int subflow_v4_conn_request(struct sock *sk, struct sk_buff *skb)
627 {
628 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
629 
630 	pr_debug("subflow=%p", subflow);
631 
632 	/* Never answer to SYNs sent to broadcast or multicast */
633 	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
634 		goto drop;
635 
636 	return tcp_conn_request(&mptcp_subflow_v4_request_sock_ops,
637 				&subflow_request_sock_ipv4_ops,
638 				sk, skb);
639 drop:
640 	tcp_listendrop(sk);
641 	return 0;
642 }
643 
subflow_v4_req_destructor(struct request_sock * req)644 static void subflow_v4_req_destructor(struct request_sock *req)
645 {
646 	subflow_req_destructor(req);
647 	tcp_request_sock_ops.destructor(req);
648 }
649 
650 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
651 static struct request_sock_ops mptcp_subflow_v6_request_sock_ops __ro_after_init;
652 static struct tcp_request_sock_ops subflow_request_sock_ipv6_ops __ro_after_init;
653 static struct inet_connection_sock_af_ops subflow_v6_specific __ro_after_init;
654 static struct inet_connection_sock_af_ops subflow_v6m_specific __ro_after_init;
655 static struct proto tcpv6_prot_override __ro_after_init;
656 
subflow_v6_conn_request(struct sock * sk,struct sk_buff * skb)657 static int subflow_v6_conn_request(struct sock *sk, struct sk_buff *skb)
658 {
659 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
660 
661 	pr_debug("subflow=%p", subflow);
662 
663 	if (skb->protocol == htons(ETH_P_IP))
664 		return subflow_v4_conn_request(sk, skb);
665 
666 	if (!ipv6_unicast_destination(skb))
667 		goto drop;
668 
669 	if (ipv6_addr_v4mapped(&ipv6_hdr(skb)->saddr)) {
670 		__IP6_INC_STATS(sock_net(sk), NULL, IPSTATS_MIB_INHDRERRORS);
671 		return 0;
672 	}
673 
674 	return tcp_conn_request(&mptcp_subflow_v6_request_sock_ops,
675 				&subflow_request_sock_ipv6_ops, sk, skb);
676 
677 drop:
678 	tcp_listendrop(sk);
679 	return 0; /* don't send reset */
680 }
681 
subflow_v6_req_destructor(struct request_sock * req)682 static void subflow_v6_req_destructor(struct request_sock *req)
683 {
684 	subflow_req_destructor(req);
685 	tcp6_request_sock_ops.destructor(req);
686 }
687 #endif
688 
mptcp_subflow_reqsk_alloc(const struct request_sock_ops * ops,struct sock * sk_listener,bool attach_listener)689 struct request_sock *mptcp_subflow_reqsk_alloc(const struct request_sock_ops *ops,
690 					       struct sock *sk_listener,
691 					       bool attach_listener)
692 {
693 	if (ops->family == AF_INET)
694 		ops = &mptcp_subflow_v4_request_sock_ops;
695 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
696 	else if (ops->family == AF_INET6)
697 		ops = &mptcp_subflow_v6_request_sock_ops;
698 #endif
699 
700 	return inet_reqsk_alloc(ops, sk_listener, attach_listener);
701 }
702 EXPORT_SYMBOL(mptcp_subflow_reqsk_alloc);
703 
704 /* validate hmac received in third ACK */
subflow_hmac_valid(const struct request_sock * req,const struct mptcp_options_received * mp_opt)705 static bool subflow_hmac_valid(const struct request_sock *req,
706 			       const struct mptcp_options_received *mp_opt)
707 {
708 	const struct mptcp_subflow_request_sock *subflow_req;
709 	u8 hmac[SHA256_DIGEST_SIZE];
710 	struct mptcp_sock *msk;
711 
712 	subflow_req = mptcp_subflow_rsk(req);
713 	msk = subflow_req->msk;
714 	if (!msk)
715 		return false;
716 
717 	subflow_generate_hmac(msk->remote_key, msk->local_key,
718 			      subflow_req->remote_nonce,
719 			      subflow_req->local_nonce, hmac);
720 
721 	return !crypto_memneq(hmac, mp_opt->hmac, MPTCPOPT_HMAC_LEN);
722 }
723 
subflow_ulp_fallback(struct sock * sk,struct mptcp_subflow_context * old_ctx)724 static void subflow_ulp_fallback(struct sock *sk,
725 				 struct mptcp_subflow_context *old_ctx)
726 {
727 	struct inet_connection_sock *icsk = inet_csk(sk);
728 
729 	mptcp_subflow_tcp_fallback(sk, old_ctx);
730 	icsk->icsk_ulp_ops = NULL;
731 	rcu_assign_pointer(icsk->icsk_ulp_data, NULL);
732 	tcp_sk(sk)->is_mptcp = 0;
733 
734 	mptcp_subflow_ops_undo_override(sk);
735 }
736 
mptcp_subflow_drop_ctx(struct sock * ssk)737 void mptcp_subflow_drop_ctx(struct sock *ssk)
738 {
739 	struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(ssk);
740 
741 	if (!ctx)
742 		return;
743 
744 	list_del(&mptcp_subflow_ctx(ssk)->node);
745 	if (inet_csk(ssk)->icsk_ulp_ops) {
746 		subflow_ulp_fallback(ssk, ctx);
747 		if (ctx->conn)
748 			sock_put(ctx->conn);
749 	}
750 
751 	kfree_rcu(ctx, rcu);
752 }
753 
__mptcp_subflow_fully_established(struct mptcp_sock * msk,struct mptcp_subflow_context * subflow,const struct mptcp_options_received * mp_opt)754 void __mptcp_subflow_fully_established(struct mptcp_sock *msk,
755 				       struct mptcp_subflow_context *subflow,
756 				       const struct mptcp_options_received *mp_opt)
757 {
758 	subflow_set_remote_key(msk, subflow, mp_opt);
759 	subflow->fully_established = 1;
760 	WRITE_ONCE(msk->fully_established, true);
761 
762 	if (subflow->is_mptfo)
763 		__mptcp_fastopen_gen_msk_ackseq(msk, subflow, mp_opt);
764 }
765 
subflow_syn_recv_sock(const struct sock * sk,struct sk_buff * skb,struct request_sock * req,struct dst_entry * dst,struct request_sock * req_unhash,bool * own_req)766 static struct sock *subflow_syn_recv_sock(const struct sock *sk,
767 					  struct sk_buff *skb,
768 					  struct request_sock *req,
769 					  struct dst_entry *dst,
770 					  struct request_sock *req_unhash,
771 					  bool *own_req)
772 {
773 	struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk);
774 	struct mptcp_subflow_request_sock *subflow_req;
775 	struct mptcp_options_received mp_opt;
776 	bool fallback, fallback_is_fatal;
777 	struct mptcp_sock *owner;
778 	struct sock *child;
779 
780 	pr_debug("listener=%p, req=%p, conn=%p", listener, req, listener->conn);
781 
782 	/* After child creation we must look for MPC even when options
783 	 * are not parsed
784 	 */
785 	mp_opt.suboptions = 0;
786 
787 	/* hopefully temporary handling for MP_JOIN+syncookie */
788 	subflow_req = mptcp_subflow_rsk(req);
789 	fallback_is_fatal = tcp_rsk(req)->is_mptcp && subflow_req->mp_join;
790 	fallback = !tcp_rsk(req)->is_mptcp;
791 	if (fallback)
792 		goto create_child;
793 
794 	/* if the sk is MP_CAPABLE, we try to fetch the client key */
795 	if (subflow_req->mp_capable) {
796 		/* we can receive and accept an in-window, out-of-order pkt,
797 		 * which may not carry the MP_CAPABLE opt even on mptcp enabled
798 		 * paths: always try to extract the peer key, and fallback
799 		 * for packets missing it.
800 		 * Even OoO DSS packets coming legitly after dropped or
801 		 * reordered MPC will cause fallback, but we don't have other
802 		 * options.
803 		 */
804 		mptcp_get_options(skb, &mp_opt);
805 		if (!(mp_opt.suboptions &
806 		      (OPTION_MPTCP_MPC_SYN | OPTION_MPTCP_MPC_ACK)))
807 			fallback = true;
808 
809 	} else if (subflow_req->mp_join) {
810 		mptcp_get_options(skb, &mp_opt);
811 		if (!(mp_opt.suboptions & OPTION_MPTCP_MPJ_ACK) ||
812 		    !subflow_hmac_valid(req, &mp_opt) ||
813 		    !mptcp_can_accept_new_subflow(subflow_req->msk)) {
814 			SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINACKMAC);
815 			fallback = true;
816 		}
817 	}
818 
819 create_child:
820 	child = listener->icsk_af_ops->syn_recv_sock(sk, skb, req, dst,
821 						     req_unhash, own_req);
822 
823 	if (child && *own_req) {
824 		struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(child);
825 
826 		tcp_rsk(req)->drop_req = false;
827 
828 		/* we need to fallback on ctx allocation failure and on pre-reqs
829 		 * checking above. In the latter scenario we additionally need
830 		 * to reset the context to non MPTCP status.
831 		 */
832 		if (!ctx || fallback) {
833 			if (fallback_is_fatal) {
834 				subflow_add_reset_reason(skb, MPTCP_RST_EMPTCP);
835 				goto dispose_child;
836 			}
837 			goto fallback;
838 		}
839 
840 		/* ssk inherits options of listener sk */
841 		ctx->setsockopt_seq = listener->setsockopt_seq;
842 
843 		if (ctx->mp_capable) {
844 			ctx->conn = mptcp_sk_clone_init(listener->conn, &mp_opt, child, req);
845 			if (!ctx->conn)
846 				goto fallback;
847 
848 			ctx->subflow_id = 1;
849 			owner = mptcp_sk(ctx->conn);
850 			mptcp_pm_new_connection(owner, child, 1);
851 
852 			/* with OoO packets we can reach here without ingress
853 			 * mpc option
854 			 */
855 			if (mp_opt.suboptions & OPTION_MPTCP_MPC_ACK) {
856 				mptcp_pm_fully_established(owner, child);
857 				ctx->pm_notified = 1;
858 			}
859 		} else if (ctx->mp_join) {
860 			owner = subflow_req->msk;
861 			if (!owner) {
862 				subflow_add_reset_reason(skb, MPTCP_RST_EPROHIBIT);
863 				goto dispose_child;
864 			}
865 
866 			/* move the msk reference ownership to the subflow */
867 			subflow_req->msk = NULL;
868 			ctx->conn = (struct sock *)owner;
869 
870 			if (subflow_use_different_sport(owner, sk)) {
871 				pr_debug("ack inet_sport=%d %d",
872 					 ntohs(inet_sk(sk)->inet_sport),
873 					 ntohs(inet_sk((struct sock *)owner)->inet_sport));
874 				if (!mptcp_pm_sport_in_anno_list(owner, sk)) {
875 					SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_MISMATCHPORTACKRX);
876 					goto dispose_child;
877 				}
878 				SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINPORTACKRX);
879 			}
880 
881 			if (!mptcp_finish_join(child))
882 				goto dispose_child;
883 
884 			SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINACKRX);
885 			tcp_rsk(req)->drop_req = true;
886 		}
887 	}
888 
889 	/* check for expected invariant - should never trigger, just help
890 	 * catching eariler subtle bugs
891 	 */
892 	WARN_ON_ONCE(child && *own_req && tcp_sk(child)->is_mptcp &&
893 		     (!mptcp_subflow_ctx(child) ||
894 		      !mptcp_subflow_ctx(child)->conn));
895 	return child;
896 
897 dispose_child:
898 	mptcp_subflow_drop_ctx(child);
899 	tcp_rsk(req)->drop_req = true;
900 	inet_csk_prepare_for_destroy_sock(child);
901 	tcp_done(child);
902 	req->rsk_ops->send_reset(sk, skb);
903 
904 	/* The last child reference will be released by the caller */
905 	return child;
906 
907 fallback:
908 	if (fallback)
909 		SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_MPCAPABLEPASSIVEFALLBACK);
910 	mptcp_subflow_drop_ctx(child);
911 	return child;
912 }
913 
914 static struct inet_connection_sock_af_ops subflow_specific __ro_after_init;
915 static struct proto tcp_prot_override __ro_after_init;
916 
917 enum mapping_status {
918 	MAPPING_OK,
919 	MAPPING_INVALID,
920 	MAPPING_EMPTY,
921 	MAPPING_DATA_FIN,
922 	MAPPING_DUMMY,
923 	MAPPING_BAD_CSUM
924 };
925 
dbg_bad_map(struct mptcp_subflow_context * subflow,u32 ssn)926 static void dbg_bad_map(struct mptcp_subflow_context *subflow, u32 ssn)
927 {
928 	pr_debug("Bad mapping: ssn=%d map_seq=%d map_data_len=%d",
929 		 ssn, subflow->map_subflow_seq, subflow->map_data_len);
930 }
931 
skb_is_fully_mapped(struct sock * ssk,struct sk_buff * skb)932 static bool skb_is_fully_mapped(struct sock *ssk, struct sk_buff *skb)
933 {
934 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
935 	unsigned int skb_consumed;
936 
937 	skb_consumed = tcp_sk(ssk)->copied_seq - TCP_SKB_CB(skb)->seq;
938 	if (WARN_ON_ONCE(skb_consumed >= skb->len))
939 		return true;
940 
941 	return skb->len - skb_consumed <= subflow->map_data_len -
942 					  mptcp_subflow_get_map_offset(subflow);
943 }
944 
validate_mapping(struct sock * ssk,struct sk_buff * skb)945 static bool validate_mapping(struct sock *ssk, struct sk_buff *skb)
946 {
947 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
948 	u32 ssn = tcp_sk(ssk)->copied_seq - subflow->ssn_offset;
949 
950 	if (unlikely(before(ssn, subflow->map_subflow_seq))) {
951 		/* Mapping covers data later in the subflow stream,
952 		 * currently unsupported.
953 		 */
954 		dbg_bad_map(subflow, ssn);
955 		return false;
956 	}
957 	if (unlikely(!before(ssn, subflow->map_subflow_seq +
958 				  subflow->map_data_len))) {
959 		/* Mapping does covers past subflow data, invalid */
960 		dbg_bad_map(subflow, ssn);
961 		return false;
962 	}
963 	return true;
964 }
965 
validate_data_csum(struct sock * ssk,struct sk_buff * skb,bool csum_reqd)966 static enum mapping_status validate_data_csum(struct sock *ssk, struct sk_buff *skb,
967 					      bool csum_reqd)
968 {
969 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
970 	u32 offset, seq, delta;
971 	__sum16 csum;
972 	int len;
973 
974 	if (!csum_reqd)
975 		return MAPPING_OK;
976 
977 	/* mapping already validated on previous traversal */
978 	if (subflow->map_csum_len == subflow->map_data_len)
979 		return MAPPING_OK;
980 
981 	/* traverse the receive queue, ensuring it contains a full
982 	 * DSS mapping and accumulating the related csum.
983 	 * Preserve the accoumlate csum across multiple calls, to compute
984 	 * the csum only once
985 	 */
986 	delta = subflow->map_data_len - subflow->map_csum_len;
987 	for (;;) {
988 		seq = tcp_sk(ssk)->copied_seq + subflow->map_csum_len;
989 		offset = seq - TCP_SKB_CB(skb)->seq;
990 
991 		/* if the current skb has not been accounted yet, csum its contents
992 		 * up to the amount covered by the current DSS
993 		 */
994 		if (offset < skb->len) {
995 			__wsum csum;
996 
997 			len = min(skb->len - offset, delta);
998 			csum = skb_checksum(skb, offset, len, 0);
999 			subflow->map_data_csum = csum_block_add(subflow->map_data_csum, csum,
1000 								subflow->map_csum_len);
1001 
1002 			delta -= len;
1003 			subflow->map_csum_len += len;
1004 		}
1005 		if (delta == 0)
1006 			break;
1007 
1008 		if (skb_queue_is_last(&ssk->sk_receive_queue, skb)) {
1009 			/* if this subflow is closed, the partial mapping
1010 			 * will be never completed; flush the pending skbs, so
1011 			 * that subflow_sched_work_if_closed() can kick in
1012 			 */
1013 			if (unlikely(ssk->sk_state == TCP_CLOSE))
1014 				while ((skb = skb_peek(&ssk->sk_receive_queue)))
1015 					sk_eat_skb(ssk, skb);
1016 
1017 			/* not enough data to validate the csum */
1018 			return MAPPING_EMPTY;
1019 		}
1020 
1021 		/* the DSS mapping for next skbs will be validated later,
1022 		 * when a get_mapping_status call will process such skb
1023 		 */
1024 		skb = skb->next;
1025 	}
1026 
1027 	/* note that 'map_data_len' accounts only for the carried data, does
1028 	 * not include the eventual seq increment due to the data fin,
1029 	 * while the pseudo header requires the original DSS data len,
1030 	 * including that
1031 	 */
1032 	csum = __mptcp_make_csum(subflow->map_seq,
1033 				 subflow->map_subflow_seq,
1034 				 subflow->map_data_len + subflow->map_data_fin,
1035 				 subflow->map_data_csum);
1036 	if (unlikely(csum)) {
1037 		MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DATACSUMERR);
1038 		return MAPPING_BAD_CSUM;
1039 	}
1040 
1041 	subflow->valid_csum_seen = 1;
1042 	return MAPPING_OK;
1043 }
1044 
get_mapping_status(struct sock * ssk,struct mptcp_sock * msk)1045 static enum mapping_status get_mapping_status(struct sock *ssk,
1046 					      struct mptcp_sock *msk)
1047 {
1048 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1049 	bool csum_reqd = READ_ONCE(msk->csum_enabled);
1050 	struct mptcp_ext *mpext;
1051 	struct sk_buff *skb;
1052 	u16 data_len;
1053 	u64 map_seq;
1054 
1055 	skb = skb_peek(&ssk->sk_receive_queue);
1056 	if (!skb)
1057 		return MAPPING_EMPTY;
1058 
1059 	if (mptcp_check_fallback(ssk))
1060 		return MAPPING_DUMMY;
1061 
1062 	mpext = mptcp_get_ext(skb);
1063 	if (!mpext || !mpext->use_map) {
1064 		if (!subflow->map_valid && !skb->len) {
1065 			/* the TCP stack deliver 0 len FIN pkt to the receive
1066 			 * queue, that is the only 0len pkts ever expected here,
1067 			 * and we can admit no mapping only for 0 len pkts
1068 			 */
1069 			if (!(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN))
1070 				WARN_ONCE(1, "0len seq %d:%d flags %x",
1071 					  TCP_SKB_CB(skb)->seq,
1072 					  TCP_SKB_CB(skb)->end_seq,
1073 					  TCP_SKB_CB(skb)->tcp_flags);
1074 			sk_eat_skb(ssk, skb);
1075 			return MAPPING_EMPTY;
1076 		}
1077 
1078 		if (!subflow->map_valid)
1079 			return MAPPING_INVALID;
1080 
1081 		goto validate_seq;
1082 	}
1083 
1084 	trace_get_mapping_status(mpext);
1085 
1086 	data_len = mpext->data_len;
1087 	if (data_len == 0) {
1088 		pr_debug("infinite mapping received");
1089 		MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_INFINITEMAPRX);
1090 		subflow->map_data_len = 0;
1091 		return MAPPING_INVALID;
1092 	}
1093 
1094 	if (mpext->data_fin == 1) {
1095 		if (data_len == 1) {
1096 			bool updated = mptcp_update_rcv_data_fin(msk, mpext->data_seq,
1097 								 mpext->dsn64);
1098 			pr_debug("DATA_FIN with no payload seq=%llu", mpext->data_seq);
1099 			if (subflow->map_valid) {
1100 				/* A DATA_FIN might arrive in a DSS
1101 				 * option before the previous mapping
1102 				 * has been fully consumed. Continue
1103 				 * handling the existing mapping.
1104 				 */
1105 				skb_ext_del(skb, SKB_EXT_MPTCP);
1106 				return MAPPING_OK;
1107 			} else {
1108 				if (updated)
1109 					mptcp_schedule_work((struct sock *)msk);
1110 
1111 				return MAPPING_DATA_FIN;
1112 			}
1113 		} else {
1114 			u64 data_fin_seq = mpext->data_seq + data_len - 1;
1115 
1116 			/* If mpext->data_seq is a 32-bit value, data_fin_seq
1117 			 * must also be limited to 32 bits.
1118 			 */
1119 			if (!mpext->dsn64)
1120 				data_fin_seq &= GENMASK_ULL(31, 0);
1121 
1122 			mptcp_update_rcv_data_fin(msk, data_fin_seq, mpext->dsn64);
1123 			pr_debug("DATA_FIN with mapping seq=%llu dsn64=%d",
1124 				 data_fin_seq, mpext->dsn64);
1125 		}
1126 
1127 		/* Adjust for DATA_FIN using 1 byte of sequence space */
1128 		data_len--;
1129 	}
1130 
1131 	map_seq = mptcp_expand_seq(READ_ONCE(msk->ack_seq), mpext->data_seq, mpext->dsn64);
1132 	WRITE_ONCE(mptcp_sk(subflow->conn)->use_64bit_ack, !!mpext->dsn64);
1133 
1134 	if (subflow->map_valid) {
1135 		/* Allow replacing only with an identical map */
1136 		if (subflow->map_seq == map_seq &&
1137 		    subflow->map_subflow_seq == mpext->subflow_seq &&
1138 		    subflow->map_data_len == data_len &&
1139 		    subflow->map_csum_reqd == mpext->csum_reqd) {
1140 			skb_ext_del(skb, SKB_EXT_MPTCP);
1141 			goto validate_csum;
1142 		}
1143 
1144 		/* If this skb data are fully covered by the current mapping,
1145 		 * the new map would need caching, which is not supported
1146 		 */
1147 		if (skb_is_fully_mapped(ssk, skb)) {
1148 			MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DSSNOMATCH);
1149 			return MAPPING_INVALID;
1150 		}
1151 
1152 		/* will validate the next map after consuming the current one */
1153 		goto validate_csum;
1154 	}
1155 
1156 	subflow->map_seq = map_seq;
1157 	subflow->map_subflow_seq = mpext->subflow_seq;
1158 	subflow->map_data_len = data_len;
1159 	subflow->map_valid = 1;
1160 	subflow->map_data_fin = mpext->data_fin;
1161 	subflow->mpc_map = mpext->mpc_map;
1162 	subflow->map_csum_reqd = mpext->csum_reqd;
1163 	subflow->map_csum_len = 0;
1164 	subflow->map_data_csum = csum_unfold(mpext->csum);
1165 
1166 	/* Cfr RFC 8684 Section 3.3.0 */
1167 	if (unlikely(subflow->map_csum_reqd != csum_reqd))
1168 		return MAPPING_INVALID;
1169 
1170 	pr_debug("new map seq=%llu subflow_seq=%u data_len=%u csum=%d:%u",
1171 		 subflow->map_seq, subflow->map_subflow_seq,
1172 		 subflow->map_data_len, subflow->map_csum_reqd,
1173 		 subflow->map_data_csum);
1174 
1175 validate_seq:
1176 	/* we revalidate valid mapping on new skb, because we must ensure
1177 	 * the current skb is completely covered by the available mapping
1178 	 */
1179 	if (!validate_mapping(ssk, skb)) {
1180 		MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DSSTCPMISMATCH);
1181 		return MAPPING_INVALID;
1182 	}
1183 
1184 	skb_ext_del(skb, SKB_EXT_MPTCP);
1185 
1186 validate_csum:
1187 	return validate_data_csum(ssk, skb, csum_reqd);
1188 }
1189 
mptcp_subflow_discard_data(struct sock * ssk,struct sk_buff * skb,u64 limit)1190 static void mptcp_subflow_discard_data(struct sock *ssk, struct sk_buff *skb,
1191 				       u64 limit)
1192 {
1193 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1194 	bool fin = TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN;
1195 	u32 incr;
1196 
1197 	incr = limit >= skb->len ? skb->len + fin : limit;
1198 
1199 	pr_debug("discarding=%d len=%d seq=%d", incr, skb->len,
1200 		 subflow->map_subflow_seq);
1201 	MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DUPDATA);
1202 	tcp_sk(ssk)->copied_seq += incr;
1203 	if (!before(tcp_sk(ssk)->copied_seq, TCP_SKB_CB(skb)->end_seq))
1204 		sk_eat_skb(ssk, skb);
1205 	if (mptcp_subflow_get_map_offset(subflow) >= subflow->map_data_len)
1206 		subflow->map_valid = 0;
1207 }
1208 
1209 /* sched mptcp worker to remove the subflow if no more data is pending */
subflow_sched_work_if_closed(struct mptcp_sock * msk,struct sock * ssk)1210 static void subflow_sched_work_if_closed(struct mptcp_sock *msk, struct sock *ssk)
1211 {
1212 	if (likely(ssk->sk_state != TCP_CLOSE))
1213 		return;
1214 
1215 	if (skb_queue_empty(&ssk->sk_receive_queue) &&
1216 	    !test_and_set_bit(MPTCP_WORK_CLOSE_SUBFLOW, &msk->flags))
1217 		mptcp_schedule_work((struct sock *)msk);
1218 }
1219 
subflow_can_fallback(struct mptcp_subflow_context * subflow)1220 static bool subflow_can_fallback(struct mptcp_subflow_context *subflow)
1221 {
1222 	struct mptcp_sock *msk = mptcp_sk(subflow->conn);
1223 
1224 	if (subflow->mp_join)
1225 		return false;
1226 	else if (READ_ONCE(msk->csum_enabled))
1227 		return !subflow->valid_csum_seen;
1228 	else
1229 		return !subflow->fully_established;
1230 }
1231 
mptcp_subflow_fail(struct mptcp_sock * msk,struct sock * ssk)1232 static void mptcp_subflow_fail(struct mptcp_sock *msk, struct sock *ssk)
1233 {
1234 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1235 	unsigned long fail_tout;
1236 
1237 	/* greceful failure can happen only on the MPC subflow */
1238 	if (WARN_ON_ONCE(ssk != READ_ONCE(msk->first)))
1239 		return;
1240 
1241 	/* since the close timeout take precedence on the fail one,
1242 	 * no need to start the latter when the first is already set
1243 	 */
1244 	if (sock_flag((struct sock *)msk, SOCK_DEAD))
1245 		return;
1246 
1247 	/* we don't need extreme accuracy here, use a zero fail_tout as special
1248 	 * value meaning no fail timeout at all;
1249 	 */
1250 	fail_tout = jiffies + TCP_RTO_MAX;
1251 	if (!fail_tout)
1252 		fail_tout = 1;
1253 	WRITE_ONCE(subflow->fail_tout, fail_tout);
1254 	tcp_send_ack(ssk);
1255 
1256 	mptcp_reset_tout_timer(msk, subflow->fail_tout);
1257 }
1258 
subflow_check_data_avail(struct sock * ssk)1259 static bool subflow_check_data_avail(struct sock *ssk)
1260 {
1261 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1262 	enum mapping_status status;
1263 	struct mptcp_sock *msk;
1264 	struct sk_buff *skb;
1265 
1266 	if (!skb_peek(&ssk->sk_receive_queue))
1267 		WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_NODATA);
1268 	if (subflow->data_avail)
1269 		return true;
1270 
1271 	msk = mptcp_sk(subflow->conn);
1272 	for (;;) {
1273 		u64 ack_seq;
1274 		u64 old_ack;
1275 
1276 		status = get_mapping_status(ssk, msk);
1277 		trace_subflow_check_data_avail(status, skb_peek(&ssk->sk_receive_queue));
1278 		if (unlikely(status == MAPPING_INVALID || status == MAPPING_DUMMY ||
1279 			     status == MAPPING_BAD_CSUM))
1280 			goto fallback;
1281 
1282 		if (status != MAPPING_OK)
1283 			goto no_data;
1284 
1285 		skb = skb_peek(&ssk->sk_receive_queue);
1286 		if (WARN_ON_ONCE(!skb))
1287 			goto no_data;
1288 
1289 		if (unlikely(!READ_ONCE(msk->can_ack)))
1290 			goto fallback;
1291 
1292 		old_ack = READ_ONCE(msk->ack_seq);
1293 		ack_seq = mptcp_subflow_get_mapped_dsn(subflow);
1294 		pr_debug("msk ack_seq=%llx subflow ack_seq=%llx", old_ack,
1295 			 ack_seq);
1296 		if (unlikely(before64(ack_seq, old_ack))) {
1297 			mptcp_subflow_discard_data(ssk, skb, old_ack - ack_seq);
1298 			continue;
1299 		}
1300 
1301 		WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_DATA_AVAIL);
1302 		break;
1303 	}
1304 	return true;
1305 
1306 no_data:
1307 	subflow_sched_work_if_closed(msk, ssk);
1308 	return false;
1309 
1310 fallback:
1311 	if (!__mptcp_check_fallback(msk)) {
1312 		/* RFC 8684 section 3.7. */
1313 		if (status == MAPPING_BAD_CSUM &&
1314 		    (subflow->mp_join || subflow->valid_csum_seen)) {
1315 			subflow->send_mp_fail = 1;
1316 
1317 			if (!READ_ONCE(msk->allow_infinite_fallback)) {
1318 				subflow->reset_transient = 0;
1319 				subflow->reset_reason = MPTCP_RST_EMIDDLEBOX;
1320 				goto reset;
1321 			}
1322 			mptcp_subflow_fail(msk, ssk);
1323 			WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_DATA_AVAIL);
1324 			return true;
1325 		}
1326 
1327 		if (!subflow_can_fallback(subflow) && subflow->map_data_len) {
1328 			/* fatal protocol error, close the socket.
1329 			 * subflow_error_report() will introduce the appropriate barriers
1330 			 */
1331 			subflow->reset_transient = 0;
1332 			subflow->reset_reason = MPTCP_RST_EMPTCP;
1333 
1334 reset:
1335 			WRITE_ONCE(ssk->sk_err, EBADMSG);
1336 			tcp_set_state(ssk, TCP_CLOSE);
1337 			while ((skb = skb_peek(&ssk->sk_receive_queue)))
1338 				sk_eat_skb(ssk, skb);
1339 			tcp_send_active_reset(ssk, GFP_ATOMIC);
1340 			WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_NODATA);
1341 			return false;
1342 		}
1343 
1344 		mptcp_do_fallback(ssk);
1345 	}
1346 
1347 	skb = skb_peek(&ssk->sk_receive_queue);
1348 	subflow->map_valid = 1;
1349 	subflow->map_seq = READ_ONCE(msk->ack_seq);
1350 	subflow->map_data_len = skb->len;
1351 	subflow->map_subflow_seq = tcp_sk(ssk)->copied_seq - subflow->ssn_offset;
1352 	WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_DATA_AVAIL);
1353 	return true;
1354 }
1355 
mptcp_subflow_data_available(struct sock * sk)1356 bool mptcp_subflow_data_available(struct sock *sk)
1357 {
1358 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1359 
1360 	/* check if current mapping is still valid */
1361 	if (subflow->map_valid &&
1362 	    mptcp_subflow_get_map_offset(subflow) >= subflow->map_data_len) {
1363 		subflow->map_valid = 0;
1364 		WRITE_ONCE(subflow->data_avail, MPTCP_SUBFLOW_NODATA);
1365 
1366 		pr_debug("Done with mapping: seq=%u data_len=%u",
1367 			 subflow->map_subflow_seq,
1368 			 subflow->map_data_len);
1369 	}
1370 
1371 	return subflow_check_data_avail(sk);
1372 }
1373 
1374 /* If ssk has an mptcp parent socket, use the mptcp rcvbuf occupancy,
1375  * not the ssk one.
1376  *
1377  * In mptcp, rwin is about the mptcp-level connection data.
1378  *
1379  * Data that is still on the ssk rx queue can thus be ignored,
1380  * as far as mptcp peer is concerned that data is still inflight.
1381  * DSS ACK is updated when skb is moved to the mptcp rx queue.
1382  */
mptcp_space(const struct sock * ssk,int * space,int * full_space)1383 void mptcp_space(const struct sock *ssk, int *space, int *full_space)
1384 {
1385 	const struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1386 	const struct sock *sk = subflow->conn;
1387 
1388 	*space = __mptcp_space(sk);
1389 	*full_space = mptcp_win_from_space(sk, READ_ONCE(sk->sk_rcvbuf));
1390 }
1391 
subflow_error_report(struct sock * ssk)1392 static void subflow_error_report(struct sock *ssk)
1393 {
1394 	struct sock *sk = mptcp_subflow_ctx(ssk)->conn;
1395 
1396 	/* bail early if this is a no-op, so that we avoid introducing a
1397 	 * problematic lockdep dependency between TCP accept queue lock
1398 	 * and msk socket spinlock
1399 	 */
1400 	if (!sk->sk_socket)
1401 		return;
1402 
1403 	mptcp_data_lock(sk);
1404 	if (!sock_owned_by_user(sk))
1405 		__mptcp_error_report(sk);
1406 	else
1407 		__set_bit(MPTCP_ERROR_REPORT,  &mptcp_sk(sk)->cb_flags);
1408 	mptcp_data_unlock(sk);
1409 }
1410 
subflow_data_ready(struct sock * sk)1411 static void subflow_data_ready(struct sock *sk)
1412 {
1413 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1414 	u16 state = 1 << inet_sk_state_load(sk);
1415 	struct sock *parent = subflow->conn;
1416 	struct mptcp_sock *msk;
1417 
1418 	trace_sk_data_ready(sk);
1419 
1420 	msk = mptcp_sk(parent);
1421 	if (state & TCPF_LISTEN) {
1422 		/* MPJ subflow are removed from accept queue before reaching here,
1423 		 * avoid stray wakeups
1424 		 */
1425 		if (reqsk_queue_empty(&inet_csk(sk)->icsk_accept_queue))
1426 			return;
1427 
1428 		parent->sk_data_ready(parent);
1429 		return;
1430 	}
1431 
1432 	WARN_ON_ONCE(!__mptcp_check_fallback(msk) && !subflow->mp_capable &&
1433 		     !subflow->mp_join && !(state & TCPF_CLOSE));
1434 
1435 	if (mptcp_subflow_data_available(sk))
1436 		mptcp_data_ready(parent, sk);
1437 	else if (unlikely(sk->sk_err))
1438 		subflow_error_report(sk);
1439 }
1440 
subflow_write_space(struct sock * ssk)1441 static void subflow_write_space(struct sock *ssk)
1442 {
1443 	struct sock *sk = mptcp_subflow_ctx(ssk)->conn;
1444 
1445 	mptcp_propagate_sndbuf(sk, ssk);
1446 	mptcp_write_space(sk);
1447 }
1448 
1449 static const struct inet_connection_sock_af_ops *
subflow_default_af_ops(struct sock * sk)1450 subflow_default_af_ops(struct sock *sk)
1451 {
1452 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1453 	if (sk->sk_family == AF_INET6)
1454 		return &subflow_v6_specific;
1455 #endif
1456 	return &subflow_specific;
1457 }
1458 
1459 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
mptcpv6_handle_mapped(struct sock * sk,bool mapped)1460 void mptcpv6_handle_mapped(struct sock *sk, bool mapped)
1461 {
1462 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1463 	struct inet_connection_sock *icsk = inet_csk(sk);
1464 	const struct inet_connection_sock_af_ops *target;
1465 
1466 	target = mapped ? &subflow_v6m_specific : subflow_default_af_ops(sk);
1467 
1468 	pr_debug("subflow=%p family=%d ops=%p target=%p mapped=%d",
1469 		 subflow, sk->sk_family, icsk->icsk_af_ops, target, mapped);
1470 
1471 	if (likely(icsk->icsk_af_ops == target))
1472 		return;
1473 
1474 	subflow->icsk_af_ops = icsk->icsk_af_ops;
1475 	icsk->icsk_af_ops = target;
1476 }
1477 #endif
1478 
mptcp_info2sockaddr(const struct mptcp_addr_info * info,struct sockaddr_storage * addr,unsigned short family)1479 void mptcp_info2sockaddr(const struct mptcp_addr_info *info,
1480 			 struct sockaddr_storage *addr,
1481 			 unsigned short family)
1482 {
1483 	memset(addr, 0, sizeof(*addr));
1484 	addr->ss_family = family;
1485 	if (addr->ss_family == AF_INET) {
1486 		struct sockaddr_in *in_addr = (struct sockaddr_in *)addr;
1487 
1488 		if (info->family == AF_INET)
1489 			in_addr->sin_addr = info->addr;
1490 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1491 		else if (ipv6_addr_v4mapped(&info->addr6))
1492 			in_addr->sin_addr.s_addr = info->addr6.s6_addr32[3];
1493 #endif
1494 		in_addr->sin_port = info->port;
1495 	}
1496 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1497 	else if (addr->ss_family == AF_INET6) {
1498 		struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)addr;
1499 
1500 		if (info->family == AF_INET)
1501 			ipv6_addr_set_v4mapped(info->addr.s_addr,
1502 					       &in6_addr->sin6_addr);
1503 		else
1504 			in6_addr->sin6_addr = info->addr6;
1505 		in6_addr->sin6_port = info->port;
1506 	}
1507 #endif
1508 }
1509 
__mptcp_subflow_connect(struct sock * sk,const struct mptcp_addr_info * loc,const struct mptcp_addr_info * remote)1510 int __mptcp_subflow_connect(struct sock *sk, const struct mptcp_addr_info *loc,
1511 			    const struct mptcp_addr_info *remote)
1512 {
1513 	struct mptcp_sock *msk = mptcp_sk(sk);
1514 	struct mptcp_subflow_context *subflow;
1515 	struct sockaddr_storage addr;
1516 	int remote_id = remote->id;
1517 	int local_id = loc->id;
1518 	int err = -ENOTCONN;
1519 	struct socket *sf;
1520 	struct sock *ssk;
1521 	u32 remote_token;
1522 	int addrlen;
1523 	int ifindex;
1524 	u8 flags;
1525 
1526 	if (!mptcp_is_fully_established(sk))
1527 		goto err_out;
1528 
1529 	err = mptcp_subflow_create_socket(sk, loc->family, &sf);
1530 	if (err)
1531 		goto err_out;
1532 
1533 	ssk = sf->sk;
1534 	subflow = mptcp_subflow_ctx(ssk);
1535 	do {
1536 		get_random_bytes(&subflow->local_nonce, sizeof(u32));
1537 	} while (!subflow->local_nonce);
1538 
1539 	if (local_id)
1540 		subflow_set_local_id(subflow, local_id);
1541 
1542 	mptcp_pm_get_flags_and_ifindex_by_id(msk, local_id,
1543 					     &flags, &ifindex);
1544 	subflow->remote_key_valid = 1;
1545 	subflow->remote_key = msk->remote_key;
1546 	subflow->local_key = msk->local_key;
1547 	subflow->token = msk->token;
1548 	mptcp_info2sockaddr(loc, &addr, ssk->sk_family);
1549 
1550 	addrlen = sizeof(struct sockaddr_in);
1551 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1552 	if (addr.ss_family == AF_INET6)
1553 		addrlen = sizeof(struct sockaddr_in6);
1554 #endif
1555 	mptcp_sockopt_sync(msk, ssk);
1556 
1557 	ssk->sk_bound_dev_if = ifindex;
1558 	err = kernel_bind(sf, (struct sockaddr *)&addr, addrlen);
1559 	if (err)
1560 		goto failed;
1561 
1562 	mptcp_crypto_key_sha(subflow->remote_key, &remote_token, NULL);
1563 	pr_debug("msk=%p remote_token=%u local_id=%d remote_id=%d", msk,
1564 		 remote_token, local_id, remote_id);
1565 	subflow->remote_token = remote_token;
1566 	WRITE_ONCE(subflow->remote_id, remote_id);
1567 	subflow->request_join = 1;
1568 	subflow->request_bkup = !!(flags & MPTCP_PM_ADDR_FLAG_BACKUP);
1569 	subflow->subflow_id = msk->subflow_id++;
1570 	mptcp_info2sockaddr(remote, &addr, ssk->sk_family);
1571 
1572 	sock_hold(ssk);
1573 	list_add_tail(&subflow->node, &msk->conn_list);
1574 	err = kernel_connect(sf, (struct sockaddr *)&addr, addrlen, O_NONBLOCK);
1575 	if (err && err != -EINPROGRESS)
1576 		goto failed_unlink;
1577 
1578 	/* discard the subflow socket */
1579 	mptcp_sock_graft(ssk, sk->sk_socket);
1580 	iput(SOCK_INODE(sf));
1581 	WRITE_ONCE(msk->allow_infinite_fallback, false);
1582 	mptcp_stop_tout_timer(sk);
1583 	return 0;
1584 
1585 failed_unlink:
1586 	list_del(&subflow->node);
1587 	sock_put(mptcp_subflow_tcp_sock(subflow));
1588 
1589 failed:
1590 	subflow->disposable = 1;
1591 	sock_release(sf);
1592 
1593 err_out:
1594 	/* we account subflows before the creation, and this failures will not
1595 	 * be caught by sk_state_change()
1596 	 */
1597 	mptcp_pm_close_subflow(msk);
1598 	return err;
1599 }
1600 
mptcp_attach_cgroup(struct sock * parent,struct sock * child)1601 static void mptcp_attach_cgroup(struct sock *parent, struct sock *child)
1602 {
1603 #ifdef CONFIG_SOCK_CGROUP_DATA
1604 	struct sock_cgroup_data *parent_skcd = &parent->sk_cgrp_data,
1605 				*child_skcd = &child->sk_cgrp_data;
1606 
1607 	/* only the additional subflows created by kworkers have to be modified */
1608 	if (cgroup_id(sock_cgroup_ptr(parent_skcd)) !=
1609 	    cgroup_id(sock_cgroup_ptr(child_skcd))) {
1610 #ifdef CONFIG_MEMCG
1611 		struct mem_cgroup *memcg = parent->sk_memcg;
1612 
1613 		mem_cgroup_sk_free(child);
1614 		if (memcg && css_tryget(&memcg->css))
1615 			child->sk_memcg = memcg;
1616 #endif /* CONFIG_MEMCG */
1617 
1618 		cgroup_sk_free(child_skcd);
1619 		*child_skcd = *parent_skcd;
1620 		cgroup_sk_clone(child_skcd);
1621 	}
1622 #endif /* CONFIG_SOCK_CGROUP_DATA */
1623 }
1624 
mptcp_subflow_ops_override(struct sock * ssk)1625 static void mptcp_subflow_ops_override(struct sock *ssk)
1626 {
1627 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1628 	if (ssk->sk_prot == &tcpv6_prot)
1629 		ssk->sk_prot = &tcpv6_prot_override;
1630 	else
1631 #endif
1632 		ssk->sk_prot = &tcp_prot_override;
1633 }
1634 
mptcp_subflow_ops_undo_override(struct sock * ssk)1635 static void mptcp_subflow_ops_undo_override(struct sock *ssk)
1636 {
1637 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1638 	if (ssk->sk_prot == &tcpv6_prot_override)
1639 		ssk->sk_prot = &tcpv6_prot;
1640 	else
1641 #endif
1642 		ssk->sk_prot = &tcp_prot;
1643 }
1644 
mptcp_subflow_create_socket(struct sock * sk,unsigned short family,struct socket ** new_sock)1645 int mptcp_subflow_create_socket(struct sock *sk, unsigned short family,
1646 				struct socket **new_sock)
1647 {
1648 	struct mptcp_subflow_context *subflow;
1649 	struct net *net = sock_net(sk);
1650 	struct socket *sf;
1651 	int err;
1652 
1653 	/* un-accepted server sockets can reach here - on bad configuration
1654 	 * bail early to avoid greater trouble later
1655 	 */
1656 	if (unlikely(!sk->sk_socket))
1657 		return -EINVAL;
1658 
1659 	err = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP, &sf);
1660 	if (err)
1661 		return err;
1662 
1663 	lock_sock_nested(sf->sk, SINGLE_DEPTH_NESTING);
1664 
1665 	err = security_mptcp_add_subflow(sk, sf->sk);
1666 	if (err)
1667 		goto release_ssk;
1668 
1669 	/* the newly created socket has to be in the same cgroup as its parent */
1670 	mptcp_attach_cgroup(sk, sf->sk);
1671 
1672 	/* kernel sockets do not by default acquire net ref, but TCP timer
1673 	 * needs it.
1674 	 * Update ns_tracker to current stack trace and refcounted tracker.
1675 	 */
1676 	__netns_tracker_free(net, &sf->sk->ns_tracker, false);
1677 	sf->sk->sk_net_refcnt = 1;
1678 	get_net_track(net, &sf->sk->ns_tracker, GFP_KERNEL);
1679 	sock_inuse_add(net, 1);
1680 	err = tcp_set_ulp(sf->sk, "mptcp");
1681 
1682 release_ssk:
1683 	release_sock(sf->sk);
1684 
1685 	if (err) {
1686 		sock_release(sf);
1687 		return err;
1688 	}
1689 
1690 	/* the newly created socket really belongs to the owning MPTCP master
1691 	 * socket, even if for additional subflows the allocation is performed
1692 	 * by a kernel workqueue. Adjust inode references, so that the
1693 	 * procfs/diag interfaces really show this one belonging to the correct
1694 	 * user.
1695 	 */
1696 	SOCK_INODE(sf)->i_ino = SOCK_INODE(sk->sk_socket)->i_ino;
1697 	SOCK_INODE(sf)->i_uid = SOCK_INODE(sk->sk_socket)->i_uid;
1698 	SOCK_INODE(sf)->i_gid = SOCK_INODE(sk->sk_socket)->i_gid;
1699 
1700 	subflow = mptcp_subflow_ctx(sf->sk);
1701 	pr_debug("subflow=%p", subflow);
1702 
1703 	*new_sock = sf;
1704 	sock_hold(sk);
1705 	subflow->conn = sk;
1706 	mptcp_subflow_ops_override(sf->sk);
1707 
1708 	return 0;
1709 }
1710 
subflow_create_ctx(struct sock * sk,gfp_t priority)1711 static struct mptcp_subflow_context *subflow_create_ctx(struct sock *sk,
1712 							gfp_t priority)
1713 {
1714 	struct inet_connection_sock *icsk = inet_csk(sk);
1715 	struct mptcp_subflow_context *ctx;
1716 
1717 	ctx = kzalloc(sizeof(*ctx), priority);
1718 	if (!ctx)
1719 		return NULL;
1720 
1721 	rcu_assign_pointer(icsk->icsk_ulp_data, ctx);
1722 	INIT_LIST_HEAD(&ctx->node);
1723 	INIT_LIST_HEAD(&ctx->delegated_node);
1724 
1725 	pr_debug("subflow=%p", ctx);
1726 
1727 	ctx->tcp_sock = sk;
1728 	WRITE_ONCE(ctx->local_id, -1);
1729 
1730 	return ctx;
1731 }
1732 
__subflow_state_change(struct sock * sk)1733 static void __subflow_state_change(struct sock *sk)
1734 {
1735 	struct socket_wq *wq;
1736 
1737 	rcu_read_lock();
1738 	wq = rcu_dereference(sk->sk_wq);
1739 	if (skwq_has_sleeper(wq))
1740 		wake_up_interruptible_all(&wq->wait);
1741 	rcu_read_unlock();
1742 }
1743 
subflow_is_done(const struct sock * sk)1744 static bool subflow_is_done(const struct sock *sk)
1745 {
1746 	return sk->sk_shutdown & RCV_SHUTDOWN || sk->sk_state == TCP_CLOSE;
1747 }
1748 
subflow_state_change(struct sock * sk)1749 static void subflow_state_change(struct sock *sk)
1750 {
1751 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1752 	struct sock *parent = subflow->conn;
1753 	struct mptcp_sock *msk;
1754 
1755 	__subflow_state_change(sk);
1756 
1757 	msk = mptcp_sk(parent);
1758 	if (subflow_simultaneous_connect(sk)) {
1759 		mptcp_do_fallback(sk);
1760 		pr_fallback(msk);
1761 		subflow->conn_finished = 1;
1762 		mptcp_propagate_state(parent, sk, subflow, NULL);
1763 	}
1764 
1765 	/* as recvmsg() does not acquire the subflow socket for ssk selection
1766 	 * a fin packet carrying a DSS can be unnoticed if we don't trigger
1767 	 * the data available machinery here.
1768 	 */
1769 	if (mptcp_subflow_data_available(sk))
1770 		mptcp_data_ready(parent, sk);
1771 	else if (unlikely(sk->sk_err))
1772 		subflow_error_report(sk);
1773 
1774 	subflow_sched_work_if_closed(mptcp_sk(parent), sk);
1775 
1776 	/* when the fallback subflow closes the rx side, trigger a 'dummy'
1777 	 * ingress data fin, so that the msk state will follow along
1778 	 */
1779 	if (__mptcp_check_fallback(msk) && subflow_is_done(sk) && msk->first == sk &&
1780 	    mptcp_update_rcv_data_fin(msk, READ_ONCE(msk->ack_seq), true))
1781 		mptcp_schedule_work(parent);
1782 }
1783 
mptcp_subflow_queue_clean(struct sock * listener_sk,struct sock * listener_ssk)1784 void mptcp_subflow_queue_clean(struct sock *listener_sk, struct sock *listener_ssk)
1785 {
1786 	struct request_sock_queue *queue = &inet_csk(listener_ssk)->icsk_accept_queue;
1787 	struct request_sock *req, *head, *tail;
1788 	struct mptcp_subflow_context *subflow;
1789 	struct sock *sk, *ssk;
1790 
1791 	/* Due to lock dependencies no relevant lock can be acquired under rskq_lock.
1792 	 * Splice the req list, so that accept() can not reach the pending ssk after
1793 	 * the listener socket is released below.
1794 	 */
1795 	spin_lock_bh(&queue->rskq_lock);
1796 	head = queue->rskq_accept_head;
1797 	tail = queue->rskq_accept_tail;
1798 	queue->rskq_accept_head = NULL;
1799 	queue->rskq_accept_tail = NULL;
1800 	spin_unlock_bh(&queue->rskq_lock);
1801 
1802 	if (!head)
1803 		return;
1804 
1805 	/* can't acquire the msk socket lock under the subflow one,
1806 	 * or will cause ABBA deadlock
1807 	 */
1808 	release_sock(listener_ssk);
1809 
1810 	for (req = head; req; req = req->dl_next) {
1811 		ssk = req->sk;
1812 		if (!sk_is_mptcp(ssk))
1813 			continue;
1814 
1815 		subflow = mptcp_subflow_ctx(ssk);
1816 		if (!subflow || !subflow->conn)
1817 			continue;
1818 
1819 		sk = subflow->conn;
1820 		sock_hold(sk);
1821 
1822 		lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
1823 		__mptcp_unaccepted_force_close(sk);
1824 		release_sock(sk);
1825 
1826 		/* lockdep will report a false positive ABBA deadlock
1827 		 * between cancel_work_sync and the listener socket.
1828 		 * The involved locks belong to different sockets WRT
1829 		 * the existing AB chain.
1830 		 * Using a per socket key is problematic as key
1831 		 * deregistration requires process context and must be
1832 		 * performed at socket disposal time, in atomic
1833 		 * context.
1834 		 * Just tell lockdep to consider the listener socket
1835 		 * released here.
1836 		 */
1837 		mutex_release(&listener_sk->sk_lock.dep_map, _RET_IP_);
1838 		mptcp_cancel_work(sk);
1839 		mutex_acquire(&listener_sk->sk_lock.dep_map, 0, 0, _RET_IP_);
1840 
1841 		sock_put(sk);
1842 	}
1843 
1844 	/* we are still under the listener msk socket lock */
1845 	lock_sock_nested(listener_ssk, SINGLE_DEPTH_NESTING);
1846 
1847 	/* restore the listener queue, to let the TCP code clean it up */
1848 	spin_lock_bh(&queue->rskq_lock);
1849 	WARN_ON_ONCE(queue->rskq_accept_head);
1850 	queue->rskq_accept_head = head;
1851 	queue->rskq_accept_tail = tail;
1852 	spin_unlock_bh(&queue->rskq_lock);
1853 }
1854 
subflow_ulp_init(struct sock * sk)1855 static int subflow_ulp_init(struct sock *sk)
1856 {
1857 	struct inet_connection_sock *icsk = inet_csk(sk);
1858 	struct mptcp_subflow_context *ctx;
1859 	struct tcp_sock *tp = tcp_sk(sk);
1860 	int err = 0;
1861 
1862 	/* disallow attaching ULP to a socket unless it has been
1863 	 * created with sock_create_kern()
1864 	 */
1865 	if (!sk->sk_kern_sock) {
1866 		err = -EOPNOTSUPP;
1867 		goto out;
1868 	}
1869 
1870 	ctx = subflow_create_ctx(sk, GFP_KERNEL);
1871 	if (!ctx) {
1872 		err = -ENOMEM;
1873 		goto out;
1874 	}
1875 
1876 	pr_debug("subflow=%p, family=%d", ctx, sk->sk_family);
1877 
1878 	tp->is_mptcp = 1;
1879 	ctx->icsk_af_ops = icsk->icsk_af_ops;
1880 	icsk->icsk_af_ops = subflow_default_af_ops(sk);
1881 	ctx->tcp_state_change = sk->sk_state_change;
1882 	ctx->tcp_error_report = sk->sk_error_report;
1883 
1884 	WARN_ON_ONCE(sk->sk_data_ready != sock_def_readable);
1885 	WARN_ON_ONCE(sk->sk_write_space != sk_stream_write_space);
1886 
1887 	sk->sk_data_ready = subflow_data_ready;
1888 	sk->sk_write_space = subflow_write_space;
1889 	sk->sk_state_change = subflow_state_change;
1890 	sk->sk_error_report = subflow_error_report;
1891 out:
1892 	return err;
1893 }
1894 
subflow_ulp_release(struct sock * ssk)1895 static void subflow_ulp_release(struct sock *ssk)
1896 {
1897 	struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(ssk);
1898 	bool release = true;
1899 	struct sock *sk;
1900 
1901 	if (!ctx)
1902 		return;
1903 
1904 	sk = ctx->conn;
1905 	if (sk) {
1906 		/* if the msk has been orphaned, keep the ctx
1907 		 * alive, will be freed by __mptcp_close_ssk(),
1908 		 * when the subflow is still unaccepted
1909 		 */
1910 		release = ctx->disposable || list_empty(&ctx->node);
1911 
1912 		/* inet_child_forget() does not call sk_state_change(),
1913 		 * explicitly trigger the socket close machinery
1914 		 */
1915 		if (!release && !test_and_set_bit(MPTCP_WORK_CLOSE_SUBFLOW,
1916 						  &mptcp_sk(sk)->flags))
1917 			mptcp_schedule_work(sk);
1918 		sock_put(sk);
1919 	}
1920 
1921 	mptcp_subflow_ops_undo_override(ssk);
1922 	if (release)
1923 		kfree_rcu(ctx, rcu);
1924 }
1925 
subflow_ulp_clone(const struct request_sock * req,struct sock * newsk,const gfp_t priority)1926 static void subflow_ulp_clone(const struct request_sock *req,
1927 			      struct sock *newsk,
1928 			      const gfp_t priority)
1929 {
1930 	struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
1931 	struct mptcp_subflow_context *old_ctx = mptcp_subflow_ctx(newsk);
1932 	struct mptcp_subflow_context *new_ctx;
1933 
1934 	if (!tcp_rsk(req)->is_mptcp ||
1935 	    (!subflow_req->mp_capable && !subflow_req->mp_join)) {
1936 		subflow_ulp_fallback(newsk, old_ctx);
1937 		return;
1938 	}
1939 
1940 	new_ctx = subflow_create_ctx(newsk, priority);
1941 	if (!new_ctx) {
1942 		subflow_ulp_fallback(newsk, old_ctx);
1943 		return;
1944 	}
1945 
1946 	new_ctx->conn_finished = 1;
1947 	new_ctx->icsk_af_ops = old_ctx->icsk_af_ops;
1948 	new_ctx->tcp_state_change = old_ctx->tcp_state_change;
1949 	new_ctx->tcp_error_report = old_ctx->tcp_error_report;
1950 	new_ctx->rel_write_seq = 1;
1951 	new_ctx->tcp_sock = newsk;
1952 
1953 	if (subflow_req->mp_capable) {
1954 		/* see comments in subflow_syn_recv_sock(), MPTCP connection
1955 		 * is fully established only after we receive the remote key
1956 		 */
1957 		new_ctx->mp_capable = 1;
1958 		new_ctx->local_key = subflow_req->local_key;
1959 		new_ctx->token = subflow_req->token;
1960 		new_ctx->ssn_offset = subflow_req->ssn_offset;
1961 		new_ctx->idsn = subflow_req->idsn;
1962 
1963 		/* this is the first subflow, id is always 0 */
1964 		subflow_set_local_id(new_ctx, 0);
1965 	} else if (subflow_req->mp_join) {
1966 		new_ctx->ssn_offset = subflow_req->ssn_offset;
1967 		new_ctx->mp_join = 1;
1968 		new_ctx->fully_established = 1;
1969 		new_ctx->remote_key_valid = 1;
1970 		new_ctx->backup = subflow_req->backup;
1971 		WRITE_ONCE(new_ctx->remote_id, subflow_req->remote_id);
1972 		new_ctx->token = subflow_req->token;
1973 		new_ctx->thmac = subflow_req->thmac;
1974 
1975 		/* the subflow req id is valid, fetched via subflow_check_req()
1976 		 * and subflow_token_join_request()
1977 		 */
1978 		subflow_set_local_id(new_ctx, subflow_req->local_id);
1979 	}
1980 }
1981 
tcp_release_cb_override(struct sock * ssk)1982 static void tcp_release_cb_override(struct sock *ssk)
1983 {
1984 	struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
1985 	long status;
1986 
1987 	/* process and clear all the pending actions, but leave the subflow into
1988 	 * the napi queue. To respect locking, only the same CPU that originated
1989 	 * the action can touch the list. mptcp_napi_poll will take care of it.
1990 	 */
1991 	status = set_mask_bits(&subflow->delegated_status, MPTCP_DELEGATE_ACTIONS_MASK, 0);
1992 	if (status)
1993 		mptcp_subflow_process_delegated(ssk, status);
1994 
1995 	tcp_release_cb(ssk);
1996 }
1997 
tcp_abort_override(struct sock * ssk,int err)1998 static int tcp_abort_override(struct sock *ssk, int err)
1999 {
2000 	/* closing a listener subflow requires a great deal of care.
2001 	 * keep it simple and just prevent such operation
2002 	 */
2003 	if (inet_sk_state_load(ssk) == TCP_LISTEN)
2004 		return -EINVAL;
2005 
2006 	return tcp_abort(ssk, err);
2007 }
2008 
2009 static struct tcp_ulp_ops subflow_ulp_ops __read_mostly = {
2010 	.name		= "mptcp",
2011 	.owner		= THIS_MODULE,
2012 	.init		= subflow_ulp_init,
2013 	.release	= subflow_ulp_release,
2014 	.clone		= subflow_ulp_clone,
2015 };
2016 
subflow_ops_init(struct request_sock_ops * subflow_ops)2017 static int subflow_ops_init(struct request_sock_ops *subflow_ops)
2018 {
2019 	subflow_ops->obj_size = sizeof(struct mptcp_subflow_request_sock);
2020 
2021 	subflow_ops->slab = kmem_cache_create(subflow_ops->slab_name,
2022 					      subflow_ops->obj_size, 0,
2023 					      SLAB_ACCOUNT |
2024 					      SLAB_TYPESAFE_BY_RCU,
2025 					      NULL);
2026 	if (!subflow_ops->slab)
2027 		return -ENOMEM;
2028 
2029 	return 0;
2030 }
2031 
mptcp_subflow_init(void)2032 void __init mptcp_subflow_init(void)
2033 {
2034 	mptcp_subflow_v4_request_sock_ops = tcp_request_sock_ops;
2035 	mptcp_subflow_v4_request_sock_ops.slab_name = "request_sock_subflow_v4";
2036 	mptcp_subflow_v4_request_sock_ops.destructor = subflow_v4_req_destructor;
2037 
2038 	if (subflow_ops_init(&mptcp_subflow_v4_request_sock_ops) != 0)
2039 		panic("MPTCP: failed to init subflow v4 request sock ops\n");
2040 
2041 	subflow_request_sock_ipv4_ops = tcp_request_sock_ipv4_ops;
2042 	subflow_request_sock_ipv4_ops.route_req = subflow_v4_route_req;
2043 	subflow_request_sock_ipv4_ops.send_synack = subflow_v4_send_synack;
2044 
2045 	subflow_specific = ipv4_specific;
2046 	subflow_specific.conn_request = subflow_v4_conn_request;
2047 	subflow_specific.syn_recv_sock = subflow_syn_recv_sock;
2048 	subflow_specific.sk_rx_dst_set = subflow_finish_connect;
2049 	subflow_specific.rebuild_header = subflow_rebuild_header;
2050 
2051 	tcp_prot_override = tcp_prot;
2052 	tcp_prot_override.release_cb = tcp_release_cb_override;
2053 	tcp_prot_override.diag_destroy = tcp_abort_override;
2054 
2055 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
2056 	/* In struct mptcp_subflow_request_sock, we assume the TCP request sock
2057 	 * structures for v4 and v6 have the same size. It should not changed in
2058 	 * the future but better to make sure to be warned if it is no longer
2059 	 * the case.
2060 	 */
2061 	BUILD_BUG_ON(sizeof(struct tcp_request_sock) != sizeof(struct tcp6_request_sock));
2062 
2063 	mptcp_subflow_v6_request_sock_ops = tcp6_request_sock_ops;
2064 	mptcp_subflow_v6_request_sock_ops.slab_name = "request_sock_subflow_v6";
2065 	mptcp_subflow_v6_request_sock_ops.destructor = subflow_v6_req_destructor;
2066 
2067 	if (subflow_ops_init(&mptcp_subflow_v6_request_sock_ops) != 0)
2068 		panic("MPTCP: failed to init subflow v6 request sock ops\n");
2069 
2070 	subflow_request_sock_ipv6_ops = tcp_request_sock_ipv6_ops;
2071 	subflow_request_sock_ipv6_ops.route_req = subflow_v6_route_req;
2072 	subflow_request_sock_ipv6_ops.send_synack = subflow_v6_send_synack;
2073 
2074 	subflow_v6_specific = ipv6_specific;
2075 	subflow_v6_specific.conn_request = subflow_v6_conn_request;
2076 	subflow_v6_specific.syn_recv_sock = subflow_syn_recv_sock;
2077 	subflow_v6_specific.sk_rx_dst_set = subflow_finish_connect;
2078 	subflow_v6_specific.rebuild_header = subflow_v6_rebuild_header;
2079 
2080 	subflow_v6m_specific = subflow_v6_specific;
2081 	subflow_v6m_specific.queue_xmit = ipv4_specific.queue_xmit;
2082 	subflow_v6m_specific.send_check = ipv4_specific.send_check;
2083 	subflow_v6m_specific.net_header_len = ipv4_specific.net_header_len;
2084 	subflow_v6m_specific.mtu_reduced = ipv4_specific.mtu_reduced;
2085 	subflow_v6m_specific.net_frag_header_len = 0;
2086 	subflow_v6m_specific.rebuild_header = subflow_rebuild_header;
2087 
2088 	tcpv6_prot_override = tcpv6_prot;
2089 	tcpv6_prot_override.release_cb = tcp_release_cb_override;
2090 	tcpv6_prot_override.diag_destroy = tcp_abort_override;
2091 #endif
2092 
2093 	mptcp_diag_subflow_init(&subflow_ulp_ops);
2094 
2095 	if (tcp_register_ulp(&subflow_ulp_ops) != 0)
2096 		panic("MPTCP: failed to register subflows to ULP\n");
2097 }
2098