Merge branches 'pm-cpuidle' and 'powercap'
[linux-2.6-block.git] / net / core / skmsg.c
CommitLineData
604326b4
DB
1// SPDX-License-Identifier: GPL-2.0
2/* Copyright (c) 2017 - 2018 Covalent IO, Inc. http://covalent.io */
3
4#include <linux/skmsg.h>
5#include <linux/skbuff.h>
6#include <linux/scatterlist.h>
7
8#include <net/sock.h>
9#include <net/tcp.h>
10
11static bool sk_msg_try_coalesce_ok(struct sk_msg *msg, int elem_first_coalesce)
12{
13 if (msg->sg.end > msg->sg.start &&
14 elem_first_coalesce < msg->sg.end)
15 return true;
16
17 if (msg->sg.end < msg->sg.start &&
18 (elem_first_coalesce > msg->sg.start ||
19 elem_first_coalesce < msg->sg.end))
20 return true;
21
22 return false;
23}
24
25int sk_msg_alloc(struct sock *sk, struct sk_msg *msg, int len,
26 int elem_first_coalesce)
27{
28 struct page_frag *pfrag = sk_page_frag(sk);
29 int ret = 0;
30
31 len -= msg->sg.size;
32 while (len > 0) {
33 struct scatterlist *sge;
34 u32 orig_offset;
35 int use, i;
36
37 if (!sk_page_frag_refill(sk, pfrag))
38 return -ENOMEM;
39
40 orig_offset = pfrag->offset;
41 use = min_t(int, len, pfrag->size - orig_offset);
42 if (!sk_wmem_schedule(sk, use))
43 return -ENOMEM;
44
45 i = msg->sg.end;
46 sk_msg_iter_var_prev(i);
47 sge = &msg->sg.data[i];
48
49 if (sk_msg_try_coalesce_ok(msg, elem_first_coalesce) &&
50 sg_page(sge) == pfrag->page &&
51 sge->offset + sge->length == orig_offset) {
52 sge->length += use;
53 } else {
54 if (sk_msg_full(msg)) {
55 ret = -ENOSPC;
56 break;
57 }
58
59 sge = &msg->sg.data[msg->sg.end];
60 sg_unmark_end(sge);
61 sg_set_page(sge, pfrag->page, use, orig_offset);
62 get_page(pfrag->page);
63 sk_msg_iter_next(msg, end);
64 }
65
66 sk_mem_charge(sk, use);
67 msg->sg.size += use;
68 pfrag->offset += use;
69 len -= use;
70 }
71
72 return ret;
73}
74EXPORT_SYMBOL_GPL(sk_msg_alloc);
75
d829e9c4
DB
76int sk_msg_clone(struct sock *sk, struct sk_msg *dst, struct sk_msg *src,
77 u32 off, u32 len)
78{
79 int i = src->sg.start;
80 struct scatterlist *sge = sk_msg_elem(src, i);
81 u32 sge_len, sge_off;
82
83 if (sk_msg_full(dst))
84 return -ENOSPC;
85
86 while (off) {
87 if (sge->length > off)
88 break;
89 off -= sge->length;
90 sk_msg_iter_var_next(i);
91 if (i == src->sg.end && off)
92 return -ENOSPC;
93 sge = sk_msg_elem(src, i);
94 }
95
96 while (len) {
5c1e7e94
VG
97 if (sk_msg_full(dst))
98 return -ENOSPC;
99
d829e9c4
DB
100 sge_len = sge->length - off;
101 sge_off = sge->offset + off;
102 if (sge_len > len)
103 sge_len = len;
104 off = 0;
105 len -= sge_len;
106 sk_msg_page_add(dst, sg_page(sge), sge_len, sge_off);
107 sk_mem_charge(sk, sge_len);
108 sk_msg_iter_var_next(i);
109 if (i == src->sg.end && len)
110 return -ENOSPC;
111 sge = sk_msg_elem(src, i);
112 }
113
114 return 0;
115}
116EXPORT_SYMBOL_GPL(sk_msg_clone);
117
604326b4
DB
118void sk_msg_return_zero(struct sock *sk, struct sk_msg *msg, int bytes)
119{
120 int i = msg->sg.start;
121
122 do {
123 struct scatterlist *sge = sk_msg_elem(msg, i);
124
125 if (bytes < sge->length) {
126 sge->length -= bytes;
127 sge->offset += bytes;
128 sk_mem_uncharge(sk, bytes);
129 break;
130 }
131
132 sk_mem_uncharge(sk, sge->length);
133 bytes -= sge->length;
134 sge->length = 0;
135 sge->offset = 0;
136 sk_msg_iter_var_next(i);
137 } while (bytes && i != msg->sg.end);
138 msg->sg.start = i;
139}
140EXPORT_SYMBOL_GPL(sk_msg_return_zero);
141
142void sk_msg_return(struct sock *sk, struct sk_msg *msg, int bytes)
143{
144 int i = msg->sg.start;
145
146 do {
147 struct scatterlist *sge = &msg->sg.data[i];
148 int uncharge = (bytes < sge->length) ? bytes : sge->length;
149
150 sk_mem_uncharge(sk, uncharge);
151 bytes -= uncharge;
152 sk_msg_iter_var_next(i);
153 } while (i != msg->sg.end);
154}
155EXPORT_SYMBOL_GPL(sk_msg_return);
156
157static int sk_msg_free_elem(struct sock *sk, struct sk_msg *msg, u32 i,
158 bool charge)
159{
160 struct scatterlist *sge = sk_msg_elem(msg, i);
161 u32 len = sge->length;
162
163 if (charge)
164 sk_mem_uncharge(sk, len);
165 if (!msg->skb)
166 put_page(sg_page(sge));
167 memset(sge, 0, sizeof(*sge));
168 return len;
169}
170
171static int __sk_msg_free(struct sock *sk, struct sk_msg *msg, u32 i,
172 bool charge)
173{
174 struct scatterlist *sge = sk_msg_elem(msg, i);
175 int freed = 0;
176
177 while (msg->sg.size) {
178 msg->sg.size -= sge->length;
179 freed += sk_msg_free_elem(sk, msg, i, charge);
180 sk_msg_iter_var_next(i);
181 sk_msg_check_to_free(msg, i, msg->sg.size);
182 sge = sk_msg_elem(msg, i);
183 }
184 if (msg->skb)
185 consume_skb(msg->skb);
186 sk_msg_init(msg);
187 return freed;
188}
189
190int sk_msg_free_nocharge(struct sock *sk, struct sk_msg *msg)
191{
192 return __sk_msg_free(sk, msg, msg->sg.start, false);
193}
194EXPORT_SYMBOL_GPL(sk_msg_free_nocharge);
195
196int sk_msg_free(struct sock *sk, struct sk_msg *msg)
197{
198 return __sk_msg_free(sk, msg, msg->sg.start, true);
199}
200EXPORT_SYMBOL_GPL(sk_msg_free);
201
202static void __sk_msg_free_partial(struct sock *sk, struct sk_msg *msg,
203 u32 bytes, bool charge)
204{
205 struct scatterlist *sge;
206 u32 i = msg->sg.start;
207
208 while (bytes) {
209 sge = sk_msg_elem(msg, i);
210 if (!sge->length)
211 break;
212 if (bytes < sge->length) {
213 if (charge)
214 sk_mem_uncharge(sk, bytes);
215 sge->length -= bytes;
216 sge->offset += bytes;
217 msg->sg.size -= bytes;
218 break;
219 }
220
221 msg->sg.size -= sge->length;
222 bytes -= sge->length;
223 sk_msg_free_elem(sk, msg, i, charge);
224 sk_msg_iter_var_next(i);
225 sk_msg_check_to_free(msg, i, bytes);
226 }
227 msg->sg.start = i;
228}
229
230void sk_msg_free_partial(struct sock *sk, struct sk_msg *msg, u32 bytes)
231{
232 __sk_msg_free_partial(sk, msg, bytes, true);
233}
234EXPORT_SYMBOL_GPL(sk_msg_free_partial);
235
236void sk_msg_free_partial_nocharge(struct sock *sk, struct sk_msg *msg,
237 u32 bytes)
238{
239 __sk_msg_free_partial(sk, msg, bytes, false);
240}
241
242void sk_msg_trim(struct sock *sk, struct sk_msg *msg, int len)
243{
244 int trim = msg->sg.size - len;
245 u32 i = msg->sg.end;
246
247 if (trim <= 0) {
248 WARN_ON(trim < 0);
249 return;
250 }
251
252 sk_msg_iter_var_prev(i);
253 msg->sg.size = len;
254 while (msg->sg.data[i].length &&
255 trim >= msg->sg.data[i].length) {
256 trim -= msg->sg.data[i].length;
257 sk_msg_free_elem(sk, msg, i, true);
258 sk_msg_iter_var_prev(i);
259 if (!trim)
260 goto out;
261 }
262
263 msg->sg.data[i].length -= trim;
264 sk_mem_uncharge(sk, trim);
265out:
266 /* If we trim data before curr pointer update copybreak and current
267 * so that any future copy operations start at new copy location.
268 * However trimed data that has not yet been used in a copy op
269 * does not require an update.
270 */
271 if (msg->sg.curr >= i) {
272 msg->sg.curr = i;
273 msg->sg.copybreak = msg->sg.data[i].length;
274 }
275 sk_msg_iter_var_next(i);
276 msg->sg.end = i;
277}
278EXPORT_SYMBOL_GPL(sk_msg_trim);
279
280int sk_msg_zerocopy_from_iter(struct sock *sk, struct iov_iter *from,
281 struct sk_msg *msg, u32 bytes)
282{
283 int i, maxpages, ret = 0, num_elems = sk_msg_elem_used(msg);
284 const int to_max_pages = MAX_MSG_FRAGS;
285 struct page *pages[MAX_MSG_FRAGS];
286 ssize_t orig, copied, use, offset;
287
288 orig = msg->sg.size;
289 while (bytes > 0) {
290 i = 0;
291 maxpages = to_max_pages - num_elems;
292 if (maxpages == 0) {
293 ret = -EFAULT;
294 goto out;
295 }
296
297 copied = iov_iter_get_pages(from, pages, bytes, maxpages,
298 &offset);
299 if (copied <= 0) {
300 ret = -EFAULT;
301 goto out;
302 }
303
304 iov_iter_advance(from, copied);
305 bytes -= copied;
306 msg->sg.size += copied;
307
308 while (copied) {
309 use = min_t(int, copied, PAGE_SIZE - offset);
310 sg_set_page(&msg->sg.data[msg->sg.end],
311 pages[i], use, offset);
312 sg_unmark_end(&msg->sg.data[msg->sg.end]);
313 sk_mem_charge(sk, use);
314
315 offset = 0;
316 copied -= use;
317 sk_msg_iter_next(msg, end);
318 num_elems++;
319 i++;
320 }
321 /* When zerocopy is mixed with sk_msg_*copy* operations we
322 * may have a copybreak set in this case clear and prefer
323 * zerocopy remainder when possible.
324 */
325 msg->sg.copybreak = 0;
326 msg->sg.curr = msg->sg.end;
327 }
328out:
329 /* Revert iov_iter updates, msg will need to use 'trim' later if it
330 * also needs to be cleared.
331 */
332 if (ret)
333 iov_iter_revert(from, msg->sg.size - orig);
334 return ret;
335}
336EXPORT_SYMBOL_GPL(sk_msg_zerocopy_from_iter);
337
338int sk_msg_memcopy_from_iter(struct sock *sk, struct iov_iter *from,
339 struct sk_msg *msg, u32 bytes)
340{
341 int ret = -ENOSPC, i = msg->sg.curr;
342 struct scatterlist *sge;
343 u32 copy, buf_size;
344 void *to;
345
346 do {
347 sge = sk_msg_elem(msg, i);
348 /* This is possible if a trim operation shrunk the buffer */
349 if (msg->sg.copybreak >= sge->length) {
350 msg->sg.copybreak = 0;
351 sk_msg_iter_var_next(i);
352 if (i == msg->sg.end)
353 break;
354 sge = sk_msg_elem(msg, i);
355 }
356
357 buf_size = sge->length - msg->sg.copybreak;
358 copy = (buf_size > bytes) ? bytes : buf_size;
359 to = sg_virt(sge) + msg->sg.copybreak;
360 msg->sg.copybreak += copy;
361 if (sk->sk_route_caps & NETIF_F_NOCACHE_COPY)
362 ret = copy_from_iter_nocache(to, copy, from);
363 else
364 ret = copy_from_iter(to, copy, from);
365 if (ret != copy) {
366 ret = -EFAULT;
367 goto out;
368 }
369 bytes -= copy;
370 if (!bytes)
371 break;
372 msg->sg.copybreak = 0;
373 sk_msg_iter_var_next(i);
374 } while (i != msg->sg.end);
375out:
376 msg->sg.curr = i;
377 return ret;
378}
379EXPORT_SYMBOL_GPL(sk_msg_memcopy_from_iter);
380
381static int sk_psock_skb_ingress(struct sk_psock *psock, struct sk_buff *skb)
382{
383 struct sock *sk = psock->sk;
384 int copied = 0, num_sge;
385 struct sk_msg *msg;
386
387 msg = kzalloc(sizeof(*msg), __GFP_NOWARN | GFP_ATOMIC);
388 if (unlikely(!msg))
389 return -EAGAIN;
390 if (!sk_rmem_schedule(sk, skb, skb->len)) {
391 kfree(msg);
392 return -EAGAIN;
393 }
394
395 sk_msg_init(msg);
396 num_sge = skb_to_sgvec(skb, msg->sg.data, 0, skb->len);
397 if (unlikely(num_sge < 0)) {
398 kfree(msg);
399 return num_sge;
400 }
401
402 sk_mem_charge(sk, skb->len);
403 copied = skb->len;
404 msg->sg.start = 0;
405 msg->sg.end = num_sge == MAX_MSG_FRAGS ? 0 : num_sge;
406 msg->skb = skb;
407
408 sk_psock_queue_msg(psock, msg);
552de910 409 sk_psock_data_ready(sk, psock);
604326b4
DB
410 return copied;
411}
412
413static int sk_psock_handle_skb(struct sk_psock *psock, struct sk_buff *skb,
414 u32 off, u32 len, bool ingress)
415{
416 if (ingress)
417 return sk_psock_skb_ingress(psock, skb);
418 else
419 return skb_send_sock_locked(psock->sk, skb, off, len);
420}
421
422static void sk_psock_backlog(struct work_struct *work)
423{
424 struct sk_psock *psock = container_of(work, struct sk_psock, work);
425 struct sk_psock_work_state *state = &psock->work_state;
426 struct sk_buff *skb;
427 bool ingress;
428 u32 len, off;
429 int ret;
430
431 /* Lock sock to avoid losing sk_socket during loop. */
432 lock_sock(psock->sk);
433 if (state->skb) {
434 skb = state->skb;
435 len = state->len;
436 off = state->off;
437 state->skb = NULL;
438 goto start;
439 }
440
441 while ((skb = skb_dequeue(&psock->ingress_skb))) {
442 len = skb->len;
443 off = 0;
444start:
445 ingress = tcp_skb_bpf_ingress(skb);
446 do {
447 ret = -EIO;
448 if (likely(psock->sk->sk_socket))
449 ret = sk_psock_handle_skb(psock, skb, off,
450 len, ingress);
451 if (ret <= 0) {
452 if (ret == -EAGAIN) {
453 state->skb = skb;
454 state->len = len;
455 state->off = off;
456 goto end;
457 }
458 /* Hard errors break pipe and stop xmit. */
459 sk_psock_report_error(psock, ret ? -ret : EPIPE);
460 sk_psock_clear_state(psock, SK_PSOCK_TX_ENABLED);
461 kfree_skb(skb);
462 goto end;
463 }
464 off += ret;
465 len -= ret;
466 } while (len);
467
468 if (!ingress)
469 kfree_skb(skb);
470 }
471end:
472 release_sock(psock->sk);
473}
474
475struct sk_psock *sk_psock_init(struct sock *sk, int node)
476{
477 struct sk_psock *psock = kzalloc_node(sizeof(*psock),
478 GFP_ATOMIC | __GFP_NOWARN,
479 node);
480 if (!psock)
481 return NULL;
482
483 psock->sk = sk;
484 psock->eval = __SK_NONE;
485
486 INIT_LIST_HEAD(&psock->link);
487 spin_lock_init(&psock->link_lock);
488
489 INIT_WORK(&psock->work, sk_psock_backlog);
490 INIT_LIST_HEAD(&psock->ingress_msg);
491 skb_queue_head_init(&psock->ingress_skb);
492
493 sk_psock_set_state(psock, SK_PSOCK_TX_ENABLED);
494 refcount_set(&psock->refcnt, 1);
495
496 rcu_assign_sk_user_data(sk, psock);
497 sock_hold(sk);
498
499 return psock;
500}
501EXPORT_SYMBOL_GPL(sk_psock_init);
502
503struct sk_psock_link *sk_psock_link_pop(struct sk_psock *psock)
504{
505 struct sk_psock_link *link;
506
507 spin_lock_bh(&psock->link_lock);
508 link = list_first_entry_or_null(&psock->link, struct sk_psock_link,
509 list);
510 if (link)
511 list_del(&link->list);
512 spin_unlock_bh(&psock->link_lock);
513 return link;
514}
515
516void __sk_psock_purge_ingress_msg(struct sk_psock *psock)
517{
518 struct sk_msg *msg, *tmp;
519
520 list_for_each_entry_safe(msg, tmp, &psock->ingress_msg, list) {
521 list_del(&msg->list);
522 sk_msg_free(psock->sk, msg);
523 kfree(msg);
524 }
525}
526
527static void sk_psock_zap_ingress(struct sk_psock *psock)
528{
529 __skb_queue_purge(&psock->ingress_skb);
530 __sk_psock_purge_ingress_msg(psock);
531}
532
533static void sk_psock_link_destroy(struct sk_psock *psock)
534{
535 struct sk_psock_link *link, *tmp;
536
537 list_for_each_entry_safe(link, tmp, &psock->link, list) {
538 list_del(&link->list);
539 sk_psock_free_link(link);
540 }
541}
542
543static void sk_psock_destroy_deferred(struct work_struct *gc)
544{
545 struct sk_psock *psock = container_of(gc, struct sk_psock, gc);
546
547 /* No sk_callback_lock since already detached. */
1d79895a 548 strp_done(&psock->parser.strp);
604326b4
DB
549
550 cancel_work_sync(&psock->work);
551
552 psock_progs_drop(&psock->progs);
553
554 sk_psock_link_destroy(psock);
555 sk_psock_cork_free(psock);
556 sk_psock_zap_ingress(psock);
557
558 if (psock->sk_redir)
559 sock_put(psock->sk_redir);
560 sock_put(psock->sk);
561 kfree(psock);
562}
563
564void sk_psock_destroy(struct rcu_head *rcu)
565{
566 struct sk_psock *psock = container_of(rcu, struct sk_psock, rcu);
567
568 INIT_WORK(&psock->gc, sk_psock_destroy_deferred);
569 schedule_work(&psock->gc);
570}
571EXPORT_SYMBOL_GPL(sk_psock_destroy);
572
573void sk_psock_drop(struct sock *sk, struct sk_psock *psock)
574{
575 rcu_assign_sk_user_data(sk, NULL);
576 sk_psock_cork_free(psock);
a136678c 577 sk_psock_zap_ingress(psock);
604326b4
DB
578 sk_psock_restore_proto(sk, psock);
579
580 write_lock_bh(&sk->sk_callback_lock);
581 if (psock->progs.skb_parser)
582 sk_psock_stop_strp(sk, psock);
583 write_unlock_bh(&sk->sk_callback_lock);
584 sk_psock_clear_state(psock, SK_PSOCK_TX_ENABLED);
585
0245b80e 586 call_rcu(&psock->rcu, sk_psock_destroy);
604326b4
DB
587}
588EXPORT_SYMBOL_GPL(sk_psock_drop);
589
590static int sk_psock_map_verd(int verdict, bool redir)
591{
592 switch (verdict) {
593 case SK_PASS:
594 return redir ? __SK_REDIRECT : __SK_PASS;
595 case SK_DROP:
596 default:
597 break;
598 }
599
600 return __SK_DROP;
601}
602
603int sk_psock_msg_verdict(struct sock *sk, struct sk_psock *psock,
604 struct sk_msg *msg)
605{
606 struct bpf_prog *prog;
607 int ret;
608
609 preempt_disable();
610 rcu_read_lock();
611 prog = READ_ONCE(psock->progs.msg_parser);
612 if (unlikely(!prog)) {
613 ret = __SK_PASS;
614 goto out;
615 }
616
617 sk_msg_compute_data_pointers(msg);
618 msg->sk = sk;
619 ret = BPF_PROG_RUN(prog, msg);
620 ret = sk_psock_map_verd(ret, msg->sk_redir);
621 psock->apply_bytes = msg->apply_bytes;
622 if (ret == __SK_REDIRECT) {
623 if (psock->sk_redir)
624 sock_put(psock->sk_redir);
625 psock->sk_redir = msg->sk_redir;
626 if (!psock->sk_redir) {
627 ret = __SK_DROP;
628 goto out;
629 }
630 sock_hold(psock->sk_redir);
631 }
632out:
633 rcu_read_unlock();
634 preempt_enable();
635 return ret;
636}
637EXPORT_SYMBOL_GPL(sk_psock_msg_verdict);
638
639static int sk_psock_bpf_run(struct sk_psock *psock, struct bpf_prog *prog,
640 struct sk_buff *skb)
641{
642 int ret;
643
644 skb->sk = psock->sk;
645 bpf_compute_data_end_sk_skb(skb);
646 preempt_disable();
647 ret = BPF_PROG_RUN(prog, skb);
648 preempt_enable();
649 /* strparser clones the skb before handing it to a upper layer,
650 * meaning skb_orphan has been called. We NULL sk on the way out
651 * to ensure we don't trigger a BUG_ON() in skb/sk operations
652 * later and because we are not charging the memory of this skb
653 * to any socket yet.
654 */
655 skb->sk = NULL;
656 return ret;
657}
658
659static struct sk_psock *sk_psock_from_strp(struct strparser *strp)
660{
661 struct sk_psock_parser *parser;
662
663 parser = container_of(strp, struct sk_psock_parser, strp);
664 return container_of(parser, struct sk_psock, parser);
665}
666
667static void sk_psock_verdict_apply(struct sk_psock *psock,
668 struct sk_buff *skb, int verdict)
669{
670 struct sk_psock *psock_other;
671 struct sock *sk_other;
672 bool ingress;
673
674 switch (verdict) {
51199405
JF
675 case __SK_PASS:
676 sk_other = psock->sk;
677 if (sock_flag(sk_other, SOCK_DEAD) ||
678 !sk_psock_test_state(psock, SK_PSOCK_TX_ENABLED)) {
679 goto out_free;
680 }
681 if (atomic_read(&sk_other->sk_rmem_alloc) <=
682 sk_other->sk_rcvbuf) {
683 struct tcp_skb_cb *tcp = TCP_SKB_CB(skb);
684
685 tcp->bpf.flags |= BPF_F_INGRESS;
686 skb_queue_tail(&psock->ingress_skb, skb);
687 schedule_work(&psock->work);
688 break;
689 }
690 goto out_free;
604326b4
DB
691 case __SK_REDIRECT:
692 sk_other = tcp_skb_bpf_redirect_fetch(skb);
693 if (unlikely(!sk_other))
694 goto out_free;
695 psock_other = sk_psock(sk_other);
696 if (!psock_other || sock_flag(sk_other, SOCK_DEAD) ||
697 !sk_psock_test_state(psock_other, SK_PSOCK_TX_ENABLED))
698 goto out_free;
699 ingress = tcp_skb_bpf_ingress(skb);
700 if ((!ingress && sock_writeable(sk_other)) ||
701 (ingress &&
702 atomic_read(&sk_other->sk_rmem_alloc) <=
703 sk_other->sk_rcvbuf)) {
704 if (!ingress)
705 skb_set_owner_w(skb, sk_other);
706 skb_queue_tail(&psock_other->ingress_skb, skb);
707 schedule_work(&psock_other->work);
708 break;
709 }
710 /* fall-through */
711 case __SK_DROP:
712 /* fall-through */
713 default:
714out_free:
715 kfree_skb(skb);
716 }
717}
718
719static void sk_psock_strp_read(struct strparser *strp, struct sk_buff *skb)
720{
721 struct sk_psock *psock = sk_psock_from_strp(strp);
722 struct bpf_prog *prog;
723 int ret = __SK_DROP;
724
725 rcu_read_lock();
726 prog = READ_ONCE(psock->progs.skb_verdict);
727 if (likely(prog)) {
728 skb_orphan(skb);
729 tcp_skb_bpf_redirect_clear(skb);
730 ret = sk_psock_bpf_run(psock, prog, skb);
731 ret = sk_psock_map_verd(ret, tcp_skb_bpf_redirect_fetch(skb));
732 }
733 rcu_read_unlock();
734 sk_psock_verdict_apply(psock, skb, ret);
735}
736
737static int sk_psock_strp_read_done(struct strparser *strp, int err)
738{
739 return err;
740}
741
742static int sk_psock_strp_parse(struct strparser *strp, struct sk_buff *skb)
743{
744 struct sk_psock *psock = sk_psock_from_strp(strp);
745 struct bpf_prog *prog;
746 int ret = skb->len;
747
748 rcu_read_lock();
749 prog = READ_ONCE(psock->progs.skb_parser);
750 if (likely(prog))
751 ret = sk_psock_bpf_run(psock, prog, skb);
752 rcu_read_unlock();
753 return ret;
754}
755
756/* Called with socket lock held. */
552de910 757static void sk_psock_strp_data_ready(struct sock *sk)
604326b4
DB
758{
759 struct sk_psock *psock;
760
761 rcu_read_lock();
762 psock = sk_psock(sk);
763 if (likely(psock)) {
764 write_lock_bh(&sk->sk_callback_lock);
765 strp_data_ready(&psock->parser.strp);
766 write_unlock_bh(&sk->sk_callback_lock);
767 }
768 rcu_read_unlock();
769}
770
771static void sk_psock_write_space(struct sock *sk)
772{
773 struct sk_psock *psock;
774 void (*write_space)(struct sock *sk);
775
776 rcu_read_lock();
777 psock = sk_psock(sk);
778 if (likely(psock && sk_psock_test_state(psock, SK_PSOCK_TX_ENABLED)))
779 schedule_work(&psock->work);
780 write_space = psock->saved_write_space;
781 rcu_read_unlock();
782 write_space(sk);
783}
784
785int sk_psock_init_strp(struct sock *sk, struct sk_psock *psock)
786{
787 static const struct strp_callbacks cb = {
788 .rcv_msg = sk_psock_strp_read,
789 .read_sock_done = sk_psock_strp_read_done,
790 .parse_msg = sk_psock_strp_parse,
791 };
792
793 psock->parser.enabled = false;
794 return strp_init(&psock->parser.strp, sk, &cb);
795}
796
797void sk_psock_start_strp(struct sock *sk, struct sk_psock *psock)
798{
799 struct sk_psock_parser *parser = &psock->parser;
800
801 if (parser->enabled)
802 return;
803
804 parser->saved_data_ready = sk->sk_data_ready;
552de910 805 sk->sk_data_ready = sk_psock_strp_data_ready;
604326b4
DB
806 sk->sk_write_space = sk_psock_write_space;
807 parser->enabled = true;
808}
809
810void sk_psock_stop_strp(struct sock *sk, struct sk_psock *psock)
811{
812 struct sk_psock_parser *parser = &psock->parser;
813
814 if (!parser->enabled)
815 return;
816
817 sk->sk_data_ready = parser->saved_data_ready;
818 parser->saved_data_ready = NULL;
819 strp_stop(&parser->strp);
820 parser->enabled = false;
821}