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, refcount_read(&skb->users), 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, refcount_read(&skb->users), 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, refcount_read(&skb->users), n, why); 46 skb_get(skb); 47 } 48 49 /* 50 * Note the destruction of a socket buffer. 51 */ 52 void rxrpc_free_skb(struct sk_buff *skb, enum rxrpc_skb_trace why) 53 { 54 if (skb) { 55 int n = atomic_dec_return(select_skb_count(skb)); 56 trace_rxrpc_skb(skb, refcount_read(&skb->users), n, why); 57 consume_skb(skb); 58 } 59 } 60 61 /* 62 * Clear a queue of socket buffers. 63 */ 64 void rxrpc_purge_queue(struct sk_buff_head *list) 65 { 66 struct sk_buff *skb; 67 68 while ((skb = skb_dequeue((list))) != NULL) { 69 int n = atomic_dec_return(select_skb_count(skb)); 70 trace_rxrpc_skb(skb, refcount_read(&skb->users), n, 71 rxrpc_skb_put_purge); 72 consume_skb(skb); 73 } 74 } 75