1 | // SPDX-License-Identifier: GPL-2.0-or-later |
2 | /* Socket buffer accounting |
3 | * |
4 | * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved. |
5 | * Written by David Howells (dhowells@redhat.com) |
6 | */ |
7 | |
8 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
9 | |
10 | #include <linux/module.h> |
11 | #include <linux/net.h> |
12 | #include <linux/skbuff.h> |
13 | #include <net/sock.h> |
14 | #include <net/af_rxrpc.h> |
15 | #include "ar-internal.h" |
16 | |
17 | #define select_skb_count(skb) (&rxrpc_n_rx_skbs) |
18 | |
19 | /* |
20 | * Note the allocation or reception of a socket buffer. |
21 | */ |
22 | void rxrpc_new_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) |
23 | { |
24 | int n = atomic_inc_return(select_skb_count(skb)); |
25 | trace_rxrpc_skb(skb, usage: refcount_read(r: &skb->users), mod_count: n, why); |
26 | } |
27 | |
28 | /* |
29 | * Note the re-emergence of a socket buffer from a queue or buffer. |
30 | */ |
31 | void rxrpc_see_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) |
32 | { |
33 | if (skb) { |
34 | int n = atomic_read(select_skb_count(skb)); |
35 | trace_rxrpc_skb(skb, usage: refcount_read(r: &skb->users), mod_count: n, why); |
36 | } |
37 | } |
38 | |
39 | /* |
40 | * Note the addition of a ref on a socket buffer. |
41 | */ |
42 | void rxrpc_get_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) |
43 | { |
44 | int n = atomic_inc_return(select_skb_count(skb)); |
45 | trace_rxrpc_skb(skb, usage: refcount_read(r: &skb->users), mod_count: n, why); |
46 | skb_get(skb); |
47 | } |
48 | |
49 | /* |
50 | * Note the dropping of a ref on a socket buffer by the core. |
51 | */ |
52 | void rxrpc_eaten_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) |
53 | { |
54 | int n = atomic_inc_return(v: &rxrpc_n_rx_skbs); |
55 | trace_rxrpc_skb(skb, usage: 0, mod_count: n, why); |
56 | } |
57 | |
58 | /* |
59 | * Note the destruction of a socket buffer. |
60 | */ |
61 | void rxrpc_free_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) |
62 | { |
63 | if (skb) { |
64 | int n = atomic_dec_return(select_skb_count(skb)); |
65 | trace_rxrpc_skb(skb, usage: refcount_read(r: &skb->users), mod_count: n, why); |
66 | consume_skb(skb); |
67 | } |
68 | } |
69 | |
70 | /* |
71 | * Clear a queue of socket buffers. |
72 | */ |
73 | void rxrpc_purge_queue(struct sk_buff_head *list) |
74 | { |
75 | struct sk_buff *skb; |
76 | |
77 | while ((skb = skb_dequeue((list))) != NULL) { |
78 | int n = atomic_dec_return(select_skb_count(skb)); |
79 | trace_rxrpc_skb(skb, usage: refcount_read(r: &skb->users), mod_count: n, |
80 | why: rxrpc_skb_put_purge); |
81 | consume_skb(skb); |
82 | } |
83 | } |
84 | |