Branch data Line data Source code
1 : : // SPDX-License-Identifier: GPL-2.0
2 : : /* Multipath TCP
3 : : *
4 : : * Copyright (c) 2022, Intel Corporation.
5 : : */
6 : :
7 : : #include "protocol.h"
8 : : #include "mib.h"
9 : : #include "mptcp_pm_gen.h"
10 : :
11 : : #define mptcp_for_each_userspace_pm_addr(__msk, __entry) \
12 : : list_for_each_entry(__entry, \
13 : : &((__msk)->pm.userspace_pm_local_addr_list), list)
14 : :
15 : 66 : void mptcp_userspace_pm_free_local_addr_list(struct mptcp_sock *msk)
16 : : {
17 : 66 : struct mptcp_pm_addr_entry *entry, *tmp;
18 : 66 : struct sock *sk = (struct sock *)msk;
19 : 66 : LIST_HEAD(free_list);
20 : :
21 : 66 : spin_lock_bh(&msk->pm.lock);
22 [ + + ]: 66 : list_splice_init(&msk->pm.userspace_pm_local_addr_list, &free_list);
23 : 66 : spin_unlock_bh(&msk->pm.lock);
24 : :
25 [ + + ]: 80 : list_for_each_entry_safe(entry, tmp, &free_list, list) {
26 : 14 : sock_kfree_s(sk, entry, sizeof(*entry));
27 : : }
28 : 66 : }
29 : :
30 : : static struct mptcp_pm_addr_entry *
31 : 72 : mptcp_userspace_pm_lookup_addr(struct mptcp_sock *msk,
32 : : const struct mptcp_addr_info *addr)
33 : : {
34 : 72 : struct mptcp_pm_addr_entry *entry;
35 : :
36 [ + + ]: 78 : mptcp_for_each_userspace_pm_addr(msk, entry) {
37 [ + + ]: 70 : if (mptcp_addresses_equal(&entry->addr, addr, false))
38 : : return entry;
39 : : }
40 : : return NULL;
41 : : }
42 : :
43 : 62 : static int mptcp_userspace_pm_append_new_local_addr(struct mptcp_sock *msk,
44 : : struct mptcp_pm_addr_entry *entry,
45 : : bool needs_id)
46 : : {
47 : 62 : DECLARE_BITMAP(id_bitmap, MPTCP_PM_MAX_ADDR_ID + 1);
48 : 62 : struct sock *sk = (struct sock *)msk;
49 : 62 : struct mptcp_pm_addr_entry *e;
50 : 62 : bool addr_match = false;
51 : 62 : bool id_match = false;
52 : 62 : int ret = -EINVAL;
53 : :
54 : 62 : bitmap_zero(id_bitmap, MPTCP_PM_MAX_ADDR_ID + 1);
55 : :
56 : 62 : spin_lock_bh(&msk->pm.lock);
57 [ + + ]: 70 : mptcp_for_each_userspace_pm_addr(msk, e) {
58 : 8 : addr_match = mptcp_addresses_equal(&e->addr, &entry->addr, true);
59 [ - + - - : 8 : if (addr_match && entry->addr.id == 0 && needs_id)
- - ]
60 : 0 : entry->addr.id = e->addr.id;
61 : 8 : id_match = (e->addr.id == entry->addr.id);
62 [ + - ]: 8 : if (addr_match || id_match)
63 : : break;
64 [ - + - - ]: 8 : __set_bit(e->addr.id, id_bitmap);
65 : : }
66 : :
67 [ + - ]: 62 : if (!addr_match && !id_match) {
68 : : /* Memory for the entry is allocated from the
69 : : * sock option buffer.
70 : : */
71 : 62 : e = sock_kmemdup(sk, entry, sizeof(*entry), GFP_ATOMIC);
72 [ - + ]: 62 : if (!e) {
73 : 0 : ret = -ENOMEM;
74 : 0 : goto append_err;
75 : : }
76 : :
77 [ + + - + ]: 62 : if (!e->addr.id && needs_id)
78 : 0 : e->addr.id = find_next_zero_bit(id_bitmap,
79 : : MPTCP_PM_MAX_ADDR_ID + 1,
80 : : 1);
81 : 62 : list_add_tail_rcu(&e->list, &msk->pm.userspace_pm_local_addr_list);
82 : 62 : msk->pm.local_addr_used++;
83 : 62 : ret = e->addr.id;
84 [ # # ]: 0 : } else if (addr_match && id_match) {
85 : 0 : ret = entry->addr.id;
86 : : }
87 : :
88 : 0 : append_err:
89 : 62 : spin_unlock_bh(&msk->pm.lock);
90 : 62 : return ret;
91 : : }
92 : :
93 : : /* If the subflow is closed from the other peer (not via a
94 : : * subflow destroy command then), we want to keep the entry
95 : : * not to assign the same ID to another address and to be
96 : : * able to send RM_ADDR after the removal of the subflow.
97 : : */
98 : 22 : static int mptcp_userspace_pm_delete_local_addr(struct mptcp_sock *msk,
99 : : struct mptcp_pm_addr_entry *addr)
100 : : {
101 : 22 : struct sock *sk = (struct sock *)msk;
102 : 22 : struct mptcp_pm_addr_entry *entry;
103 : :
104 : 22 : entry = mptcp_userspace_pm_lookup_addr(msk, &addr->addr);
105 [ + + ]: 22 : if (!entry)
106 : : return -EINVAL;
107 : :
108 : : /* TODO: a refcount is needed because the entry can
109 : : * be used multiple times (e.g. fullmesh mode).
110 : : */
111 : 20 : list_del_rcu(&entry->list);
112 : 20 : sock_kfree_s(sk, entry, sizeof(*entry));
113 : 20 : msk->pm.local_addr_used--;
114 : 20 : return 0;
115 : : }
116 : :
117 : : static struct mptcp_pm_addr_entry *
118 : 0 : mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id)
119 : : {
120 : 8 : struct mptcp_pm_addr_entry *entry;
121 : :
122 [ + - + + ]: 46 : mptcp_for_each_userspace_pm_addr(msk, entry) {
123 [ + + + + ]: 44 : if (entry->addr.id == id)
124 : : return entry;
125 : : }
126 : : return NULL;
127 : : }
128 : :
129 : 22 : int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
130 : : struct mptcp_pm_addr_entry *skc)
131 : : {
132 : 22 : __be16 msk_sport = ((struct inet_sock *)
133 : 22 : inet_sk((struct sock *)msk))->inet_sport;
134 : 22 : struct mptcp_pm_addr_entry *entry;
135 : :
136 : 22 : spin_lock_bh(&msk->pm.lock);
137 : 22 : entry = mptcp_userspace_pm_lookup_addr(msk, &skc->addr);
138 : 22 : spin_unlock_bh(&msk->pm.lock);
139 [ + - ]: 22 : if (entry)
140 : 22 : return entry->addr.id;
141 : :
142 [ # # ]: 0 : if (skc->addr.port == msk_sport)
143 : 0 : skc->addr.port = 0;
144 : :
145 : 0 : return mptcp_userspace_pm_append_new_local_addr(msk, skc, true);
146 : : }
147 : :
148 : 26 : bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk,
149 : : struct mptcp_addr_info *skc)
150 : : {
151 : 26 : struct mptcp_pm_addr_entry *entry;
152 : 26 : bool backup;
153 : :
154 : 26 : spin_lock_bh(&msk->pm.lock);
155 : 26 : entry = mptcp_userspace_pm_lookup_addr(msk, skc);
156 [ + + + - ]: 26 : backup = entry && !!(entry->flags & MPTCP_PM_ADDR_FLAG_BACKUP);
157 : 26 : spin_unlock_bh(&msk->pm.lock);
158 : :
159 : 26 : return backup;
160 : : }
161 : :
162 : 146 : static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info)
163 : : {
164 : 146 : struct mptcp_sock *msk;
165 : 146 : struct nlattr *token;
166 : :
167 [ + - - - ]: 146 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_TOKEN))
168 : 0 : return NULL;
169 : :
170 : 146 : token = info->attrs[MPTCP_PM_ATTR_TOKEN];
171 : 146 : msk = mptcp_token_get_sock(genl_info_net(info), nla_get_u32(token));
172 [ + + ]: 146 : if (!msk) {
173 [ + - ]: 4 : NL_SET_ERR_MSG_ATTR(info->extack, token, "invalid token");
174 : 4 : return NULL;
175 : : }
176 : :
177 [ - + ]: 142 : if (!mptcp_pm_is_userspace(msk)) {
178 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, token,
179 : : "userspace PM not selected");
180 : 0 : sock_put((struct sock *)msk);
181 : 0 : return NULL;
182 : : }
183 : :
184 : : return msk;
185 : : }
186 : :
187 : 36 : int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info)
188 : : {
189 : 36 : struct mptcp_pm_addr_entry addr_val;
190 : 36 : struct mptcp_sock *msk;
191 : 36 : struct nlattr *addr;
192 : 36 : int err = -EINVAL;
193 : 36 : struct sock *sk;
194 : :
195 [ + - - - ]: 36 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR))
196 : 0 : return err;
197 : :
198 : 36 : msk = mptcp_userspace_pm_get_sock(info);
199 [ + + ]: 36 : if (!msk)
200 : : return err;
201 : :
202 : 34 : sk = (struct sock *)msk;
203 : :
204 : 34 : addr = info->attrs[MPTCP_PM_ATTR_ADDR];
205 : 34 : err = mptcp_pm_parse_entry(addr, info, true, &addr_val);
206 [ - + ]: 34 : if (err < 0)
207 : 0 : goto announce_err;
208 : :
209 [ - + ]: 34 : if (addr_val.addr.id == 0) {
210 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, addr, "invalid addr id");
211 : 0 : err = -EINVAL;
212 : 0 : goto announce_err;
213 : : }
214 : :
215 [ - + ]: 34 : if (!(addr_val.flags & MPTCP_PM_ADDR_FLAG_SIGNAL)) {
216 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, addr, "invalid addr flags");
217 : 0 : err = -EINVAL;
218 : 0 : goto announce_err;
219 : : }
220 : :
221 : 34 : err = mptcp_userspace_pm_append_new_local_addr(msk, &addr_val, false);
222 [ - + ]: 34 : if (err < 0) {
223 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, addr,
224 : : "did not match address and id");
225 : 0 : goto announce_err;
226 : : }
227 : :
228 : 34 : lock_sock(sk);
229 : 34 : spin_lock_bh(&msk->pm.lock);
230 : :
231 [ + - ]: 34 : if (mptcp_pm_alloc_anno_list(msk, &addr_val.addr)) {
232 : 34 : msk->pm.add_addr_signaled++;
233 : 34 : mptcp_pm_announce_addr(msk, &addr_val.addr, false);
234 : 34 : mptcp_pm_addr_send_ack(msk);
235 : : }
236 : :
237 : 34 : spin_unlock_bh(&msk->pm.lock);
238 : 34 : release_sock(sk);
239 : :
240 : 34 : err = 0;
241 : 34 : announce_err:
242 : 34 : sock_put(sk);
243 : 34 : return err;
244 : : }
245 : :
246 : 2 : static int mptcp_userspace_pm_remove_id_zero_address(struct mptcp_sock *msk)
247 : : {
248 : 2 : struct mptcp_rm_list list = { .nr = 0 };
249 : 2 : struct mptcp_subflow_context *subflow;
250 : 2 : struct sock *sk = (struct sock *)msk;
251 : 2 : bool has_id_0 = false;
252 : 2 : int err = -EINVAL;
253 : :
254 : 2 : lock_sock(sk);
255 [ + - ]: 2 : mptcp_for_each_subflow(msk, subflow) {
256 [ - + ]: 2 : if (READ_ONCE(subflow->local_id) == 0) {
257 : : has_id_0 = true;
258 : : break;
259 : : }
260 : : }
261 [ - + ]: 2 : if (!has_id_0)
262 : 0 : goto remove_err;
263 : :
264 : 2 : list.ids[list.nr++] = 0;
265 : :
266 : 2 : spin_lock_bh(&msk->pm.lock);
267 : 2 : mptcp_pm_remove_addr(msk, &list);
268 : 2 : spin_unlock_bh(&msk->pm.lock);
269 : :
270 : 2 : err = 0;
271 : :
272 : 2 : remove_err:
273 : 2 : release_sock(sk);
274 : 2 : return err;
275 : : }
276 : :
277 : 28 : void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
278 : : struct mptcp_pm_addr_entry *entry)
279 : : {
280 : 28 : struct mptcp_rm_list alist = { .nr = 0 };
281 : 28 : int anno_nr = 0;
282 : :
283 : : /* only delete if either announced or matching a subflow */
284 [ - + ]: 28 : if (mptcp_remove_anno_list_by_saddr(msk, &entry->addr))
285 : : anno_nr++;
286 [ # # ]: 0 : else if (!mptcp_lookup_subflow_by_saddr(&msk->conn_list, &entry->addr))
287 : 0 : return;
288 : :
289 : 28 : alist.ids[alist.nr++] = entry->addr.id;
290 : :
291 : 28 : spin_lock_bh(&msk->pm.lock);
292 : 28 : msk->pm.add_addr_signaled -= anno_nr;
293 : 28 : mptcp_pm_remove_addr(msk, &alist);
294 : 28 : spin_unlock_bh(&msk->pm.lock);
295 : : }
296 : :
297 : 34 : int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
298 : : {
299 : 34 : struct mptcp_pm_addr_entry *match;
300 : 34 : struct mptcp_sock *msk;
301 : 34 : struct nlattr *id;
302 : 34 : int err = -EINVAL;
303 : 34 : struct sock *sk;
304 : 34 : u8 id_val;
305 : :
306 [ + - - - ]: 34 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_LOC_ID))
307 : 0 : return err;
308 : :
309 : 34 : id = info->attrs[MPTCP_PM_ATTR_LOC_ID];
310 : 34 : id_val = nla_get_u8(id);
311 : :
312 : 34 : msk = mptcp_userspace_pm_get_sock(info);
313 [ + + ]: 34 : if (!msk)
314 : : return err;
315 : :
316 : 32 : sk = (struct sock *)msk;
317 : :
318 [ + + ]: 32 : if (id_val == 0) {
319 : 2 : err = mptcp_userspace_pm_remove_id_zero_address(msk);
320 : 2 : goto out;
321 : : }
322 : :
323 : 30 : lock_sock(sk);
324 : :
325 : 30 : spin_lock_bh(&msk->pm.lock);
326 : 30 : match = mptcp_userspace_pm_lookup_addr_by_id(msk, id_val);
327 [ + + ]: 30 : if (!match) {
328 : 2 : spin_unlock_bh(&msk->pm.lock);
329 : 2 : release_sock(sk);
330 : 2 : goto out;
331 : : }
332 : :
333 : 28 : list_del_rcu(&match->list);
334 : 28 : spin_unlock_bh(&msk->pm.lock);
335 : :
336 : 28 : mptcp_pm_remove_addr_entry(msk, match);
337 : :
338 : 28 : release_sock(sk);
339 : :
340 : 28 : kfree_rcu_mightsleep(match);
341 : : /* Adjust sk_omem_alloc like sock_kfree_s() does, to match
342 : : * with allocation of this memory by sock_kmemdup()
343 : : */
344 : 28 : atomic_sub(sizeof(*match), &sk->sk_omem_alloc);
345 : :
346 : 28 : err = 0;
347 : 0 : out:
348 [ + - ]: 32 : if (err)
349 [ - + - + : 2 : NL_SET_ERR_MSG_ATTR_FMT(info->extack, id,
- - ]
350 : : "address with id %u not found",
351 : : id_val);
352 : :
353 : 32 : sock_put(sk);
354 : 32 : return err;
355 : : }
356 : :
357 : 28 : int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info)
358 : : {
359 : 28 : struct mptcp_pm_addr_entry entry = { 0 };
360 : 28 : struct mptcp_addr_info addr_r;
361 : 28 : struct nlattr *raddr, *laddr;
362 : 28 : struct mptcp_pm_local local;
363 : 28 : struct mptcp_sock *msk;
364 : 28 : int err = -EINVAL;
365 : 28 : struct sock *sk;
366 : :
367 [ + - - - ]: 28 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR) ||
368 [ + - - - ]: 28 : GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR_REMOTE))
369 : 0 : return err;
370 : :
371 : 28 : msk = mptcp_userspace_pm_get_sock(info);
372 [ - + ]: 28 : if (!msk)
373 : : return err;
374 : :
375 : 28 : sk = (struct sock *)msk;
376 : :
377 : 28 : laddr = info->attrs[MPTCP_PM_ATTR_ADDR];
378 : 28 : err = mptcp_pm_parse_entry(laddr, info, true, &entry);
379 [ - + ]: 28 : if (err < 0)
380 : 0 : goto create_err;
381 : :
382 [ - + ]: 28 : if (entry.flags & MPTCP_PM_ADDR_FLAG_SIGNAL) {
383 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, laddr, "invalid addr flags");
384 : 0 : err = -EINVAL;
385 : 0 : goto create_err;
386 : : }
387 : 28 : entry.flags |= MPTCP_PM_ADDR_FLAG_SUBFLOW;
388 : :
389 : 28 : raddr = info->attrs[MPTCP_PM_ATTR_ADDR_REMOTE];
390 : 28 : err = mptcp_pm_parse_addr(raddr, info, &addr_r);
391 [ - + ]: 28 : if (err < 0)
392 : 0 : goto create_err;
393 : :
394 [ - + ]: 28 : if (!mptcp_pm_addr_families_match(sk, &entry.addr, &addr_r)) {
395 [ # # ]: 0 : GENL_SET_ERR_MSG(info, "families mismatch");
396 : 0 : err = -EINVAL;
397 : 0 : goto create_err;
398 : : }
399 : :
400 : 28 : err = mptcp_userspace_pm_append_new_local_addr(msk, &entry, false);
401 [ - + ]: 28 : if (err < 0) {
402 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, laddr,
403 : : "did not match address and id");
404 : 0 : goto create_err;
405 : : }
406 : :
407 : 28 : local.addr = entry.addr;
408 : 28 : local.flags = entry.flags;
409 : 28 : local.ifindex = entry.ifindex;
410 : :
411 : 28 : spin_lock_bh(&msk->pm.lock);
412 : 28 : msk->pm.extra_subflows++;
413 : 28 : spin_unlock_bh(&msk->pm.lock);
414 : :
415 : 28 : lock_sock(sk);
416 : 28 : err = __mptcp_subflow_connect(sk, &local, &addr_r);
417 : 28 : release_sock(sk);
418 : :
419 [ + + ]: 28 : if (err) {
420 [ + - - + : 2 : GENL_SET_ERR_MSG_FMT(info, "connect error: %d", err);
- - ]
421 : :
422 : 2 : spin_lock_bh(&msk->pm.lock);
423 : 2 : mptcp_userspace_pm_delete_local_addr(msk, &entry);
424 : 2 : spin_unlock_bh(&msk->pm.lock);
425 : : }
426 : :
427 : 26 : create_err:
428 : 28 : sock_put(sk);
429 : 28 : return err;
430 : : }
431 : :
432 : 20 : static struct sock *mptcp_nl_find_ssk(struct mptcp_sock *msk,
433 : : const struct mptcp_addr_info *local,
434 : : const struct mptcp_addr_info *remote)
435 : : {
436 : 20 : struct mptcp_subflow_context *subflow;
437 : :
438 [ - + ]: 20 : if (local->family != remote->family)
439 : : return NULL;
440 : :
441 [ - + ]: 38 : mptcp_for_each_subflow(msk, subflow) {
442 : 38 : const struct inet_sock *issk;
443 : 38 : struct sock *ssk;
444 : :
445 [ + + ]: 38 : ssk = mptcp_subflow_tcp_sock(subflow);
446 : :
447 [ + + ]: 38 : if (local->family != ssk->sk_family)
448 : 2 : continue;
449 : :
450 : 36 : issk = inet_sk(ssk);
451 : :
452 [ + + - ]: 36 : switch (ssk->sk_family) {
453 : 24 : case AF_INET:
454 [ + + ]: 24 : if (issk->inet_saddr != local->addr.s_addr ||
455 [ - + ]: 14 : issk->inet_daddr != remote->addr.s_addr)
456 : 10 : continue;
457 : : break;
458 : : #if IS_ENABLED(CONFIG_MPTCP_IPV6)
459 : 12 : case AF_INET6: {
460 [ + + - + ]: 12 : if (!ipv6_addr_equal(&local->addr6, &issk->pinet6->saddr) ||
461 [ # # ]: 0 : !ipv6_addr_equal(&remote->addr6, &ssk->sk_v6_daddr))
462 : 6 : continue;
463 : : break;
464 : : }
465 : : #endif
466 : 0 : default:
467 : 0 : continue;
468 : : }
469 : :
470 [ + - ]: 20 : if (issk->inet_sport == local->port &&
471 [ - + ]: 20 : issk->inet_dport == remote->port)
472 : : return ssk;
473 : : }
474 : :
475 : : return NULL;
476 : : }
477 : :
478 : 20 : int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info)
479 : : {
480 : 20 : struct mptcp_pm_addr_entry addr_l;
481 : 20 : struct mptcp_addr_info addr_r;
482 : 20 : struct nlattr *raddr, *laddr;
483 : 20 : struct mptcp_sock *msk;
484 : 20 : struct sock *sk, *ssk;
485 : 20 : int err = -EINVAL;
486 : :
487 [ + - - - ]: 20 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR) ||
488 [ + - - - ]: 20 : GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR_REMOTE))
489 : 0 : return err;
490 : :
491 : 20 : msk = mptcp_userspace_pm_get_sock(info);
492 [ - + ]: 20 : if (!msk)
493 : : return err;
494 : :
495 : 20 : sk = (struct sock *)msk;
496 : :
497 : 20 : laddr = info->attrs[MPTCP_PM_ATTR_ADDR];
498 : 20 : err = mptcp_pm_parse_entry(laddr, info, true, &addr_l);
499 [ - + ]: 20 : if (err < 0)
500 : 0 : goto destroy_err;
501 : :
502 : 20 : raddr = info->attrs[MPTCP_PM_ATTR_ADDR_REMOTE];
503 : 20 : err = mptcp_pm_parse_addr(raddr, info, &addr_r);
504 [ - + ]: 20 : if (err < 0)
505 : 0 : goto destroy_err;
506 : :
507 : : #if IS_ENABLED(CONFIG_MPTCP_IPV6)
508 [ + + - + ]: 20 : if (addr_l.addr.family == AF_INET && ipv6_addr_v4mapped(&addr_r.addr6)) {
509 [ # # ]: 0 : ipv6_addr_set_v4mapped(addr_l.addr.addr.s_addr, &addr_l.addr.addr6);
510 : 0 : addr_l.addr.family = AF_INET6;
511 : : }
512 [ + + - + ]: 20 : if (addr_r.family == AF_INET && ipv6_addr_v4mapped(&addr_l.addr.addr6)) {
513 [ # # ]: 0 : ipv6_addr_set_v4mapped(addr_r.addr.s_addr, &addr_r.addr6);
514 : 0 : addr_r.family = AF_INET6;
515 : : }
516 : : #endif
517 [ - + ]: 20 : if (addr_l.addr.family != addr_r.family) {
518 [ # # ]: 0 : GENL_SET_ERR_MSG(info, "address families do not match");
519 : 0 : err = -EINVAL;
520 : 0 : goto destroy_err;
521 : : }
522 : :
523 [ - + ]: 20 : if (!addr_l.addr.port) {
524 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, laddr, "missing local port");
525 : 0 : err = -EINVAL;
526 : 0 : goto destroy_err;
527 : : }
528 : :
529 [ - + ]: 20 : if (!addr_r.port) {
530 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, raddr, "missing remote port");
531 : 0 : err = -EINVAL;
532 : 0 : goto destroy_err;
533 : : }
534 : :
535 : 20 : lock_sock(sk);
536 : 20 : ssk = mptcp_nl_find_ssk(msk, &addr_l.addr, &addr_r);
537 [ - + ]: 20 : if (!ssk) {
538 [ # # ]: 0 : GENL_SET_ERR_MSG(info, "subflow not found");
539 : 0 : err = -ESRCH;
540 : 0 : goto release_sock;
541 : : }
542 : :
543 : 20 : spin_lock_bh(&msk->pm.lock);
544 : 20 : mptcp_userspace_pm_delete_local_addr(msk, &addr_l);
545 : 20 : spin_unlock_bh(&msk->pm.lock);
546 : 20 : mptcp_subflow_shutdown(sk, ssk, RCV_SHUTDOWN | SEND_SHUTDOWN);
547 : 20 : mptcp_close_ssk(sk, ssk, mptcp_subflow_ctx(ssk));
548 [ + - ]: 20 : MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_RMSUBFLOW);
549 : 20 : release_sock:
550 : 20 : release_sock(sk);
551 : :
552 : 20 : destroy_err:
553 : 20 : sock_put(sk);
554 : 20 : return err;
555 : : }
556 : :
557 : 2 : int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local,
558 : : struct genl_info *info)
559 : : {
560 : 2 : struct mptcp_addr_info rem = { .family = AF_UNSPEC, };
561 : 2 : struct mptcp_pm_addr_entry *entry;
562 : 2 : struct nlattr *attr, *attr_rem;
563 : 2 : struct mptcp_sock *msk;
564 : 2 : int ret = -EINVAL;
565 : 2 : struct sock *sk;
566 : 2 : u8 bkup = 0;
567 : :
568 [ + - - - ]: 2 : if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR_REMOTE))
569 : 0 : return ret;
570 : :
571 : 2 : msk = mptcp_userspace_pm_get_sock(info);
572 [ - + ]: 2 : if (!msk)
573 : : return ret;
574 : :
575 : 2 : sk = (struct sock *)msk;
576 : :
577 : 2 : attr = info->attrs[MPTCP_PM_ATTR_ADDR];
578 [ - + ]: 2 : if (local->addr.family == AF_UNSPEC) {
579 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, attr,
580 : : "invalid local address family");
581 : 0 : ret = -EINVAL;
582 : 0 : goto set_flags_err;
583 : : }
584 : :
585 : 2 : attr_rem = info->attrs[MPTCP_PM_ATTR_ADDR_REMOTE];
586 : 2 : ret = mptcp_pm_parse_addr(attr_rem, info, &rem);
587 [ - + ]: 2 : if (ret < 0)
588 : 0 : goto set_flags_err;
589 : :
590 [ - + ]: 2 : if (rem.family == AF_UNSPEC) {
591 [ # # ]: 0 : NL_SET_ERR_MSG_ATTR(info->extack, attr_rem,
592 : : "invalid remote address family");
593 : 0 : ret = -EINVAL;
594 : 0 : goto set_flags_err;
595 : : }
596 : :
597 [ + - ]: 2 : if (local->flags & MPTCP_PM_ADDR_FLAG_BACKUP)
598 : 2 : bkup = 1;
599 : :
600 : 2 : spin_lock_bh(&msk->pm.lock);
601 : 2 : entry = mptcp_userspace_pm_lookup_addr(msk, &local->addr);
602 [ - + ]: 2 : if (entry) {
603 [ # # ]: 0 : if (bkup)
604 : 0 : entry->flags |= MPTCP_PM_ADDR_FLAG_BACKUP;
605 : : else
606 : 0 : entry->flags &= ~MPTCP_PM_ADDR_FLAG_BACKUP;
607 : : }
608 : 2 : spin_unlock_bh(&msk->pm.lock);
609 : :
610 : 2 : lock_sock(sk);
611 : 2 : ret = mptcp_pm_mp_prio_send_ack(msk, &local->addr, &rem, bkup);
612 : 2 : release_sock(sk);
613 : :
614 : : /* mptcp_pm_mp_prio_send_ack() only fails in one case */
615 [ + - ]: 2 : if (ret < 0)
616 [ # # ]: 0 : GENL_SET_ERR_MSG(info, "subflow not found");
617 : :
618 : 0 : set_flags_err:
619 : 2 : sock_put(sk);
620 : 2 : return ret;
621 : : }
622 : :
623 : 20 : int mptcp_userspace_pm_dump_addr(struct sk_buff *msg,
624 : : struct netlink_callback *cb)
625 : : {
626 : 20 : struct id_bitmap {
627 : : DECLARE_BITMAP(map, MPTCP_PM_MAX_ADDR_ID + 1);
628 : 0 : } *bitmap;
629 : 20 : const struct genl_info *info = genl_info_dump(cb);
630 : 20 : struct mptcp_pm_addr_entry *entry;
631 : 20 : struct mptcp_sock *msk;
632 : 20 : int ret = -EINVAL;
633 : 20 : struct sock *sk;
634 : :
635 : 20 : BUILD_BUG_ON(sizeof(struct id_bitmap) > sizeof(cb->ctx));
636 : :
637 : 20 : bitmap = (struct id_bitmap *)cb->ctx;
638 : :
639 : 20 : msk = mptcp_userspace_pm_get_sock(info);
640 [ + - ]: 20 : if (!msk)
641 : : return ret;
642 : :
643 : 20 : sk = (struct sock *)msk;
644 : :
645 : 20 : lock_sock(sk);
646 : 20 : spin_lock_bh(&msk->pm.lock);
647 [ + + ]: 40 : mptcp_for_each_userspace_pm_addr(msk, entry) {
648 [ - + - - : 40 : if (test_bit(entry->addr.id, bitmap->map))
- - - - +
+ ][ - + -
- - - - -
- + + + ]
649 : 10 : continue;
650 : :
651 [ + - ]: 10 : if (mptcp_pm_genl_fill_addr(msg, cb, entry) < 0)
652 : : break;
653 : :
654 [ - + - - : 10 : __set_bit(entry->addr.id, bitmap->map);
- - - - ]
655 : : }
656 : 20 : spin_unlock_bh(&msk->pm.lock);
657 : 20 : release_sock(sk);
658 : 20 : ret = msg->len;
659 : :
660 : 20 : sock_put(sk);
661 : 20 : return ret;
662 : : }
663 : :
664 : 6 : int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr,
665 : : struct genl_info *info)
666 : : {
667 : 6 : struct mptcp_pm_addr_entry *entry;
668 : 6 : struct mptcp_sock *msk;
669 : 6 : int ret = -EINVAL;
670 : 6 : struct sock *sk;
671 : :
672 : 6 : msk = mptcp_userspace_pm_get_sock(info);
673 [ + - ]: 6 : if (!msk)
674 : : return ret;
675 : :
676 : 6 : sk = (struct sock *)msk;
677 : :
678 : 6 : lock_sock(sk);
679 : 6 : spin_lock_bh(&msk->pm.lock);
680 : 6 : entry = mptcp_userspace_pm_lookup_addr_by_id(msk, id);
681 [ + - ]: 6 : if (entry) {
682 : 6 : *addr = *entry;
683 : 6 : ret = 0;
684 : : }
685 : 6 : spin_unlock_bh(&msk->pm.lock);
686 : 6 : release_sock(sk);
687 : :
688 : 6 : sock_put(sk);
689 : 6 : return ret;
690 : : }
691 : :
692 : : static struct mptcp_pm_ops mptcp_pm_userspace = {
693 : : .name = "userspace",
694 : : .owner = THIS_MODULE,
695 : : };
696 : :
697 : 6 : void __init mptcp_pm_userspace_register(void)
698 : : {
699 : 6 : mptcp_pm_register(&mptcp_pm_userspace);
700 : 6 : }
|