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/sha.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 #endif
22 #include <net/mptcp.h>
23 #include <uapi/linux/mptcp.h>
24 #include "protocol.h"
25 #include "mib.h"
26
SUBFLOW_REQ_INC_STATS(struct request_sock * req,enum linux_mptcp_mib_field field)27 static void SUBFLOW_REQ_INC_STATS(struct request_sock *req,
28 enum linux_mptcp_mib_field field)
29 {
30 MPTCP_INC_STATS(sock_net(req_to_sk(req)), field);
31 }
32
subflow_req_destructor(struct request_sock * req)33 static void subflow_req_destructor(struct request_sock *req)
34 {
35 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
36
37 pr_debug("subflow_req=%p", subflow_req);
38
39 if (subflow_req->msk)
40 sock_put((struct sock *)subflow_req->msk);
41
42 mptcp_token_destroy_request(req);
43 tcp_request_sock_ops.destructor(req);
44 }
45
subflow_generate_hmac(u64 key1,u64 key2,u32 nonce1,u32 nonce2,void * hmac)46 static void subflow_generate_hmac(u64 key1, u64 key2, u32 nonce1, u32 nonce2,
47 void *hmac)
48 {
49 u8 msg[8];
50
51 put_unaligned_be32(nonce1, &msg[0]);
52 put_unaligned_be32(nonce2, &msg[4]);
53
54 mptcp_crypto_hmac_sha(key1, key2, msg, 8, hmac);
55 }
56
mptcp_can_accept_new_subflow(const struct mptcp_sock * msk)57 static bool mptcp_can_accept_new_subflow(const struct mptcp_sock *msk)
58 {
59 return mptcp_is_fully_established((void *)msk) &&
60 READ_ONCE(msk->pm.accept_subflow);
61 }
62
63 /* validate received token and create truncated hmac and nonce for SYN-ACK */
subflow_token_join_request(struct request_sock * req,const struct sk_buff * skb)64 static struct mptcp_sock *subflow_token_join_request(struct request_sock *req,
65 const struct sk_buff *skb)
66 {
67 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
68 u8 hmac[SHA256_DIGEST_SIZE];
69 struct mptcp_sock *msk;
70 int local_id;
71
72 msk = mptcp_token_get_sock(sock_net(req_to_sk(req)), subflow_req->token);
73 if (!msk) {
74 SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINNOTOKEN);
75 return NULL;
76 }
77
78 local_id = mptcp_pm_get_local_id(msk, (struct sock_common *)req);
79 if (local_id < 0) {
80 sock_put((struct sock *)msk);
81 return NULL;
82 }
83 subflow_req->local_id = local_id;
84
85 get_random_bytes(&subflow_req->local_nonce, sizeof(u32));
86
87 subflow_generate_hmac(msk->local_key, msk->remote_key,
88 subflow_req->local_nonce,
89 subflow_req->remote_nonce, hmac);
90
91 subflow_req->thmac = get_unaligned_be64(hmac);
92 return msk;
93 }
94
__subflow_init_req(struct request_sock * req,const struct sock * sk_listener)95 static int __subflow_init_req(struct request_sock *req, const struct sock *sk_listener)
96 {
97 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
98
99 subflow_req->mp_capable = 0;
100 subflow_req->mp_join = 0;
101 subflow_req->msk = NULL;
102 mptcp_token_init_request(req);
103
104 #ifdef CONFIG_TCP_MD5SIG
105 /* no MPTCP if MD5SIG is enabled on this socket or we may run out of
106 * TCP option space.
107 */
108 if (rcu_access_pointer(tcp_sk(sk_listener)->md5sig_info))
109 return -EINVAL;
110 #endif
111
112 return 0;
113 }
114
subflow_init_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)115 static void subflow_init_req(struct request_sock *req,
116 const struct sock *sk_listener,
117 struct sk_buff *skb)
118 {
119 struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk_listener);
120 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
121 struct mptcp_options_received mp_opt;
122 int ret;
123
124 pr_debug("subflow_req=%p, listener=%p", subflow_req, listener);
125
126 ret = __subflow_init_req(req, sk_listener);
127 if (ret)
128 return;
129
130 mptcp_get_options(skb, &mp_opt);
131
132 if (mp_opt.mp_capable) {
133 SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_MPCAPABLEPASSIVE);
134
135 if (mp_opt.mp_join)
136 return;
137 } else if (mp_opt.mp_join) {
138 SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINSYNRX);
139 }
140
141 if (mp_opt.mp_capable && listener->request_mptcp) {
142 int err, retries = 4;
143
144 subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq;
145 again:
146 do {
147 get_random_bytes(&subflow_req->local_key, sizeof(subflow_req->local_key));
148 } while (subflow_req->local_key == 0);
149
150 if (unlikely(req->syncookie)) {
151 mptcp_crypto_key_sha(subflow_req->local_key,
152 &subflow_req->token,
153 &subflow_req->idsn);
154 if (mptcp_token_exists(subflow_req->token)) {
155 if (retries-- > 0)
156 goto again;
157 } else {
158 subflow_req->mp_capable = 1;
159 }
160 return;
161 }
162
163 err = mptcp_token_new_request(req);
164 if (err == 0)
165 subflow_req->mp_capable = 1;
166 else if (retries-- > 0)
167 goto again;
168
169 } else if (mp_opt.mp_join && listener->request_mptcp) {
170 subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq;
171 subflow_req->mp_join = 1;
172 subflow_req->backup = mp_opt.backup;
173 subflow_req->remote_id = mp_opt.join_id;
174 subflow_req->token = mp_opt.token;
175 subflow_req->remote_nonce = mp_opt.nonce;
176 subflow_req->msk = subflow_token_join_request(req, skb);
177
178 if (unlikely(req->syncookie) && subflow_req->msk) {
179 if (mptcp_can_accept_new_subflow(subflow_req->msk))
180 subflow_init_req_cookie_join_save(subflow_req, skb);
181 }
182
183 pr_debug("token=%u, remote_nonce=%u msk=%p", subflow_req->token,
184 subflow_req->remote_nonce, subflow_req->msk);
185 }
186 }
187
mptcp_subflow_init_cookie_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)188 int mptcp_subflow_init_cookie_req(struct request_sock *req,
189 const struct sock *sk_listener,
190 struct sk_buff *skb)
191 {
192 struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk_listener);
193 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
194 struct mptcp_options_received mp_opt;
195 int err;
196
197 err = __subflow_init_req(req, sk_listener);
198 if (err)
199 return err;
200
201 mptcp_get_options(skb, &mp_opt);
202
203 if (mp_opt.mp_capable && mp_opt.mp_join)
204 return -EINVAL;
205
206 if (mp_opt.mp_capable && listener->request_mptcp) {
207 if (mp_opt.sndr_key == 0)
208 return -EINVAL;
209
210 subflow_req->local_key = mp_opt.rcvr_key;
211 err = mptcp_token_new_request(req);
212 if (err)
213 return err;
214
215 subflow_req->mp_capable = 1;
216 subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq - 1;
217 } else if (mp_opt.mp_join && listener->request_mptcp) {
218 if (!mptcp_token_join_cookie_init_state(subflow_req, skb))
219 return -EINVAL;
220
221 if (mptcp_can_accept_new_subflow(subflow_req->msk))
222 subflow_req->mp_join = 1;
223
224 subflow_req->ssn_offset = TCP_SKB_CB(skb)->seq - 1;
225 }
226
227 return 0;
228 }
229 EXPORT_SYMBOL_GPL(mptcp_subflow_init_cookie_req);
230
subflow_v4_init_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)231 static void subflow_v4_init_req(struct request_sock *req,
232 const struct sock *sk_listener,
233 struct sk_buff *skb)
234 {
235 tcp_rsk(req)->is_mptcp = 1;
236
237 tcp_request_sock_ipv4_ops.init_req(req, sk_listener, skb);
238
239 subflow_init_req(req, sk_listener, skb);
240 }
241
242 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
subflow_v6_init_req(struct request_sock * req,const struct sock * sk_listener,struct sk_buff * skb)243 static void subflow_v6_init_req(struct request_sock *req,
244 const struct sock *sk_listener,
245 struct sk_buff *skb)
246 {
247 tcp_rsk(req)->is_mptcp = 1;
248
249 tcp_request_sock_ipv6_ops.init_req(req, sk_listener, skb);
250
251 subflow_init_req(req, sk_listener, skb);
252 }
253 #endif
254
255 /* validate received truncated hmac and create hmac for third ACK */
subflow_thmac_valid(struct mptcp_subflow_context * subflow)256 static bool subflow_thmac_valid(struct mptcp_subflow_context *subflow)
257 {
258 u8 hmac[SHA256_DIGEST_SIZE];
259 u64 thmac;
260
261 subflow_generate_hmac(subflow->remote_key, subflow->local_key,
262 subflow->remote_nonce, subflow->local_nonce,
263 hmac);
264
265 thmac = get_unaligned_be64(hmac);
266 pr_debug("subflow=%p, token=%u, thmac=%llu, subflow->thmac=%llu\n",
267 subflow, subflow->token,
268 (unsigned long long)thmac,
269 (unsigned long long)subflow->thmac);
270
271 return thmac == subflow->thmac;
272 }
273
mptcp_subflow_reset(struct sock * ssk)274 void mptcp_subflow_reset(struct sock *ssk)
275 {
276 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
277 struct sock *sk = subflow->conn;
278
279 tcp_set_state(ssk, TCP_CLOSE);
280 tcp_send_active_reset(ssk, GFP_ATOMIC);
281 tcp_done(ssk);
282 if (!test_and_set_bit(MPTCP_WORK_CLOSE_SUBFLOW, &mptcp_sk(sk)->flags) &&
283 schedule_work(&mptcp_sk(sk)->work))
284 sock_hold(sk);
285 }
286
subflow_finish_connect(struct sock * sk,const struct sk_buff * skb)287 static void subflow_finish_connect(struct sock *sk, const struct sk_buff *skb)
288 {
289 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
290 struct mptcp_options_received mp_opt;
291 struct sock *parent = subflow->conn;
292
293 subflow->icsk_af_ops->sk_rx_dst_set(sk, skb);
294
295 if (inet_sk_state_load(parent) == TCP_SYN_SENT) {
296 inet_sk_state_store(parent, TCP_ESTABLISHED);
297 parent->sk_state_change(parent);
298 }
299
300 /* be sure no special action on any packet other than syn-ack */
301 if (subflow->conn_finished)
302 return;
303
304 subflow->rel_write_seq = 1;
305 subflow->conn_finished = 1;
306 subflow->ssn_offset = TCP_SKB_CB(skb)->seq;
307 pr_debug("subflow=%p synack seq=%x", subflow, subflow->ssn_offset);
308
309 mptcp_get_options(skb, &mp_opt);
310 if (subflow->request_mptcp) {
311 if (!mp_opt.mp_capable) {
312 MPTCP_INC_STATS(sock_net(sk),
313 MPTCP_MIB_MPCAPABLEACTIVEFALLBACK);
314 mptcp_do_fallback(sk);
315 pr_fallback(mptcp_sk(subflow->conn));
316 goto fallback;
317 }
318
319 subflow->mp_capable = 1;
320 subflow->can_ack = 1;
321 subflow->remote_key = mp_opt.sndr_key;
322 pr_debug("subflow=%p, remote_key=%llu", subflow,
323 subflow->remote_key);
324 mptcp_finish_connect(sk);
325 } else if (subflow->request_join) {
326 u8 hmac[SHA256_DIGEST_SIZE];
327
328 if (!mp_opt.mp_join)
329 goto do_reset;
330
331 subflow->thmac = mp_opt.thmac;
332 subflow->remote_nonce = mp_opt.nonce;
333 pr_debug("subflow=%p, thmac=%llu, remote_nonce=%u", subflow,
334 subflow->thmac, subflow->remote_nonce);
335
336 if (!subflow_thmac_valid(subflow)) {
337 MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_JOINACKMAC);
338 goto do_reset;
339 }
340
341 if (!mptcp_finish_join(sk))
342 goto do_reset;
343
344 subflow_generate_hmac(subflow->local_key, subflow->remote_key,
345 subflow->local_nonce,
346 subflow->remote_nonce,
347 hmac);
348 memcpy(subflow->hmac, hmac, MPTCPOPT_HMAC_LEN);
349
350 subflow->mp_join = 1;
351 MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_JOINSYNACKRX);
352 } else if (mptcp_check_fallback(sk)) {
353 fallback:
354 mptcp_rcv_space_init(mptcp_sk(parent), sk);
355 }
356 return;
357
358 do_reset:
359 mptcp_subflow_reset(sk);
360 }
361
362 struct request_sock_ops mptcp_subflow_request_sock_ops;
363 EXPORT_SYMBOL_GPL(mptcp_subflow_request_sock_ops);
364 static struct tcp_request_sock_ops subflow_request_sock_ipv4_ops;
365
subflow_v4_conn_request(struct sock * sk,struct sk_buff * skb)366 static int subflow_v4_conn_request(struct sock *sk, struct sk_buff *skb)
367 {
368 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
369
370 pr_debug("subflow=%p", subflow);
371
372 /* Never answer to SYNs sent to broadcast or multicast */
373 if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
374 goto drop;
375
376 return tcp_conn_request(&mptcp_subflow_request_sock_ops,
377 &subflow_request_sock_ipv4_ops,
378 sk, skb);
379 drop:
380 tcp_listendrop(sk);
381 return 0;
382 }
383
384 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
385 static struct tcp_request_sock_ops subflow_request_sock_ipv6_ops;
386 static struct inet_connection_sock_af_ops subflow_v6_specific;
387 static struct inet_connection_sock_af_ops subflow_v6m_specific;
388
subflow_v6_conn_request(struct sock * sk,struct sk_buff * skb)389 static int subflow_v6_conn_request(struct sock *sk, struct sk_buff *skb)
390 {
391 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
392
393 pr_debug("subflow=%p", subflow);
394
395 if (skb->protocol == htons(ETH_P_IP))
396 return subflow_v4_conn_request(sk, skb);
397
398 if (!ipv6_unicast_destination(skb))
399 goto drop;
400
401 if (ipv6_addr_v4mapped(&ipv6_hdr(skb)->saddr)) {
402 __IP6_INC_STATS(sock_net(sk), NULL, IPSTATS_MIB_INHDRERRORS);
403 return 0;
404 }
405
406 return tcp_conn_request(&mptcp_subflow_request_sock_ops,
407 &subflow_request_sock_ipv6_ops, sk, skb);
408
409 drop:
410 tcp_listendrop(sk);
411 return 0; /* don't send reset */
412 }
413 #endif
414
415 /* validate hmac received in third ACK */
subflow_hmac_valid(const struct request_sock * req,const struct mptcp_options_received * mp_opt)416 static bool subflow_hmac_valid(const struct request_sock *req,
417 const struct mptcp_options_received *mp_opt)
418 {
419 const struct mptcp_subflow_request_sock *subflow_req;
420 u8 hmac[SHA256_DIGEST_SIZE];
421 struct mptcp_sock *msk;
422
423 subflow_req = mptcp_subflow_rsk(req);
424 msk = subflow_req->msk;
425 if (!msk)
426 return false;
427
428 subflow_generate_hmac(msk->remote_key, msk->local_key,
429 subflow_req->remote_nonce,
430 subflow_req->local_nonce, hmac);
431
432 return !crypto_memneq(hmac, mp_opt->hmac, MPTCPOPT_HMAC_LEN);
433 }
434
mptcp_sock_destruct(struct sock * sk)435 static void mptcp_sock_destruct(struct sock *sk)
436 {
437 /* if new mptcp socket isn't accepted, it is free'd
438 * from the tcp listener sockets request queue, linked
439 * from req->sk. The tcp socket is released.
440 * This calls the ULP release function which will
441 * also remove the mptcp socket, via
442 * sock_put(ctx->conn).
443 *
444 * Problem is that the mptcp socket will be in
445 * ESTABLISHED state and will not have the SOCK_DEAD flag.
446 * Both result in warnings from inet_sock_destruct.
447 */
448 if ((1 << sk->sk_state) & (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)) {
449 sk->sk_state = TCP_CLOSE;
450 WARN_ON_ONCE(sk->sk_socket);
451 sock_orphan(sk);
452 }
453
454 mptcp_destroy_common(mptcp_sk(sk));
455 inet_sock_destruct(sk);
456 }
457
mptcp_force_close(struct sock * sk)458 static void mptcp_force_close(struct sock *sk)
459 {
460 inet_sk_state_store(sk, TCP_CLOSE);
461 sk_common_release(sk);
462 }
463
subflow_ulp_fallback(struct sock * sk,struct mptcp_subflow_context * old_ctx)464 static void subflow_ulp_fallback(struct sock *sk,
465 struct mptcp_subflow_context *old_ctx)
466 {
467 struct inet_connection_sock *icsk = inet_csk(sk);
468
469 mptcp_subflow_tcp_fallback(sk, old_ctx);
470 icsk->icsk_ulp_ops = NULL;
471 rcu_assign_pointer(icsk->icsk_ulp_data, NULL);
472 tcp_sk(sk)->is_mptcp = 0;
473 }
474
subflow_drop_ctx(struct sock * ssk)475 static void subflow_drop_ctx(struct sock *ssk)
476 {
477 struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(ssk);
478
479 if (!ctx)
480 return;
481
482 subflow_ulp_fallback(ssk, ctx);
483 if (ctx->conn)
484 sock_put(ctx->conn);
485
486 kfree_rcu(ctx, rcu);
487 }
488
mptcp_subflow_fully_established(struct mptcp_subflow_context * subflow,struct mptcp_options_received * mp_opt)489 void mptcp_subflow_fully_established(struct mptcp_subflow_context *subflow,
490 struct mptcp_options_received *mp_opt)
491 {
492 struct mptcp_sock *msk = mptcp_sk(subflow->conn);
493
494 subflow->remote_key = mp_opt->sndr_key;
495 subflow->fully_established = 1;
496 subflow->can_ack = 1;
497 WRITE_ONCE(msk->fully_established, true);
498 }
499
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)500 static struct sock *subflow_syn_recv_sock(const struct sock *sk,
501 struct sk_buff *skb,
502 struct request_sock *req,
503 struct dst_entry *dst,
504 struct request_sock *req_unhash,
505 bool *own_req)
506 {
507 struct mptcp_subflow_context *listener = mptcp_subflow_ctx(sk);
508 struct mptcp_subflow_request_sock *subflow_req;
509 struct mptcp_options_received mp_opt;
510 bool fallback, fallback_is_fatal;
511 struct sock *new_msk = NULL;
512 struct sock *child;
513
514 pr_debug("listener=%p, req=%p, conn=%p", listener, req, listener->conn);
515
516 /* After child creation we must look for 'mp_capable' even when options
517 * are not parsed
518 */
519 mp_opt.mp_capable = 0;
520
521 /* hopefully temporary handling for MP_JOIN+syncookie */
522 subflow_req = mptcp_subflow_rsk(req);
523 fallback_is_fatal = tcp_rsk(req)->is_mptcp && subflow_req->mp_join;
524 fallback = !tcp_rsk(req)->is_mptcp;
525 if (fallback)
526 goto create_child;
527
528 /* if the sk is MP_CAPABLE, we try to fetch the client key */
529 if (subflow_req->mp_capable) {
530 /* we can receive and accept an in-window, out-of-order pkt,
531 * which may not carry the MP_CAPABLE opt even on mptcp enabled
532 * paths: always try to extract the peer key, and fallback
533 * for packets missing it.
534 * Even OoO DSS packets coming legitly after dropped or
535 * reordered MPC will cause fallback, but we don't have other
536 * options.
537 */
538 mptcp_get_options(skb, &mp_opt);
539 if (!mp_opt.mp_capable) {
540 fallback = true;
541 goto create_child;
542 }
543
544 new_msk = mptcp_sk_clone(listener->conn, &mp_opt, req);
545 if (!new_msk)
546 fallback = true;
547 } else if (subflow_req->mp_join) {
548 mptcp_get_options(skb, &mp_opt);
549 if (!mp_opt.mp_join || !subflow_hmac_valid(req, &mp_opt) ||
550 !mptcp_can_accept_new_subflow(subflow_req->msk)) {
551 SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINACKMAC);
552 fallback = true;
553 }
554 }
555
556 create_child:
557 child = listener->icsk_af_ops->syn_recv_sock(sk, skb, req, dst,
558 req_unhash, own_req);
559
560 if (child && *own_req) {
561 struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(child);
562
563 tcp_rsk(req)->drop_req = false;
564
565 /* we need to fallback on ctx allocation failure and on pre-reqs
566 * checking above. In the latter scenario we additionally need
567 * to reset the context to non MPTCP status.
568 */
569 if (!ctx || fallback) {
570 if (fallback_is_fatal)
571 goto dispose_child;
572
573 subflow_drop_ctx(child);
574 goto out;
575 }
576
577 if (ctx->mp_capable) {
578 /* this can't race with mptcp_close(), as the msk is
579 * not yet exposted to user-space
580 */
581 inet_sk_state_store((void *)new_msk, TCP_ESTABLISHED);
582
583 /* new mpc subflow takes ownership of the newly
584 * created mptcp socket
585 */
586 new_msk->sk_destruct = mptcp_sock_destruct;
587 mptcp_pm_new_connection(mptcp_sk(new_msk), 1);
588 mptcp_token_accept(subflow_req, mptcp_sk(new_msk));
589 ctx->conn = new_msk;
590 new_msk = NULL;
591
592 /* with OoO packets we can reach here without ingress
593 * mpc option
594 */
595 if (mp_opt.mp_capable)
596 mptcp_subflow_fully_established(ctx, &mp_opt);
597 } else if (ctx->mp_join) {
598 struct mptcp_sock *owner;
599
600 owner = subflow_req->msk;
601 if (!owner)
602 goto dispose_child;
603
604 /* move the msk reference ownership to the subflow */
605 subflow_req->msk = NULL;
606 ctx->conn = (struct sock *)owner;
607 if (!mptcp_finish_join(child))
608 goto dispose_child;
609
610 SUBFLOW_REQ_INC_STATS(req, MPTCP_MIB_JOINACKRX);
611 tcp_rsk(req)->drop_req = true;
612 }
613 }
614
615 out:
616 /* dispose of the left over mptcp master, if any */
617 if (unlikely(new_msk))
618 mptcp_force_close(new_msk);
619
620 /* check for expected invariant - should never trigger, just help
621 * catching eariler subtle bugs
622 */
623 WARN_ON_ONCE(child && *own_req && tcp_sk(child)->is_mptcp &&
624 (!mptcp_subflow_ctx(child) ||
625 !mptcp_subflow_ctx(child)->conn));
626 return child;
627
628 dispose_child:
629 subflow_drop_ctx(child);
630 tcp_rsk(req)->drop_req = true;
631 inet_csk_prepare_for_destroy_sock(child);
632 tcp_done(child);
633 req->rsk_ops->send_reset(sk, skb);
634
635 /* The last child reference will be released by the caller */
636 return child;
637 }
638
639 static struct inet_connection_sock_af_ops subflow_specific;
640
641 enum mapping_status {
642 MAPPING_OK,
643 MAPPING_INVALID,
644 MAPPING_EMPTY,
645 MAPPING_DATA_FIN,
646 MAPPING_DUMMY
647 };
648
expand_seq(u64 old_seq,u16 old_data_len,u64 seq)649 static u64 expand_seq(u64 old_seq, u16 old_data_len, u64 seq)
650 {
651 if ((u32)seq == (u32)old_seq)
652 return old_seq;
653
654 /* Assume map covers data not mapped yet. */
655 return seq | ((old_seq + old_data_len + 1) & GENMASK_ULL(63, 32));
656 }
657
dbg_bad_map(struct mptcp_subflow_context * subflow,u32 ssn)658 static void dbg_bad_map(struct mptcp_subflow_context *subflow, u32 ssn)
659 {
660 pr_debug("Bad mapping: ssn=%d map_seq=%d map_data_len=%d",
661 ssn, subflow->map_subflow_seq, subflow->map_data_len);
662 }
663
skb_is_fully_mapped(struct sock * ssk,struct sk_buff * skb)664 static bool skb_is_fully_mapped(struct sock *ssk, struct sk_buff *skb)
665 {
666 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
667 unsigned int skb_consumed;
668
669 skb_consumed = tcp_sk(ssk)->copied_seq - TCP_SKB_CB(skb)->seq;
670 if (WARN_ON_ONCE(skb_consumed >= skb->len))
671 return true;
672
673 return skb->len - skb_consumed <= subflow->map_data_len -
674 mptcp_subflow_get_map_offset(subflow);
675 }
676
validate_mapping(struct sock * ssk,struct sk_buff * skb)677 static bool validate_mapping(struct sock *ssk, struct sk_buff *skb)
678 {
679 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
680 u32 ssn = tcp_sk(ssk)->copied_seq - subflow->ssn_offset;
681
682 if (unlikely(before(ssn, subflow->map_subflow_seq))) {
683 /* Mapping covers data later in the subflow stream,
684 * currently unsupported.
685 */
686 dbg_bad_map(subflow, ssn);
687 return false;
688 }
689 if (unlikely(!before(ssn, subflow->map_subflow_seq +
690 subflow->map_data_len))) {
691 /* Mapping does covers past subflow data, invalid */
692 dbg_bad_map(subflow, ssn);
693 return false;
694 }
695 return true;
696 }
697
get_mapping_status(struct sock * ssk,struct mptcp_sock * msk)698 static enum mapping_status get_mapping_status(struct sock *ssk,
699 struct mptcp_sock *msk)
700 {
701 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
702 struct mptcp_ext *mpext;
703 struct sk_buff *skb;
704 u16 data_len;
705 u64 map_seq;
706
707 skb = skb_peek(&ssk->sk_receive_queue);
708 if (!skb)
709 return MAPPING_EMPTY;
710
711 if (mptcp_check_fallback(ssk))
712 return MAPPING_DUMMY;
713
714 mpext = mptcp_get_ext(skb);
715 if (!mpext || !mpext->use_map) {
716 if (!subflow->map_valid && !skb->len) {
717 /* the TCP stack deliver 0 len FIN pkt to the receive
718 * queue, that is the only 0len pkts ever expected here,
719 * and we can admit no mapping only for 0 len pkts
720 */
721 if (!(TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN))
722 WARN_ONCE(1, "0len seq %d:%d flags %x",
723 TCP_SKB_CB(skb)->seq,
724 TCP_SKB_CB(skb)->end_seq,
725 TCP_SKB_CB(skb)->tcp_flags);
726 sk_eat_skb(ssk, skb);
727 return MAPPING_EMPTY;
728 }
729
730 if (!subflow->map_valid)
731 return MAPPING_INVALID;
732
733 goto validate_seq;
734 }
735
736 pr_debug("seq=%llu is64=%d ssn=%u data_len=%u data_fin=%d",
737 mpext->data_seq, mpext->dsn64, mpext->subflow_seq,
738 mpext->data_len, mpext->data_fin);
739
740 data_len = mpext->data_len;
741 if (data_len == 0) {
742 MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_INFINITEMAPRX);
743 return MAPPING_INVALID;
744 }
745
746 if (mpext->data_fin == 1) {
747 if (data_len == 1) {
748 bool updated = mptcp_update_rcv_data_fin(msk, mpext->data_seq,
749 mpext->dsn64);
750 pr_debug("DATA_FIN with no payload seq=%llu", mpext->data_seq);
751 if (subflow->map_valid) {
752 /* A DATA_FIN might arrive in a DSS
753 * option before the previous mapping
754 * has been fully consumed. Continue
755 * handling the existing mapping.
756 */
757 skb_ext_del(skb, SKB_EXT_MPTCP);
758 return MAPPING_OK;
759 } else {
760 if (updated && schedule_work(&msk->work))
761 sock_hold((struct sock *)msk);
762
763 return MAPPING_DATA_FIN;
764 }
765 } else {
766 u64 data_fin_seq = mpext->data_seq + data_len - 1;
767
768 /* If mpext->data_seq is a 32-bit value, data_fin_seq
769 * must also be limited to 32 bits.
770 */
771 if (!mpext->dsn64)
772 data_fin_seq &= GENMASK_ULL(31, 0);
773
774 mptcp_update_rcv_data_fin(msk, data_fin_seq, mpext->dsn64);
775 pr_debug("DATA_FIN with mapping seq=%llu dsn64=%d",
776 data_fin_seq, mpext->dsn64);
777 }
778
779 /* Adjust for DATA_FIN using 1 byte of sequence space */
780 data_len--;
781 }
782
783 if (!mpext->dsn64) {
784 map_seq = expand_seq(subflow->map_seq, subflow->map_data_len,
785 mpext->data_seq);
786 pr_debug("expanded seq=%llu", subflow->map_seq);
787 } else {
788 map_seq = mpext->data_seq;
789 }
790 WRITE_ONCE(mptcp_sk(subflow->conn)->use_64bit_ack, !!mpext->dsn64);
791
792 if (subflow->map_valid) {
793 /* Allow replacing only with an identical map */
794 if (subflow->map_seq == map_seq &&
795 subflow->map_subflow_seq == mpext->subflow_seq &&
796 subflow->map_data_len == data_len) {
797 skb_ext_del(skb, SKB_EXT_MPTCP);
798 return MAPPING_OK;
799 }
800
801 /* If this skb data are fully covered by the current mapping,
802 * the new map would need caching, which is not supported
803 */
804 if (skb_is_fully_mapped(ssk, skb)) {
805 MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DSSNOMATCH);
806 return MAPPING_INVALID;
807 }
808
809 /* will validate the next map after consuming the current one */
810 return MAPPING_OK;
811 }
812
813 subflow->map_seq = map_seq;
814 subflow->map_subflow_seq = mpext->subflow_seq;
815 subflow->map_data_len = data_len;
816 subflow->map_valid = 1;
817 subflow->mpc_map = mpext->mpc_map;
818 pr_debug("new map seq=%llu subflow_seq=%u data_len=%u",
819 subflow->map_seq, subflow->map_subflow_seq,
820 subflow->map_data_len);
821
822 validate_seq:
823 /* we revalidate valid mapping on new skb, because we must ensure
824 * the current skb is completely covered by the available mapping
825 */
826 if (!validate_mapping(ssk, skb))
827 return MAPPING_INVALID;
828
829 skb_ext_del(skb, SKB_EXT_MPTCP);
830 return MAPPING_OK;
831 }
832
mptcp_subflow_discard_data(struct sock * ssk,struct sk_buff * skb,u64 limit)833 static void mptcp_subflow_discard_data(struct sock *ssk, struct sk_buff *skb,
834 u64 limit)
835 {
836 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
837 bool fin = TCP_SKB_CB(skb)->tcp_flags & TCPHDR_FIN;
838 u32 incr;
839
840 incr = limit >= skb->len ? skb->len + fin : limit;
841
842 pr_debug("discarding=%d len=%d seq=%d", incr, skb->len,
843 subflow->map_subflow_seq);
844 MPTCP_INC_STATS(sock_net(ssk), MPTCP_MIB_DUPDATA);
845 tcp_sk(ssk)->copied_seq += incr;
846 if (!before(tcp_sk(ssk)->copied_seq, TCP_SKB_CB(skb)->end_seq))
847 sk_eat_skb(ssk, skb);
848 if (mptcp_subflow_get_map_offset(subflow) >= subflow->map_data_len)
849 subflow->map_valid = 0;
850 if (incr)
851 tcp_cleanup_rbuf(ssk, incr);
852 }
853
subflow_check_data_avail(struct sock * ssk)854 static bool subflow_check_data_avail(struct sock *ssk)
855 {
856 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
857 enum mapping_status status;
858 struct mptcp_sock *msk;
859 struct sk_buff *skb;
860
861 pr_debug("msk=%p ssk=%p data_avail=%d skb=%p", subflow->conn, ssk,
862 subflow->data_avail, skb_peek(&ssk->sk_receive_queue));
863 if (!skb_peek(&ssk->sk_receive_queue))
864 subflow->data_avail = 0;
865 if (subflow->data_avail)
866 return true;
867
868 msk = mptcp_sk(subflow->conn);
869 for (;;) {
870 u64 ack_seq;
871 u64 old_ack;
872
873 status = get_mapping_status(ssk, msk);
874 pr_debug("msk=%p ssk=%p status=%d", msk, ssk, status);
875 if (status == MAPPING_INVALID) {
876 ssk->sk_err = EBADMSG;
877 goto fatal;
878 }
879 if (status == MAPPING_DUMMY) {
880 __mptcp_do_fallback(msk);
881 skb = skb_peek(&ssk->sk_receive_queue);
882 subflow->map_valid = 1;
883 subflow->map_seq = READ_ONCE(msk->ack_seq);
884 subflow->map_data_len = skb->len;
885 subflow->map_subflow_seq = tcp_sk(ssk)->copied_seq -
886 subflow->ssn_offset;
887 subflow->data_avail = MPTCP_SUBFLOW_DATA_AVAIL;
888 return true;
889 }
890
891 if (status != MAPPING_OK)
892 return false;
893
894 skb = skb_peek(&ssk->sk_receive_queue);
895 if (WARN_ON_ONCE(!skb))
896 return false;
897
898 /* if msk lacks the remote key, this subflow must provide an
899 * MP_CAPABLE-based mapping
900 */
901 if (unlikely(!READ_ONCE(msk->can_ack))) {
902 if (!subflow->mpc_map) {
903 ssk->sk_err = EBADMSG;
904 goto fatal;
905 }
906 WRITE_ONCE(msk->remote_key, subflow->remote_key);
907 WRITE_ONCE(msk->ack_seq, subflow->map_seq);
908 WRITE_ONCE(msk->can_ack, true);
909 }
910
911 old_ack = READ_ONCE(msk->ack_seq);
912 ack_seq = mptcp_subflow_get_mapped_dsn(subflow);
913 pr_debug("msk ack_seq=%llx subflow ack_seq=%llx", old_ack,
914 ack_seq);
915 if (ack_seq == old_ack) {
916 subflow->data_avail = MPTCP_SUBFLOW_DATA_AVAIL;
917 break;
918 } else if (after64(ack_seq, old_ack)) {
919 subflow->data_avail = MPTCP_SUBFLOW_OOO_DATA;
920 break;
921 }
922
923 /* only accept in-sequence mapping. Old values are spurious
924 * retransmission
925 */
926 mptcp_subflow_discard_data(ssk, skb, old_ack - ack_seq);
927 }
928 return true;
929
930 fatal:
931 /* fatal protocol error, close the socket */
932 /* This barrier is coupled with smp_rmb() in tcp_poll() */
933 smp_wmb();
934 ssk->sk_error_report(ssk);
935 tcp_set_state(ssk, TCP_CLOSE);
936 tcp_send_active_reset(ssk, GFP_ATOMIC);
937 subflow->data_avail = 0;
938 return false;
939 }
940
mptcp_subflow_data_available(struct sock * sk)941 bool mptcp_subflow_data_available(struct sock *sk)
942 {
943 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
944
945 /* check if current mapping is still valid */
946 if (subflow->map_valid &&
947 mptcp_subflow_get_map_offset(subflow) >= subflow->map_data_len) {
948 subflow->map_valid = 0;
949 subflow->data_avail = 0;
950
951 pr_debug("Done with mapping: seq=%u data_len=%u",
952 subflow->map_subflow_seq,
953 subflow->map_data_len);
954 }
955
956 return subflow_check_data_avail(sk);
957 }
958
959 /* If ssk has an mptcp parent socket, use the mptcp rcvbuf occupancy,
960 * not the ssk one.
961 *
962 * In mptcp, rwin is about the mptcp-level connection data.
963 *
964 * Data that is still on the ssk rx queue can thus be ignored,
965 * as far as mptcp peer is concerened that data is still inflight.
966 * DSS ACK is updated when skb is moved to the mptcp rx queue.
967 */
mptcp_space(const struct sock * ssk,int * space,int * full_space)968 void mptcp_space(const struct sock *ssk, int *space, int *full_space)
969 {
970 const struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(ssk);
971 const struct sock *sk = subflow->conn;
972
973 *space = tcp_space(sk);
974 *full_space = tcp_full_space(sk);
975 }
976
subflow_data_ready(struct sock * sk)977 static void subflow_data_ready(struct sock *sk)
978 {
979 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
980 u16 state = 1 << inet_sk_state_load(sk);
981 struct sock *parent = subflow->conn;
982 struct mptcp_sock *msk;
983
984 msk = mptcp_sk(parent);
985 if (state & TCPF_LISTEN) {
986 /* MPJ subflow are removed from accept queue before reaching here,
987 * avoid stray wakeups
988 */
989 if (reqsk_queue_empty(&inet_csk(sk)->icsk_accept_queue))
990 return;
991
992 set_bit(MPTCP_DATA_READY, &msk->flags);
993 parent->sk_data_ready(parent);
994 return;
995 }
996
997 WARN_ON_ONCE(!__mptcp_check_fallback(msk) && !subflow->mp_capable &&
998 !subflow->mp_join && !(state & TCPF_CLOSE));
999
1000 if (mptcp_subflow_data_available(sk))
1001 mptcp_data_ready(parent, sk);
1002 }
1003
subflow_write_space(struct sock * sk)1004 static void subflow_write_space(struct sock *sk)
1005 {
1006 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1007 struct sock *parent = subflow->conn;
1008
1009 if (!sk_stream_is_writeable(sk))
1010 return;
1011
1012 if (sk_stream_is_writeable(parent)) {
1013 set_bit(MPTCP_SEND_SPACE, &mptcp_sk(parent)->flags);
1014 smp_mb__after_atomic();
1015 /* set SEND_SPACE before sk_stream_write_space clears NOSPACE */
1016 sk_stream_write_space(parent);
1017 }
1018 }
1019
1020 static struct inet_connection_sock_af_ops *
subflow_default_af_ops(struct sock * sk)1021 subflow_default_af_ops(struct sock *sk)
1022 {
1023 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1024 if (sk->sk_family == AF_INET6)
1025 return &subflow_v6_specific;
1026 #endif
1027 return &subflow_specific;
1028 }
1029
1030 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
mptcpv6_handle_mapped(struct sock * sk,bool mapped)1031 void mptcpv6_handle_mapped(struct sock *sk, bool mapped)
1032 {
1033 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1034 struct inet_connection_sock *icsk = inet_csk(sk);
1035 struct inet_connection_sock_af_ops *target;
1036
1037 target = mapped ? &subflow_v6m_specific : subflow_default_af_ops(sk);
1038
1039 pr_debug("subflow=%p family=%d ops=%p target=%p mapped=%d",
1040 subflow, sk->sk_family, icsk->icsk_af_ops, target, mapped);
1041
1042 if (likely(icsk->icsk_af_ops == target))
1043 return;
1044
1045 subflow->icsk_af_ops = icsk->icsk_af_ops;
1046 icsk->icsk_af_ops = target;
1047 }
1048 #endif
1049
mptcp_info2sockaddr(const struct mptcp_addr_info * info,struct sockaddr_storage * addr)1050 static void mptcp_info2sockaddr(const struct mptcp_addr_info *info,
1051 struct sockaddr_storage *addr)
1052 {
1053 memset(addr, 0, sizeof(*addr));
1054 addr->ss_family = info->family;
1055 if (addr->ss_family == AF_INET) {
1056 struct sockaddr_in *in_addr = (struct sockaddr_in *)addr;
1057
1058 in_addr->sin_addr = info->addr;
1059 in_addr->sin_port = info->port;
1060 }
1061 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1062 else if (addr->ss_family == AF_INET6) {
1063 struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)addr;
1064
1065 in6_addr->sin6_addr = info->addr6;
1066 in6_addr->sin6_port = info->port;
1067 }
1068 #endif
1069 }
1070
__mptcp_subflow_connect(struct sock * sk,const struct mptcp_addr_info * loc,const struct mptcp_addr_info * remote)1071 int __mptcp_subflow_connect(struct sock *sk, const struct mptcp_addr_info *loc,
1072 const struct mptcp_addr_info *remote)
1073 {
1074 struct mptcp_sock *msk = mptcp_sk(sk);
1075 struct mptcp_subflow_context *subflow;
1076 struct sockaddr_storage addr;
1077 int remote_id = remote->id;
1078 int local_id = loc->id;
1079 struct socket *sf;
1080 struct sock *ssk;
1081 u32 remote_token;
1082 int addrlen;
1083 int err;
1084
1085 if (!mptcp_is_fully_established(sk))
1086 return -ENOTCONN;
1087
1088 err = mptcp_subflow_create_socket(sk, &sf);
1089 if (err)
1090 return err;
1091
1092 ssk = sf->sk;
1093 subflow = mptcp_subflow_ctx(ssk);
1094 do {
1095 get_random_bytes(&subflow->local_nonce, sizeof(u32));
1096 } while (!subflow->local_nonce);
1097
1098 if (!local_id) {
1099 err = mptcp_pm_get_local_id(msk, (struct sock_common *)ssk);
1100 if (err < 0)
1101 goto failed;
1102
1103 local_id = err;
1104 }
1105
1106 subflow->remote_key = msk->remote_key;
1107 subflow->local_key = msk->local_key;
1108 subflow->token = msk->token;
1109 mptcp_info2sockaddr(loc, &addr);
1110
1111 addrlen = sizeof(struct sockaddr_in);
1112 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1113 if (loc->family == AF_INET6)
1114 addrlen = sizeof(struct sockaddr_in6);
1115 #endif
1116 ssk->sk_bound_dev_if = loc->ifindex;
1117 err = kernel_bind(sf, (struct sockaddr *)&addr, addrlen);
1118 if (err)
1119 goto failed;
1120
1121 mptcp_crypto_key_sha(subflow->remote_key, &remote_token, NULL);
1122 pr_debug("msk=%p remote_token=%u local_id=%d remote_id=%d", msk,
1123 remote_token, local_id, remote_id);
1124 subflow->remote_token = remote_token;
1125 subflow->local_id = local_id;
1126 subflow->remote_id = remote_id;
1127 subflow->request_join = 1;
1128 subflow->request_bkup = !!(loc->flags & MPTCP_PM_ADDR_FLAG_BACKUP);
1129 mptcp_info2sockaddr(remote, &addr);
1130
1131 err = kernel_connect(sf, (struct sockaddr *)&addr, addrlen, O_NONBLOCK);
1132 if (err && err != -EINPROGRESS)
1133 goto failed;
1134
1135 spin_lock_bh(&msk->join_list_lock);
1136 list_add_tail(&subflow->node, &msk->join_list);
1137 spin_unlock_bh(&msk->join_list_lock);
1138
1139 return err;
1140
1141 failed:
1142 sock_release(sf);
1143 return err;
1144 }
1145
mptcp_subflow_create_socket(struct sock * sk,struct socket ** new_sock)1146 int mptcp_subflow_create_socket(struct sock *sk, struct socket **new_sock)
1147 {
1148 struct mptcp_subflow_context *subflow;
1149 struct net *net = sock_net(sk);
1150 struct socket *sf;
1151 int err;
1152
1153 /* un-accepted server sockets can reach here - on bad configuration
1154 * bail early to avoid greater trouble later
1155 */
1156 if (unlikely(!sk->sk_socket))
1157 return -EINVAL;
1158
1159 err = sock_create_kern(net, sk->sk_family, SOCK_STREAM, IPPROTO_TCP,
1160 &sf);
1161 if (err)
1162 return err;
1163
1164 lock_sock(sf->sk);
1165
1166 /* kernel sockets do not by default acquire net ref, but TCP timer
1167 * needs it.
1168 */
1169 sf->sk->sk_net_refcnt = 1;
1170 get_net(net);
1171 #ifdef CONFIG_PROC_FS
1172 this_cpu_add(*net->core.sock_inuse, 1);
1173 #endif
1174 err = tcp_set_ulp(sf->sk, "mptcp");
1175 release_sock(sf->sk);
1176
1177 if (err) {
1178 sock_release(sf);
1179 return err;
1180 }
1181
1182 /* the newly created socket really belongs to the owning MPTCP master
1183 * socket, even if for additional subflows the allocation is performed
1184 * by a kernel workqueue. Adjust inode references, so that the
1185 * procfs/diag interaces really show this one belonging to the correct
1186 * user.
1187 */
1188 SOCK_INODE(sf)->i_ino = SOCK_INODE(sk->sk_socket)->i_ino;
1189 SOCK_INODE(sf)->i_uid = SOCK_INODE(sk->sk_socket)->i_uid;
1190 SOCK_INODE(sf)->i_gid = SOCK_INODE(sk->sk_socket)->i_gid;
1191
1192 subflow = mptcp_subflow_ctx(sf->sk);
1193 pr_debug("subflow=%p", subflow);
1194
1195 *new_sock = sf;
1196 sock_hold(sk);
1197 subflow->conn = sk;
1198
1199 return 0;
1200 }
1201
subflow_create_ctx(struct sock * sk,gfp_t priority)1202 static struct mptcp_subflow_context *subflow_create_ctx(struct sock *sk,
1203 gfp_t priority)
1204 {
1205 struct inet_connection_sock *icsk = inet_csk(sk);
1206 struct mptcp_subflow_context *ctx;
1207
1208 ctx = kzalloc(sizeof(*ctx), priority);
1209 if (!ctx)
1210 return NULL;
1211
1212 rcu_assign_pointer(icsk->icsk_ulp_data, ctx);
1213 INIT_LIST_HEAD(&ctx->node);
1214
1215 pr_debug("subflow=%p", ctx);
1216
1217 ctx->tcp_sock = sk;
1218
1219 return ctx;
1220 }
1221
__subflow_state_change(struct sock * sk)1222 static void __subflow_state_change(struct sock *sk)
1223 {
1224 struct socket_wq *wq;
1225
1226 rcu_read_lock();
1227 wq = rcu_dereference(sk->sk_wq);
1228 if (skwq_has_sleeper(wq))
1229 wake_up_interruptible_all(&wq->wait);
1230 rcu_read_unlock();
1231 }
1232
subflow_is_done(const struct sock * sk)1233 static bool subflow_is_done(const struct sock *sk)
1234 {
1235 return sk->sk_shutdown & RCV_SHUTDOWN || sk->sk_state == TCP_CLOSE;
1236 }
1237
subflow_state_change(struct sock * sk)1238 static void subflow_state_change(struct sock *sk)
1239 {
1240 struct mptcp_subflow_context *subflow = mptcp_subflow_ctx(sk);
1241 struct sock *parent = subflow->conn;
1242
1243 __subflow_state_change(sk);
1244
1245 if (subflow_simultaneous_connect(sk)) {
1246 mptcp_do_fallback(sk);
1247 mptcp_rcv_space_init(mptcp_sk(parent), sk);
1248 pr_fallback(mptcp_sk(parent));
1249 subflow->conn_finished = 1;
1250 if (inet_sk_state_load(parent) == TCP_SYN_SENT) {
1251 inet_sk_state_store(parent, TCP_ESTABLISHED);
1252 parent->sk_state_change(parent);
1253 }
1254 }
1255
1256 /* as recvmsg() does not acquire the subflow socket for ssk selection
1257 * a fin packet carrying a DSS can be unnoticed if we don't trigger
1258 * the data available machinery here.
1259 */
1260 if (mptcp_subflow_data_available(sk))
1261 mptcp_data_ready(parent, sk);
1262
1263 if (__mptcp_check_fallback(mptcp_sk(parent)) &&
1264 !(parent->sk_shutdown & RCV_SHUTDOWN) &&
1265 !subflow->rx_eof && subflow_is_done(sk)) {
1266 subflow->rx_eof = 1;
1267 mptcp_subflow_eof(parent);
1268 }
1269 }
1270
subflow_ulp_init(struct sock * sk)1271 static int subflow_ulp_init(struct sock *sk)
1272 {
1273 struct inet_connection_sock *icsk = inet_csk(sk);
1274 struct mptcp_subflow_context *ctx;
1275 struct tcp_sock *tp = tcp_sk(sk);
1276 int err = 0;
1277
1278 /* disallow attaching ULP to a socket unless it has been
1279 * created with sock_create_kern()
1280 */
1281 if (!sk->sk_kern_sock) {
1282 err = -EOPNOTSUPP;
1283 goto out;
1284 }
1285
1286 ctx = subflow_create_ctx(sk, GFP_KERNEL);
1287 if (!ctx) {
1288 err = -ENOMEM;
1289 goto out;
1290 }
1291
1292 pr_debug("subflow=%p, family=%d", ctx, sk->sk_family);
1293
1294 tp->is_mptcp = 1;
1295 ctx->icsk_af_ops = icsk->icsk_af_ops;
1296 icsk->icsk_af_ops = subflow_default_af_ops(sk);
1297 ctx->tcp_data_ready = sk->sk_data_ready;
1298 ctx->tcp_state_change = sk->sk_state_change;
1299 ctx->tcp_write_space = sk->sk_write_space;
1300 sk->sk_data_ready = subflow_data_ready;
1301 sk->sk_write_space = subflow_write_space;
1302 sk->sk_state_change = subflow_state_change;
1303 out:
1304 return err;
1305 }
1306
subflow_ulp_release(struct sock * sk)1307 static void subflow_ulp_release(struct sock *sk)
1308 {
1309 struct mptcp_subflow_context *ctx = mptcp_subflow_ctx(sk);
1310
1311 if (!ctx)
1312 return;
1313
1314 if (ctx->conn)
1315 sock_put(ctx->conn);
1316
1317 kfree_rcu(ctx, rcu);
1318 }
1319
subflow_ulp_clone(const struct request_sock * req,struct sock * newsk,const gfp_t priority)1320 static void subflow_ulp_clone(const struct request_sock *req,
1321 struct sock *newsk,
1322 const gfp_t priority)
1323 {
1324 struct mptcp_subflow_request_sock *subflow_req = mptcp_subflow_rsk(req);
1325 struct mptcp_subflow_context *old_ctx = mptcp_subflow_ctx(newsk);
1326 struct mptcp_subflow_context *new_ctx;
1327
1328 if (!tcp_rsk(req)->is_mptcp ||
1329 (!subflow_req->mp_capable && !subflow_req->mp_join)) {
1330 subflow_ulp_fallback(newsk, old_ctx);
1331 return;
1332 }
1333
1334 new_ctx = subflow_create_ctx(newsk, priority);
1335 if (!new_ctx) {
1336 subflow_ulp_fallback(newsk, old_ctx);
1337 return;
1338 }
1339
1340 new_ctx->conn_finished = 1;
1341 new_ctx->icsk_af_ops = old_ctx->icsk_af_ops;
1342 new_ctx->tcp_data_ready = old_ctx->tcp_data_ready;
1343 new_ctx->tcp_state_change = old_ctx->tcp_state_change;
1344 new_ctx->tcp_write_space = old_ctx->tcp_write_space;
1345 new_ctx->rel_write_seq = 1;
1346 new_ctx->tcp_sock = newsk;
1347
1348 if (subflow_req->mp_capable) {
1349 /* see comments in subflow_syn_recv_sock(), MPTCP connection
1350 * is fully established only after we receive the remote key
1351 */
1352 new_ctx->mp_capable = 1;
1353 new_ctx->local_key = subflow_req->local_key;
1354 new_ctx->token = subflow_req->token;
1355 new_ctx->ssn_offset = subflow_req->ssn_offset;
1356 new_ctx->idsn = subflow_req->idsn;
1357 } else if (subflow_req->mp_join) {
1358 new_ctx->ssn_offset = subflow_req->ssn_offset;
1359 new_ctx->mp_join = 1;
1360 new_ctx->fully_established = 1;
1361 new_ctx->backup = subflow_req->backup;
1362 new_ctx->local_id = subflow_req->local_id;
1363 new_ctx->remote_id = subflow_req->remote_id;
1364 new_ctx->token = subflow_req->token;
1365 new_ctx->thmac = subflow_req->thmac;
1366 }
1367 }
1368
1369 static struct tcp_ulp_ops subflow_ulp_ops __read_mostly = {
1370 .name = "mptcp",
1371 .owner = THIS_MODULE,
1372 .init = subflow_ulp_init,
1373 .release = subflow_ulp_release,
1374 .clone = subflow_ulp_clone,
1375 };
1376
subflow_ops_init(struct request_sock_ops * subflow_ops)1377 static int subflow_ops_init(struct request_sock_ops *subflow_ops)
1378 {
1379 subflow_ops->obj_size = sizeof(struct mptcp_subflow_request_sock);
1380 subflow_ops->slab_name = "request_sock_subflow";
1381
1382 subflow_ops->slab = kmem_cache_create(subflow_ops->slab_name,
1383 subflow_ops->obj_size, 0,
1384 SLAB_ACCOUNT |
1385 SLAB_TYPESAFE_BY_RCU,
1386 NULL);
1387 if (!subflow_ops->slab)
1388 return -ENOMEM;
1389
1390 subflow_ops->destructor = subflow_req_destructor;
1391
1392 return 0;
1393 }
1394
mptcp_subflow_init(void)1395 void __init mptcp_subflow_init(void)
1396 {
1397 mptcp_subflow_request_sock_ops = tcp_request_sock_ops;
1398 if (subflow_ops_init(&mptcp_subflow_request_sock_ops) != 0)
1399 panic("MPTCP: failed to init subflow request sock ops\n");
1400
1401 subflow_request_sock_ipv4_ops = tcp_request_sock_ipv4_ops;
1402 subflow_request_sock_ipv4_ops.init_req = subflow_v4_init_req;
1403
1404 subflow_specific = ipv4_specific;
1405 subflow_specific.conn_request = subflow_v4_conn_request;
1406 subflow_specific.syn_recv_sock = subflow_syn_recv_sock;
1407 subflow_specific.sk_rx_dst_set = subflow_finish_connect;
1408
1409 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
1410 subflow_request_sock_ipv6_ops = tcp_request_sock_ipv6_ops;
1411 subflow_request_sock_ipv6_ops.init_req = subflow_v6_init_req;
1412
1413 subflow_v6_specific = ipv6_specific;
1414 subflow_v6_specific.conn_request = subflow_v6_conn_request;
1415 subflow_v6_specific.syn_recv_sock = subflow_syn_recv_sock;
1416 subflow_v6_specific.sk_rx_dst_set = subflow_finish_connect;
1417
1418 subflow_v6m_specific = subflow_v6_specific;
1419 subflow_v6m_specific.queue_xmit = ipv4_specific.queue_xmit;
1420 subflow_v6m_specific.send_check = ipv4_specific.send_check;
1421 subflow_v6m_specific.net_header_len = ipv4_specific.net_header_len;
1422 subflow_v6m_specific.mtu_reduced = ipv4_specific.mtu_reduced;
1423 subflow_v6m_specific.net_frag_header_len = 0;
1424 #endif
1425
1426 mptcp_diag_subflow_init(&subflow_ulp_ops);
1427
1428 if (tcp_register_ulp(&subflow_ulp_ops) != 0)
1429 panic("MPTCP: failed to register subflows to ULP\n");
1430 }
1431